File tree 1 file changed +9
-3
lines changed
1 file changed +9
-3
lines changed Original file line number Diff line number Diff line change @@ -253,13 +253,19 @@ def extract_context_from_sqs_or_sns_event_or_context(event, lambda_context):
253
253
# SQS uses dataType and binaryValue/stringValue
254
254
# SNS uses Type and Value
255
255
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" )
257
259
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" )
259
263
if dd_json_data :
260
264
dd_json_data = base64 .b64decode (dd_json_data )
261
265
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" )
263
269
else :
264
270
logger .debug (
265
271
"Datadog Lambda Python only supports extracting trace"
You can’t perform that action at this time.
0 commit comments