renamed function

Author: michael-zhao459

datadog_lambda/dsm.py

@@ -1,8 +1,8 @@
 import logging
 from datadog_lambda.trigger import EventTypes, _EventSource
 from datadog_lambda.tracing import (
-    extract_datadog_context_from_kinesis_event,
-    extract_datadog_context_from_sqs_or_sns_event,
+    extract_context_from_kinesis_record,
+    extract_context_from_sqs_or_sns_record,
 )
 
 logger = logging.getLogger(__name__)
@@ -35,11 +35,9 @@ def set_dsm_context(event, event_source: _EventSource):
             context_json = None
             try:
                 context_json = (
-                    extract_datadog_context_from_kinesis_event(
-                        record.get("kinesis", {})
-                    )
+                    extract_context_from_kinesis_record(record.get("kinesis", {}))
                     if event_source.equals(EventTypes.KINESIS)
-                    else extract_datadog_context_from_sqs_or_sns_event(record)
+                    else extract_context_from_sqs_or_sns_record(record)
                 )
             except Exception as e:
                 logger.debug(

datadog_lambda/tracing.py

@@ -227,7 +227,7 @@ def extract_context_from_sqs_or_sns_event_or_context(event, lambda_context):
 
     try:
         first_record = event.get("Records")[0]
-        dd_data = extract_datadog_context_from_sqs_or_sns_event(first_record)
+        dd_data = extract_context_from_sqs_or_sns_record(first_record)
         if dd_data:
             if is_step_function_event(dd_data):
                 try:
@@ -266,7 +266,7 @@ def extract_context_from_sqs_or_sns_event_or_context(event, lambda_context):
         return extract_context_from_lambda_context(lambda_context)
 
 
-def extract_datadog_context_from_sqs_or_sns_event(record):
+def extract_context_from_sqs_or_sns_record(record):
     if "body" in record:
         body_str = record.get("body")
         try:
@@ -368,15 +368,15 @@ def extract_context_from_kinesis_event(event, lambda_context):
         kinesis = record.get("kinesis")
         if not kinesis:
             return extract_context_from_lambda_context(lambda_context)
-        dd_ctx = extract_datadog_context_from_kinesis_event(kinesis)
+        dd_ctx = extract_context_from_kinesis_record(kinesis)
         if dd_ctx:
             return propagator.extract(dd_ctx)
     except Exception as e:
         logger.debug("The trace extractor returned with error %s", e)
     return extract_context_from_lambda_context(lambda_context)
 
 
-def extract_datadog_context_from_kinesis_event(record_kinesis_field):
+def extract_context_from_kinesis_record(record_kinesis_field):
     data = record_kinesis_field.get("data")
     if data:
         import base64