feat: Support DD_TRACE_OTEL_ENABLED env var (DataDog#361)

astuyve · web-flow · commit 1ddb8e83a21a · 2023-08-22T07:58:28.000-04:00
* feat: Support DD_TRACE_OTEL_ENABLED env var

* fix: lint

* fix: black

* fix: Seems that only the ENV var must be set for ddtrace to care

* fix: remove space
diff --git a/datadog_lambda/tracing.py b/datadog_lambda/tracing.py
@@ -41,6 +41,16 @@
     EventSubtypes,
 )
 
+dd_trace_otel_enabled = (
+    os.environ.get("DD_TRACE_OTEL_ENABLED", "false").lower() == "true"
+)
+if dd_trace_otel_enabled:
+    from opentelemetry.trace import set_tracer_provider
+    from ddtrace.opentelemetry import TracerProvider
+
+    set_tracer_provider(TracerProvider())
+
+
 logger = logging.getLogger(__name__)
 
 dd_trace_context = {}
