diff --git a/datadog_lambda/tracing.py b/datadog_lambda/tracing.py index bed55eba..d8ecca1d 100644 --- a/datadog_lambda/tracing.py +++ b/datadog_lambda/tracing.py @@ -139,7 +139,7 @@ def extract_context_from_sqs_event_or_context(event, lambda_context): try: first_record = event["Records"][0] msg_attributes = first_record.get("messageAttributes", {}) - dd_json_data = msg_attributes.get("_datadog", {}).get("StringValue", r"{}") + dd_json_data = msg_attributes.get("_datadog", {}).get("stringValue", r"{}") dd_data = json.loads(dd_json_data) trace_id = dd_data.get(TraceHeader.TRACE_ID) parent_id = dd_data.get(TraceHeader.PARENT_ID) diff --git a/tests/integration/input_events/sqs.json b/tests/integration/input_events/sqs.json index 3f672703..0ec29df4 100644 --- a/tests/integration/input_events/sqs.json +++ b/tests/integration/input_events/sqs.json @@ -12,7 +12,7 @@ }, "messageAttributes": { "_datadog": { - "StringValue": "{\"x-datadog-trace-id\":\"666\",\"x-datadog-parent-id\":\"777\",\"x-datadog-sampling-priority\":\"1\"}" + "stringValue": "{\"x-datadog-trace-id\":\"666\",\"x-datadog-parent-id\":\"777\",\"x-datadog-sampling-priority\":\"1\"}" } }, "md5OfBody": "e4e68fb7bd0e697a0ae8f1bb342846b3", diff --git a/tests/test_tracing.py b/tests/test_tracing.py index 2b93882f..0c8f8154 100644 --- a/tests/test_tracing.py +++ b/tests/test_tracing.py @@ -166,7 +166,7 @@ def test_with_sqs_distributed_datadog_trace_data(self): }, "messageAttributes": { "_datadog": { - "StringValue": json.dumps( + "stringValue": json.dumps( { TraceHeader.TRACE_ID: "123", TraceHeader.PARENT_ID: "321",