File tree Expand file tree Collapse file tree 1 file changed +7
-3
lines changed
Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Original file line number Diff line number Diff line change @@ -12,23 +12,27 @@ def _dsm_set_sqs_context(event):
1212 from datadog_lambda .wrapper import format_err_with_traceback
1313
1414 from ddtrace .internal .datastreams .processor import (
15- DataStreamsProcessor as processor ,
15+ DataStreamsProcessor ,
1616 DsmPathwayCodec ,
1717 )
1818 from ddtrace .internal .datastreams .botocore import (
1919 get_datastreams_context ,
2020 calculate_sqs_payload_size ,
2121 )
2222
23- records = event .get ("Records" , [])
23+ records = event .get ("Records" )
24+ if records is None :
25+ return
26+ processor = DataStreamsProcessor ()
27+
2428 for record in records :
2529 try :
2630 queue_arn = record .get ("eventSourceARN" , "" )
2731
2832 contextjson = get_datastreams_context (record )
2933 payload_size = calculate_sqs_payload_size (record )
3034
31- ctx = DsmPathwayCodec .decode (contextjson , processor () )
35+ ctx = DsmPathwayCodec .decode (contextjson , processor )
3236 ctx .set_checkpoint (
3337 ["direction:in" , f"topic:{ queue_arn } " , "type:sqs" ],
3438 payload_size = payload_size ,
You can’t perform that action at this time.
0 commit comments