Part of otel integration wip

Author: sl0thentr0py

sentry_sdk/integrations/opentelemetry/integration.py

@@ -3,7 +3,7 @@
 are experimental and not suitable for production use. They may be changed or
 removed at any time without prior notice.
 """
-
+import sentry_sdk
 from sentry_sdk.integrations import DidNotEnable, Integration
 from sentry_sdk.integrations.opentelemetry.propagator import SentryPropagator
 from sentry_sdk.integrations.opentelemetry.span_processor import SentrySpanProcessor
@@ -13,9 +13,16 @@
     from opentelemetry import trace
     from opentelemetry.propagate import set_global_textmap
     from opentelemetry.sdk.trace import TracerProvider
+    from opentelemetry.sdk.trace.export import BatchSpanProcessor
 except ImportError:
     raise DidNotEnable("opentelemetry not installed")
 
+try:
+    from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+except ImportError:
+    OTLPSpanExporter = None
+
+
 try:
     from opentelemetry.instrumentation.django import DjangoInstrumentor  # type: ignore[import-not-found]
 except ImportError:
@@ -30,6 +37,20 @@
 class OpenTelemetryIntegration(Integration):
     identifier = "opentelemetry"
 
+    def __init__(self, enable_span_processor=True, enable_otlp_exporter=False, enable_propagator=True):
+        # type: (bool, bool, bool) -> None
+        self.enable_span_processor = enable_span_processor
+        self.enable_otlp_exporter = enable_otlp_exporter
+        self.enable_propagator = enable_propagator
+
+        if self.enable_otlp_exporter and OTLPSpanExporter is None:
+            logger.warning("[Otel] OTLPSpanExporter not installed.")
+            self.enable_otlp_exporter = False
+
+        if self.enable_span_processor and self.enable_otlp_exporter:
+            logger.warning("[Otel] Disabling span processor because otlp exporter is set.")
+            self.enable_span_processor = False
+
     @staticmethod
     def setup_once():
         # type: () -> None
@@ -46,10 +67,26 @@ def setup_once():
 
 def _setup_sentry_tracing():
     # type: () -> None
+    integration = sentry_sdk.get_client().get_integration(OpenTelemetryIntegration)
+    if integration is None:
+        return
+
+    # TODO provider oblivious
     provider = TracerProvider()
-    provider.add_span_processor(SentrySpanProcessor())
+
+    if integration.enable_otlp_exporter:
+        otlp_exporter = OTLPSpanExporter(
+            # endpoint="___OTLP_TRACES_URL___",
+            # headers={"x-sentry-auth": "sentry sentry_key=___PUBLIC_KEY___"},
+        )
+        otlp_span_processor = BatchSpanProcessor(otlp_exporter)
+        provider.add_span_processor(otlp_span_processor)
+    elif integration.enable_span_processor:
+        provider.add_span_processor(SentrySpanProcessor())
     trace.set_tracer_provider(provider)
-    set_global_textmap(SentryPropagator())
+
+    if integration.enable_propagator:
+        set_global_textmap(SentryPropagator())
 
 
 def _setup_instrumentors():

setup.py

@@ -70,7 +70,7 @@ def get_file_text(file_name):
         "loguru": ["loguru>=0.5"],
         "openai": ["openai>=1.0.0", "tiktoken>=0.3.0"],
         "openfeature": ["openfeature-sdk>=0.7.1"],
-        "opentelemetry": ["opentelemetry-distro>=0.35b0"],
+        "opentelemetry": ["opentelemetry-distro>=0.35b0", "opentelemetry-exporter-otlp-proto-http>=1.0.0"],
         "opentelemetry-experimental": ["opentelemetry-distro"],
         "pure-eval": ["pure_eval", "executing", "asttokens"],
         "pydantic_ai": ["pydantic-ai>=1.0.0"],

tox.ini

@@ -335,6 +335,7 @@ deps =
 
     # OpenTelemetry (OTel)
     opentelemetry: opentelemetry-distro
+    opentelemetry: opentelemetry-exporter-otlp-proto-http
 
     # OpenTelemetry Experimental (POTel)
     potel: -e .[opentelemetry-experimental]