want to keep looping

michael-zhao459 · michael-zhao459 · commit 63e41cb940e2 · 2025-08-06T11:40:08.000-04:00
diff --git a/datadog_lambda/tracing.py b/datadog_lambda/tracing.py
@@ -328,11 +328,13 @@ def extract_context_from_sqs_or_sns_event_or_context(
                             logger.debug(
                                 "Found dd-trace injected trace context from AWSTraceHeader"
                             )
-                            return Context(
+                            context = Context(
                                 trace_id=int(trace_id_parts[2][8:], 16),
                                 span_id=int(x_ray_context["parent_id"], 16),
                                 sampling_priority=float(x_ray_context["sampled"]),
                             )
+                            if not config.data_streams_enabled:
+                                break
         except Exception as e:
             logger.debug("The trace extractor returned with error %s", e)
 

Original file line number	Diff line number	Diff line change
`@@ -328,11 +328,13 @@ def extract_context_from_sqs_or_sns_event_or_context(`
`328`	`328`	`logger.debug(`
`329`	`329`	`"Found dd-trace injected trace context from AWSTraceHeader"`
`330`	`330`	`)`
`331`		`- return Context(`
	`331`	`+ context = Context(`
`332`	`332`	`trace_id=int(trace_id_parts[2][8:], 16),`
`333`	`333`	`span_id=int(x_ray_context["parent_id"], 16),`
`334`	`334`	`sampling_priority=float(x_ray_context["sampled"]),`
`335`	`335`	`)`
	`336`	`+ if not config.data_streams_enabled:`
	`337`	`+ break`
`336`	`338`	`except Exception as e:`
`337`	`339`	`logger.debug("The trace extractor returned with error %s", e)`
`338`	`340`