Skip to content

Commit 28ad948

Browse files
committed
Revert "Simplify SQS/SNS data extraction logic."
This reverts commit 7d1753f.
1 parent 7d1753f commit 28ad948

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

datadog_lambda/tracing.py

+9-3
Original file line numberDiff line numberDiff line change
@@ -253,13 +253,19 @@ def extract_context_from_sqs_or_sns_event_or_context(event, lambda_context):
253253
# SQS uses dataType and binaryValue/stringValue
254254
# SNS uses Type and Value
255255
dd_json_data = None
256-
dd_json_data_type = dd_payload.get("Type") or dd_payload.get("dataType")
256+
dd_json_data_type = dd_payload.get("Type")
257+
if dd_json_data_type is None:
258+
dd_json_data_type = dd_payload.get("dataType")
257259
if dd_json_data_type == "Binary":
258-
dd_json_data = dd_payload.get("binaryValue") or dd_payload.get("Value")
260+
dd_json_data = dd_payload.get("binaryValue")
261+
if dd_json_data is None:
262+
dd_json_data = dd_payload.get("Value")
259263
if dd_json_data:
260264
dd_json_data = base64.b64decode(dd_json_data)
261265
elif dd_json_data_type == "String":
262-
dd_json_data = dd_payload.get("stringValue") or dd_payload.get("Value")
266+
dd_json_data = dd_payload.get("stringValue")
267+
if dd_json_data is None:
268+
dd_json_data = dd_payload.get("Value")
263269
else:
264270
logger.debug(
265271
"Datadog Lambda Python only supports extracting trace"

0 commit comments

Comments
 (0)