encore-dev 2.0.0a7__py3-none-any.whl

encore/tracing/_tracer.py

@@ -0,0 +1,93 @@
+"""An ``opentelemetry.trace.Tracer`` backed by the runtime's pipeline."""
+
+from __future__ import annotations
+
+from typing import Iterator, Optional, Sequence
+
+from opentelemetry import context as context_api
+from opentelemetry import trace
+from opentelemetry.trace import Link, SpanKind
+from opentelemetry.util import types
+from opentelemetry.util._decorator import _agnosticcontextmanager
+
+from ._context import span_ctx_bytes
+from ._encode import Scope, encode_start
+from ._span import Backend, Span
+
+
+class Tracer(trace.Tracer):
+    """A tracer whose spans share an instrumentation scope.
+
+    The runtime assigns span identifiers and timestamps.
+    """
+
+    def __init__(self, backend: Backend, scope: Scope) -> None:
+        self._backend = backend
+        self._scope = scope
+
+    def start_span(
+        self,
+        name: str,
+        context: Optional[context_api.Context] = None,
+        kind: SpanKind = SpanKind.INTERNAL,
+        attributes: types.Attributes = None,
+        links: Optional[Sequence[Link]] = None,
+        start_time: Optional[int] = None,
+        record_exception: bool = True,
+        set_status_on_exception: bool = True,
+    ) -> trace.Span:
+        parent = trace.get_current_span(context)
+        parent_ctx = parent.get_span_context()
+        parent_bytes = span_ctx_bytes(parent) if parent_ctx.is_valid else None
+        ctx, tracestate, start_unix, start_mono = self._backend.start_span(
+            name,
+            kind.value,
+            parent_bytes,
+            parent_ctx.trace_state.to_header() if parent_bytes else None,
+            start_time,
+            encode_start(self._scope, attributes, links),
+        )
+        return Span(
+            backend=self._backend,
+            scope=self._scope,
+            ctx_bytes=ctx,
+            tracestate=tracestate,
+            name=name,
+            kind=kind,
+            parent_span_id=parent_bytes[16:24] if parent_bytes else None,
+            start_unix_nanos=start_unix,
+            start_mono_nanos=start_mono,
+            attributes=attributes,
+            links=links,
+        )
+
+    @_agnosticcontextmanager
+    def start_as_current_span(
+        self,
+        name: str,
+        context: Optional[context_api.Context] = None,
+        kind: SpanKind = SpanKind.INTERNAL,
+        attributes: types.Attributes = None,
+        links: Optional[Sequence[Link]] = None,
+        start_time: Optional[int] = None,
+        record_exception: bool = True,
+        set_status_on_exception: bool = True,
+        end_on_exit: bool = True,
+    ) -> Iterator[trace.Span]:
+        span = self.start_span(
+            name,
+            context,
+            kind,
+            attributes,
+            links,
+            start_time,
+            record_exception,
+            set_status_on_exception,
+        )
+        with trace.use_span(
+            span,
+            end_on_exit=end_on_exit,
+            record_exception=record_exception,
+            set_status_on_exception=set_status_on_exception,
+        ) as s:
+            yield s