Skip to content

Commit 779e6a5

Browse files
fixes
1 parent 9dbf2de commit 779e6a5

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

datadog_lambda/dsm.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff 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,

0 commit comments

Comments
 (0)