Improve error log message when no sns->sqs context found.

purple4reina · purple4reina · commit dfef34ffe72b · 2023-11-21T12:18:38.000-08:00
diff --git a/datadog_lambda/tracing.py b/datadog_lambda/tracing.py
@@ -266,6 +266,7 @@ def extract_context_from_sqs_or_sns_event_or_context(event, lambda_context):
                 "Datadog Lambda Python only supports extracting trace"
                 "context from String or Binary SQS/SNS message attributes"
             )
+            return extract_context_from_lambda_context(lambda_context)
         dd_data = json.loads(dd_json_data)
         return propagator.extract(dd_data)
     except Exception as e:

Original file line number	Diff line number	Diff line change
`@@ -266,6 +266,7 @@ def extract_context_from_sqs_or_sns_event_or_context(event, lambda_context):`
`266`	`266`	`"Datadog Lambda Python only supports extracting trace"`
`267`	`267`	`"context from String or Binary SQS/SNS message attributes"`
`268`	`268`	`)`
	`269`	`+ return extract_context_from_lambda_context(lambda_context)`
`269`	`270`	`dd_data = json.loads(dd_json_data)`
`270`	`271`	`return propagator.extract(dd_data)`
`271`	`272`	`except Exception as e:`