datadog_lambda 8.122.0__tar.gz → 8.123.0__tar.gz

{datadog_lambda-8.122.0 → datadog_lambda-8.123.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: datadog_lambda
-Version: 8.122.0
+Version: 8.123.0
 Summary: The Datadog AWS Lambda Library
 Home-page: https://github.com/DataDog/datadog-lambda-python
 License: Apache-2.0

{datadog_lambda-8.122.0 → datadog_lambda-8.123.0}/datadog_lambda/cold_start.py

@@ -63,6 +63,19 @@ def get_proactive_init_tag():
     )
 
 
+def is_managed_instances_mode():
+    """
+    Checks if the Lambda function is running in managed instances mode.
+    In managed instances mode, we should not create cold start tracing spans
+    as the gap between the sandbox initialization and the first
+    invocation might not be be a great experience.
+
+    Returns:
+        bool: True if running in managed instances mode, False otherwise
+    """
+    return config.aws_lambda_initialization_type == "lambda-managed-instances"
+
+
 class ImportNode(object):
     def __init__(self, module_name, full_file_path, start_time_ns, end_time_ns=None):
         self.module_name = module_name
@@ -145,6 +158,10 @@ def wrap_find_spec(original_find_spec):
 
 
 def initialize_cold_start_tracing():
+    # Skip cold start tracing initialization in managed instances mode
+    if is_managed_instances_mode():
+        return
+
     if is_new_sandbox() and config.cold_start_tracing:
         from sys import meta_path
 

{datadog_lambda-8.122.0 → datadog_lambda-8.123.0}/datadog_lambda/config.py

@@ -98,6 +98,7 @@ class Config:
     integration_test = _get_env("DD_INTEGRATION_TEST", "false", as_bool)
 
     aws_lambda_function_name = _get_env("AWS_LAMBDA_FUNCTION_NAME")
+    aws_lambda_initialization_type = _get_env("AWS_LAMBDA_INITIALIZATION_TYPE")
 
     @property
     def function_name(self):

datadog_lambda-8.123.0/datadog_lambda/durable.py

@@ -0,0 +1,49 @@
+# Unless explicitly stated otherwise all files in this repository are licensed
+# under the Apache License Version 2.0.
+# This product includes software developed at Datadog (https://www.datadoghq.com/).
+# Copyright 2019 Datadog, Inc.
+import logging
+import re
+
+logger = logging.getLogger(__name__)
+
+
+def _parse_durable_execution_arn(arn):
+    """
+    Parses a DurableExecutionArn to extract execution name and ID.
+    ARN format:
+        arn:aws:lambda:{region}:{account}:function:{func}:{version}/durable-execution/{name}/{id}
+    Returns (execution_name, execution_id) or None if parsing fails.
+    """
+    match = re.search(r"/durable-execution/([^/]+)/([^/]+)$", arn)
+    if not match:
+        return None
+    execution_name, execution_id = match.group(1), match.group(2)
+    if not execution_name or not execution_id:
+        return None
+    return execution_name, execution_id
+
+
+def extract_durable_function_tags(event):
+    """
+    Extracts durable function tags from the Lambda event payload.
+    Returns a dict with durable function tags, or an empty dict if the event
+    is not a durable function invocation.
+    """
+    if not isinstance(event, dict):
+        return {}
+
+    durable_execution_arn = event.get("DurableExecutionArn")
+    if not isinstance(durable_execution_arn, str):
+        return {}
+
+    parsed = _parse_durable_execution_arn(durable_execution_arn)
+    if not parsed:
+        logger.error("Failed to parse DurableExecutionArn: %s", durable_execution_arn)
+        return {}
+
+    execution_name, execution_id = parsed
+    return {
+        "durable_function_execution_name": execution_name,
+        "durable_function_execution_id": execution_id,
+    }

{datadog_lambda-8.122.0 → datadog_lambda-8.123.0}/datadog_lambda/tracing.py

@@ -1449,6 +1449,7 @@ def create_function_execution_span(
     trace_context_source,
     merge_xray_traces,
     trigger_tags,
+    durable_function_tags=None,
     parent_span=None,
     span_pointers=None,
 ):
@@ -1477,6 +1478,8 @@ def create_function_execution_span(
     if trace_context_source == TraceContextSource.XRAY and merge_xray_traces:
         tags["_dd.parent_source"] = trace_context_source
     tags.update(trigger_tags)
+    if durable_function_tags:
+        tags.update(durable_function_tags)
     tracer.set_tags(_dd_origin)
     # Determine service name based on config and env var
     if config.service:

datadog_lambda-8.123.0/datadog_lambda/version.py

@@ -0,0 +1 @@
+__version__ = "8.123.0"

{datadog_lambda-8.122.0 → datadog_lambda-8.123.0}/datadog_lambda/wrapper.py

@@ -16,6 +16,7 @@ from datadog_lambda.cold_start import (
     is_cold_start,
     is_proactive_init,
     is_new_sandbox,
+    is_managed_instances_mode,
     ColdStartTracer,
 )
 from datadog_lambda.config import config
@@ -41,6 +42,7 @@ from datadog_lambda.tracing import (
     tracer,
     propagator,
 )
+from datadog_lambda.durable import extract_durable_function_tags
 from datadog_lambda.trigger import (
     extract_trigger_tags,
     extract_http_status_code_tag,
@@ -242,6 +244,7 @@ class _LambdaDecorator(object):
                 submit_invocations_metric(context)
 
             self.trigger_tags = extract_trigger_tags(event, context)
+            self.durable_function_tags = extract_durable_function_tags(event)
             # Extract Datadog trace context and source from incoming requests
             dd_context, trace_context_source, event_source = extract_dd_trace_context(
                 event,
@@ -279,6 +282,7 @@ class _LambdaDecorator(object):
                     trace_context_source=trace_context_source,
                     merge_xray_traces=config.merge_xray_traces,
                     trigger_tags=self.trigger_tags,
+                    durable_function_tags=self.durable_function_tags,
                     parent_span=self.inferred_span,
                     span_pointers=calculate_span_pointers(event_source, event),
                 )
@@ -288,7 +292,7 @@ class _LambdaDecorator(object):
             else:
                 set_correlation_ids()
             if config.profiling_enabled and profiler and is_new_sandbox():
-                self.prof.start(stop_on_exit=False, profile_children=True)
+                self.prof.start()
             logger.debug("datadog_lambda_wrapper _before() done")
         except Exception as e:
             logger.error(format_err_with_traceback(e))
@@ -338,7 +342,13 @@ class _LambdaDecorator(object):
                 create_dd_dummy_metadata_subsegment(
                     self.trigger_tags, XraySubsegment.LAMBDA_FUNCTION_TAGS_KEY
                 )
-            should_trace_cold_start = config.cold_start_tracing and is_new_sandbox()
+            # Skip creating cold start spans in managed instances mode
+            # In managed instances, the tracer library handles cold start independently
+            should_trace_cold_start = (
+                config.cold_start_tracing
+                and is_new_sandbox()
+                and not is_managed_instances_mode()
+            )
             if should_trace_cold_start:
                 trace_ctx = tracer.current_trace_context()
 

{datadog_lambda-8.122.0 → datadog_lambda-8.123.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "datadog_lambda"
-version = "8.122.0"
+version = "8.123.0"
 description = "The Datadog AWS Lambda Library"
 authors = ["Datadog, Inc. <dev@datadoghq.com>"]
 license = "Apache-2.0"
@@ -54,4 +54,4 @@ requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"
 
 [tool.pytest.ini_options]
-addopts = "--benchmark-disable --benchmark-autosave"
+addopts = "--benchmark-disable --benchmark-autosave"

datadog_lambda-8.122.0/datadog_lambda/version.py

@@ -1 +0,0 @@
-__version__ = "8.122.0"