jararaca 0.3.28__tar.gz → 0.3.29__tar.gz

{jararaca-0.3.28 → jararaca-0.3.29}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: jararaca
-Version: 0.3.28
+Version: 0.3.29
 Summary: A simple and fast API framework for Python
 Home-page: https://github.com/LuscasLeo/jararaca
 Author: Lucas S

{jararaca-0.3.28 → jararaca-0.3.29}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "jararaca"
-version = "0.3.28"
+version = "0.3.29"
 description = "A simple and fast API framework for Python"
 authors = ["Lucas S <me@luscasleo.dev>"]
 readme = "README.md"

{jararaca-0.3.28 → jararaca-0.3.29}/src/jararaca/__init__.py

@@ -23,6 +23,9 @@ if TYPE_CHECKING:
     )
     from jararaca.observability.hooks import (
         add_event,
+        add_span_link,
+        get_current_span,
+        get_current_span_context,
         get_tracing_provider,
         record_exception,
         set_span_attribute,
@@ -239,6 +242,9 @@ if TYPE_CHECKING:
         "TracedClass",
         "traced_class",
         "start_span",
+        "add_span_link",
+        "get_current_span",
+        "get_current_span_context",
         "spawn_trace",
         "set_span_attribute",
         "add_event",
@@ -427,6 +433,9 @@ _dynamic_imports: "dict[str, tuple[str, str, str | None]]" = {
     ),
     "set_span_attribute": (__SPEC_PARENT__, "observability.hooks", None),
     "start_span": (__SPEC_PARENT__, "observability.hooks", None),
+    "add_span_link": (__SPEC_PARENT__, "observability.hooks", None),
+    "get_current_span": (__SPEC_PARENT__, "observability.hooks", None),
+    "get_current_span_context": (__SPEC_PARENT__, "observability.hooks", None),
     "add_event": (__SPEC_PARENT__, "observability.hooks", None),
     "set_span_status": (__SPEC_PARENT__, "observability.hooks", None),
     "record_exception": (__SPEC_PARENT__, "observability.hooks", None),

{jararaca-0.3.28 → jararaca-0.3.29}/src/jararaca/observability/decorators.py

@@ -35,6 +35,12 @@ AttributeValue = Union[
 AttributeMap = Mapping[str, AttributeValue]
 
 
+class TracingSpan(Protocol): ...
+
+
+class TracingSpanContext(Protocol): ...
+
+
 class TracingContextProvider(Protocol):
 
     def start_span_context(
@@ -63,6 +69,13 @@ class TracingContextProvider(Protocol):
     ) -> None: ...
 
     def update_span_name(self, new_name: str) -> None: ...
+
+    def add_link(self, span_context: TracingSpanContext) -> None: ...
+
+    def get_current_span(self) -> TracingSpan | None: ...
+    def get_current_span_context(self) -> TracingSpanContext | None: ...
+
+
 class TracingContextProviderFactory(Protocol):
 
     def root_setup(

{jararaca-0.3.28 → jararaca-0.3.29}/src/jararaca/observability/hooks.py

@@ -6,6 +6,8 @@ from jararaca.observability.decorators import (
     AttributeMap,
     AttributeValue,
     TracingContextProvider,
+    TracingSpan,
+    TracingSpanContext,
     get_tracing_ctx_provider,
 )
 
@@ -84,3 +86,23 @@ def set_span_attribute(
 
 def get_tracing_provider() -> TracingContextProvider | None:
     return get_tracing_ctx_provider()
+
+
+def get_current_span_context() -> TracingSpanContext | None:
+
+    if trace_context_provider := get_tracing_ctx_provider():
+        return trace_context_provider.get_current_span_context()
+    return None
+
+
+def get_current_span() -> TracingSpan | None:
+
+    if trace_context_provider := get_tracing_ctx_provider():
+        return trace_context_provider.get_current_span()
+    return None
+
+
+def add_span_link(span_context: TracingSpanContext) -> None:
+
+    if trace_context_provider := get_tracing_ctx_provider():
+        trace_context_provider.add_link(span_context=span_context)

{jararaca-0.3.28 → jararaca-0.3.29}/src/jararaca/observability/providers/otel.py

@@ -40,12 +40,26 @@ from jararaca.observability.decorators import (
     AttributeValue,
     TracingContextProvider,
     TracingContextProviderFactory,
+    TracingSpan,
+    TracingSpanContext,
 )
 from jararaca.observability.interceptor import ObservabilityProvider
 
 tracer: trace.Tracer = trace.get_tracer(__name__)
 
 
+class OtelTracingSpan(TracingSpan):
+
+    def __init__(self, span: trace.Span) -> None:
+        self.span = span
+
+
+class OtelTracingSpanContext(TracingSpanContext):
+
+    def __init__(self, span_context: trace.SpanContext) -> None:
+        self.span_context = span_context
+
+
 class OtelTracingContextProvider(TracingContextProvider):
 
     def __init__(self, app_context: AppTransactionContext) -> None:
@@ -90,8 +104,21 @@ class OtelTracingContextProvider(TracingContextProvider):
 
     def update_span_name(self, new_name: str) -> None:
         span = trace.get_current_span()
+
         span.update_name(new_name)
 
+    def add_link(self, span_context: TracingSpanContext) -> None:
+        if not isinstance(span_context, OtelTracingSpanContext):
+            return
+        span = trace.get_current_span()
+        span.add_link(span_context.span_context)
+
+    def get_current_span(self) -> TracingSpan | None:
+        return OtelTracingSpan(trace.get_current_span())
+
+    def get_current_span_context(self) -> TracingSpanContext | None:
+        return OtelTracingSpanContext(trace.get_current_span().get_span_context())
+
 
 class OtelTracingContextProviderFactory(TracingContextProviderFactory):