Add profiling_enabled.

Author: purple4reina

datadog_lambda/config.py

@@ -58,6 +58,7 @@ class Config:
     merge_xray_traces = _get_env("DD_MERGE_XRAY_TRACES", "false", as_bool)
     trace_extractor = _get_env("DD_TRACE_EXTRACTOR")
     capture_payload_max_depth = _get_env("DD_CAPTURE_PAYLOAD_MAX_DEPTH", 10, int)
+    profiling_enabled = _get_env("DD_PROFILING_ENABLED", "false", as_bool)
 
     @property
     def fips_mode_enabled(self):

datadog_lambda/wrapper.py

@@ -45,8 +45,7 @@
     extract_http_status_code_tag,
 )
 
-profiling_env_var = os.environ.get("DD_PROFILING_ENABLED", "false").lower() == "true"
-if profiling_env_var:
+if config.profiling_enabled:
     from ddtrace.profiling import profiler
 
 llmobs_env_var = os.environ.get("DD_LLMOBS_ENABLED", "false").lower() in ("true", "1")
@@ -181,7 +180,7 @@ def __init__(self, func):
                 except Exception:
                     logger.debug(f"Malformatted for env {DD_COLD_START_TRACE_SKIP_LIB}")
             self.response = None
-            if profiling_env_var:
+            if config.profiling_enabled:
                 self.prof = profiler.Profiler(env=env_env_var, service=config.service)
             if config.trace_extractor:
                 extractor_parts = config.trace_extractor.rsplit(".", 1)
@@ -311,7 +310,7 @@ def _before(self, event, context):
                 )
             else:
                 set_correlation_ids()
-            if profiling_env_var and is_new_sandbox():
+            if config.profiling_enabled and is_new_sandbox():
                 self.prof.start(stop_on_exit=False, profile_children=True)
             logger.debug("datadog_lambda_wrapper _before() done")
         except Exception as e:

tests/test_config.py

@@ -137,6 +137,15 @@ def set_env(key, value):
     ("DD_CAPTURE_PAYLOAD_MAX_DEPTH", "capture_payload_max_depth", "2.5", 10),
     ("DD_CAPTURE_PAYLOAD_MAX_DEPTH", "capture_payload_max_depth", "-1", -1),
     ("DD_CAPTURE_PAYLOAD_MAX_DEPTH", "capture_payload_max_depth", "purple", 10),
+    ("DD_PROFILING_ENABLED", "profiling_enabled", None, False),
+    ("DD_PROFILING_ENABLED", "profiling_enabled", "", False),
+    ("DD_PROFILING_ENABLED", "profiling_enabled", "true", True),
+    ("DD_PROFILING_ENABLED", "profiling_enabled", "TRUE", True),
+    ("DD_PROFILING_ENABLED", "profiling_enabled", "false", False),
+    ("DD_PROFILING_ENABLED", "profiling_enabled", "FALSE", False),
+    ("DD_PROFILING_ENABLED", "profiling_enabled", "1", True),  # CHANGED
+    ("DD_PROFILING_ENABLED", "profiling_enabled", "0", False),
+    ("DD_PROFILING_ENABLED", "profiling_enabled", "purple", False),
 )