make private

Author: michael-zhao459

datadog_lambda/tracing.py

@@ -70,7 +70,7 @@
 
 def _dsm_set_context_sqs_or_sns_event(event, event_type):
     for record in event.get("Records", [])[1:]:
-        if arn := get_source_arn_sqs_or_sns(record, event_type):
+        if arn := _get_source_arn_sqs_or_sns(record, event_type):
             context = _extract_context_from_sqs_or_sns_record(record)
             _dsm_set_checkpoint(context, event_type, arn)
 
@@ -82,7 +82,7 @@ def _dsm_set_context_kinesis_event(event):
             _dsm_set_checkpoint(context, "kinesis", arn)
 
 
-def get_source_arn_sqs_or_sns(record, event_type):
+def _get_source_arn_sqs_or_sns(record, event_type):
     return (
         record.get("Sns", {}).get("TopicArn")
         if event_type == "sns"
@@ -271,7 +271,7 @@ def extract_context_from_sqs_or_sns_event_or_context(
 
     try:
         first_record = event.get("Records")[0]
-        source_arn = get_source_arn_sqs_or_sns(first_record, event_type)
+        source_arn = _get_source_arn_sqs_or_sns(first_record, event_type)
         dd_data = _extract_context_from_sqs_or_sns_record(first_record)
         if dd_data:
             if is_step_function_event(dd_data):