jararaca 0.3.28__py3-none-any.whl → 0.3.29__py3-none-any.whl

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/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/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/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):
 

{jararaca-0.3.28.dist-info → jararaca-0.3.29.dist-info}/METADATA

@@ -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.dist-info → jararaca-0.3.29.dist-info}/RECORD

@@ -1,7 +1,7 @@
 LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
 README.md,sha256=YmCngjU8llW0l7L3tuXkkfr8qH7V9aBMgfp2jEzeiKg,3517
-pyproject.toml,sha256=CEt_9xw3r5w6k6x7hD62Gb3uU3hHAyMaAmldYFK7jjM,2739
-jararaca/__init__.py,sha256=gp2ib_dYm5EP97FY3E8Av4CDCbgX6vuQhGNvNBpu0IQ,23866
+pyproject.toml,sha256=onnmUnr1FXl1m5F5BLHvwPeJjoQrCzetXPaqxh7F2Ao,2739
+jararaca/__init__.py,sha256=LH6u63yz8UTABwH4EMSrOprxw6Szn0nQWJB5A9U4VjY,24259
 jararaca/__main__.py,sha256=-O3vsB5lHdqNFjUtoELDF81IYFtR-DSiiFMzRaiSsv4,67
 jararaca/broker_backend/__init__.py,sha256=GzEIuHR1xzgCJD4FE3harNjoaYzxHMHoEL0_clUaC-k,3528
 jararaca/broker_backend/mapper.py,sha256=vTsi7sWpNvlga1PWPFg0rCJ5joJ0cdzykkIc2Tuvenc,696
@@ -27,12 +27,12 @@ jararaca/messagebus/publisher.py,sha256=JTkxdKbvxvDWT8nK8PVEyyX061vYYbKQMxRHXrZt
 jararaca/messagebus/worker.py,sha256=DiKDUhcU4rEjjA_3KqCmnN3X5yyokZq_-SbLryjS_PM,69237
 jararaca/microservice.py,sha256=DW4RVeqgrx4J-dAg17sbzTn_sLXuvV1UOQFde2fpqds,11471
 jararaca/observability/constants.py,sha256=quhqXBjDWVrr8Vr3tglgYo7zsQyz_y-z_dtqfUzYaa8,63
-jararaca/observability/decorators.py,sha256=W6RpGBUPJJX-FiYhp3HaCC-fh7-FZwDG4gBVZiE9dKU,6229
+jararaca/observability/decorators.py,sha256=1mxIhBduikOYayq-cW9Pq1hbTzztUkjx0uzKJSSMEU4,6511
 jararaca/observability/fastapi_exception_handler.py,sha256=jSLhuR5Og3vRnIoZucRA5jPnvR6DEsM1ax7tQuEo5rc,1264
-jararaca/observability/hooks.py,sha256=ATGFPQxhkry1NdW18ywJ9onstlnW8Yx2FD487eY9M8o,2156
+jararaca/observability/hooks.py,sha256=RRv5q57bRurOUR4AIZpl1K07j9tSduj6LrBNc4xwfRs,2778
 jararaca/observability/interceptor.py,sha256=U4ZLM0f8j6Q7gMUKKnA85bnvD-Qa0ii79Qa_X8KsXAQ,1498
 jararaca/observability/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-jararaca/observability/providers/otel.py,sha256=vsaiIDWuSbpZ4UyzRMSH64o4BxzsZzO7P39mJHUD6rU,11442
+jararaca/observability/providers/otel.py,sha256=08G1-dHzckKo9DQiteJl8KUukH_5tL31b2YrTDEnuDA,12262
 jararaca/persistence/base.py,sha256=xnGUbsLNz3gO-9iJt-Sn5NY13Yc9-misP8wLwQuGGoM,1024
 jararaca/persistence/exports.py,sha256=Ghx4yoFaB4QVTb9WxrFYgmcSATXMNvrOvT8ybPNKXCA,62
 jararaca/persistence/interceptors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -78,8 +78,8 @@ jararaca/tools/typescript/interface_parser.py,sha256=VBx-TYQPjAiHLiYgzGdRl_p4xDb
 jararaca/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 jararaca/utils/rabbitmq_utils.py,sha256=ytdAFUyv-OBkaVnxezuJaJoLrmN7giZgtKeet_IsMBs,10918
 jararaca/utils/retry.py,sha256=DzPX_fXUvTqej6BQ8Mt2dvLo9nNlTBm7Kx2pFZ26P2Q,4668
-jararaca-0.3.28.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-jararaca-0.3.28.dist-info/METADATA,sha256=5yQgnOm6ZduUsne5HQP-uq8_wgAZX-bOA3QeX4oTbgI,5149
-jararaca-0.3.28.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-jararaca-0.3.28.dist-info/entry_points.txt,sha256=WIh3aIvz8LwUJZIDfs4EeH3VoFyCGEk7cWJurW38q0I,45
-jararaca-0.3.28.dist-info/RECORD,,
+jararaca-0.3.29.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+jararaca-0.3.29.dist-info/METADATA,sha256=SF-nYBMhJ-gK7OxaQ3Y75sVExlh7vCcyldV5hWT1NBg,5149
+jararaca-0.3.29.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+jararaca-0.3.29.dist-info/entry_points.txt,sha256=WIh3aIvz8LwUJZIDfs4EeH3VoFyCGEk7cWJurW38q0I,45
+jararaca-0.3.29.dist-info/RECORD,,

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"