ref: Move integration out of opentelemetry/integrations/

Author: sentrivana

sentry_sdk/_init_implementation.py

@@ -3,6 +3,10 @@
 import sentry_sdk
 from sentry_sdk.consts import ClientConstructor
 from sentry_sdk.opentelemetry.scope import setup_scope_context_management
+from sentry_sdk.opentelemetry.integration import (
+    patch_readable_span,
+    setup_sentry_tracing,
+)
 
 if TYPE_CHECKING:
     from typing import Any, Optional
@@ -22,6 +26,8 @@ def _init(*args, **kwargs):
 
     This takes the same arguments as the client constructor.
     """
+    patch_readable_span()
+    setup_sentry_tracing()
     setup_scope_context_management()
     client = sentry_sdk.Client(*args, **kwargs)
     sentry_sdk.get_global_scope().set_client(client)

sentry_sdk/integrations/opentelemetry/integration.py

@@ -0,0 +1,33 @@
+from opentelemetry import trace
+from opentelemetry.propagate import set_global_textmap
+from opentelemetry.sdk.trace import TracerProvider, Span, ReadableSpan
+
+from sentry_sdk.opentelemetry.propagator import SentryPropagator
+from sentry_sdk.opentelemetry.sampler import SentrySampler
+from sentry_sdk.opentelemetry.span_processor import SentrySpanProcessor
+
+
+def patch_readable_span():
+    # type: () -> None
+    """
+    We need to pass through sentry specific metadata/objects from Span to ReadableSpan
+    to work with them consistently in the SpanProcessor.
+    """
+    old_readable_span = Span._readable_span
+
+    def sentry_patched_readable_span(self):
+        # type: (Span) -> ReadableSpan
+        readable_span = old_readable_span(self)
+        readable_span._sentry_meta = getattr(self, "_sentry_meta", {})  # type: ignore[attr-defined]
+        return readable_span
+
+    Span._readable_span = sentry_patched_readable_span  # type: ignore[method-assign]
+
+
+def setup_sentry_tracing():
+    # type: () -> None
+    provider = TracerProvider(sampler=SentrySampler())
+    provider.add_span_processor(SentrySpanProcessor())
+    trace.set_tracer_provider(provider)
+
+    set_global_textmap(SentryPropagator())