Allow python X-Ray to be set via env

Closes lambci#61
joaoasrosa · Jan 22, 2018 · c967e90 · c967e90
1 parent f94bdd5
commit c967e90
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python2.7/run/runtime-mock.py b/python2.7/run/runtime-mock.py
@@ -52,7 +52,7 @@ def _arn(region, account_id, fct_name):
 _GLOBAL_XRAY_TRACE_ID = None
 _GLOBAL_XRAY_PARENT_ID = None
 _GLOBAL_XRAY_SAMPLED = None
-_GLOBAL_X_AMZN_TRACE_ID = None
+_GLOBAL_X_AMZN_TRACE_ID = os.environ.get('_X_AMZN_TRACE_ID', None)
 _GLOBAL_INVOKED = False
 _GLOBAL_ERRORED = False
 _GLOBAL_START_TIME = None

diff --git a/python3.6/run/runtime-mock.py b/python3.6/run/runtime-mock.py
@@ -52,7 +52,7 @@ def _arn(region, account_id, fct_name):
 _GLOBAL_XRAY_TRACE_ID = None
 _GLOBAL_XRAY_PARENT_ID = None
 _GLOBAL_XRAY_SAMPLED = None
-_GLOBAL_X_AMZN_TRACE_ID = None
+_GLOBAL_X_AMZN_TRACE_ID = os.environ.get('_X_AMZN_TRACE_ID', None)
 _GLOBAL_INVOKED = False
 _GLOBAL_ERRORED = False
 _GLOBAL_START_TIME = None