Add metadata to track X-Ray segments

DarcyRaynerDD · DarcyRaynerDD · commit 3b47b3963eda · 2020-04-03T15:11:05.000-04:00
diff --git a/datadog_lambda/tracing.py b/datadog_lambda/tracing.py
@@ -233,7 +233,9 @@ def set_dd_trace_py_root(trace_context, merge_xray_traces):
         tracer.context_provider.activate(span_context)
 
 
-def create_function_execution_span(context, function_name, is_cold_start):
+def create_function_execution_span(
+    context, function_name, is_cold_start, trace_context
+):
     tags = {}
     if context:
         tags = {
@@ -242,6 +244,10 @@ def create_function_execution_span(context, function_name, is_cold_start):
             "request_id": context.aws_request_id,
             "resource_names": context.function_name,
         }
+    source = trace_context["source"]
+    if source != TraceContextSource.DDTRACE:
+        tags["_dd.parent_source"] = source
+
     args = {
         "service": "aws.lambda",
         "resource": function_name,
diff --git a/datadog_lambda/wrapper.py b/datadog_lambda/wrapper.py
@@ -123,11 +123,10 @@ def _before(self, event, context):
             dd_context = extract_dd_trace_context(event)
 
             self.span = None
-
             if dd_tracing_enabled:
                 set_dd_trace_py_root(dd_context, self.merge_xray_traces)
                 self.span = create_function_execution_span(
-                    context, self.function_name, is_cold_start()
+                    context, self.function_name, is_cold_start(), dd_context
                 )
             else:
                 set_correlation_ids()
