Merge pull request #45 from PADAS/tracing-enable-flag

Tracing enable flag

Author: chrisdoehring

cdip_connector/core/cdip_settings.py

@@ -56,6 +56,7 @@
 # How many items should be posted to Sensors API in each request.
 INTEGRATION_LOAD_BATCH_SIZE = env.int("INTEGRATION_LOAD_BATCH_SIZE", 25)
 
+TRACING_ENABLED = env.bool("TRACING_ENABLED", True)
 
 # Coerce task count and index into common variables (using CronJob variables). 
 # This allows them to be provided by a Kubernetes CronJob or by a Cloud Run Job.

cdip_connector/core/tracing.py

@@ -11,26 +11,27 @@
 from opentelemetry.instrumentation.requests import RequestsInstrumentor
 from opentelemetry.instrumentation.httpx import HTTPXClientInstrumentor
 from opentelemetry.instrumentation.confluent_kafka import ConfluentKafkaInstrumentor
-
+from cdip_connector.core import cdip_settings
 
 def configure_tracer(name: str, version: str = ""):
-    resource = Resource.create(
-        {
-            "service.name": name,
-            "service.version": version,
-        }
-    )
-    tracer_provider = TracerProvider(resource=resource)
-    cloud_trace_exporter = CloudTraceSpanExporter()
-    tracer_provider.add_span_processor(
-        # BatchSpanProcessor buffers spans and sends them in batches in a
-        # background thread. The default parameters are sensible, but can be
-        # tweaked to optimize your performance
-        BatchSpanProcessor(cloud_trace_exporter)
-    )
-    trace.set_tracer_provider(tracer_provider)
-    # Using the X-Cloud-Trace-Context header
-    set_global_textmap(CloudTraceFormatPropagator())
+    if cdip_settings.TRACING_ENABLED:
+        resource = Resource.create(
+            {
+                "service.name": name,
+                "service.version": version,
+            }
+        )
+        tracer_provider = TracerProvider(resource=resource)
+        cloud_trace_exporter = CloudTraceSpanExporter()
+        tracer_provider.add_span_processor(
+            # BatchSpanProcessor buffers spans and sends them in batches in a
+            # background thread. The default parameters are sensible, but can be
+            # tweaked to optimize your performance
+            BatchSpanProcessor(cloud_trace_exporter)
+        )
+        trace.set_tracer_provider(tracer_provider)
+        # Using the X-Cloud-Trace-Context header
+        set_global_textmap(CloudTraceFormatPropagator())
     return trace.get_tracer(name, version)