Use config.cold_start_tracing.

Author: purple4reina

datadog_lambda/config.py

@@ -49,7 +49,7 @@ class Config:
     service = _get_env("DD_SERVICE")
     env = _get_env("DD_ENV")
 
-    cold_start_tracing = _get_env("DD_COLD_START_TRACING", "true", as_bool)
+    cold_start_tracing = _get_env("DD_COLD_START_TRACING", "true", as_bool, depends_on_tracing=True)
     min_cold_start_trace_duration = _get_env("DD_MIN_COLD_START_DURATION", 3, int)
     cold_start_trace_skip_lib = _get_env(
             "DD_COLD_START_TRACE_SKIP_LIB", "ddtrace.internal.compat,ddtrace.filters",

datadog_lambda/wrapper.py

@@ -145,8 +145,10 @@ def __init__(self, func):
                 except Exception:
                     logger.debug(f"Malformatted for env {DD_COLD_START_TRACE_SKIP_LIB}")
             self.response = None
+
             if config.profiling_enabled:
                 self.prof = profiler.Profiler(env=config.env, service=config.service)
+
             if config.trace_extractor:
                 extractor_parts = config.trace_extractor.rsplit(".", 1)
                 if len(extractor_parts) == 2:
@@ -296,7 +298,7 @@ def _after(self, event, context):
                 create_dd_dummy_metadata_subsegment(
                     self.trigger_tags, XraySubsegment.LAMBDA_FUNCTION_TAGS_KEY
                 )
-            should_trace_cold_start = self.cold_start_tracing and is_new_sandbox()
+            should_trace_cold_start = config.cold_start_tracing and is_new_sandbox()
             if should_trace_cold_start:
                 trace_ctx = tracer.current_trace_context()