warpzone-sdk 15.0.0.dev4__py3-none-any.whl → 15.0.0.dev5__py3-none-any.whl

warpzone/function/monitor.py

@@ -28,7 +28,8 @@ SUBJECT_IDENTIFIER = "<Subject>"
 @contextmanager
 def run_in_trace_context(context: func.Context):
     trace_context = context.trace_context
-    with traces.set_trace_context(trace_context.trace_parent):
+    span_name = context.function_name or "azure_function"
+    with traces.set_trace_context(trace_context.trace_parent, span_name=span_name):
         yield
 
 

warpzone/monitor/traces.py

@@ -63,36 +63,27 @@ class AzureSDKTraceFilter(SpanProcessor):
 
 
 @contextmanager
-def set_trace_context(trace_parent: str):
-    """Context manager for setting the trace context.
+def set_trace_context(trace_parent: str, span_name: str = "function_execution"):
+    """Context manager for setting the trace context with a recording span.
 
-    Attaches the propagated trace context with a NonRecordingSpan so that
-    logs emitted within this context are properly correlated with the trace
-    created by Azure Functions.
-
-    The NonRecordingSpan is necessary because the LoggingHandler needs an
-    actual span (not just a SpanContext) to extract trace correlation IDs.
-    Using NonRecordingSpan avoids creating duplicate spans while still
-    providing the trace_id and span_id for log correlation.
+    Creates a child span of the propagated trace context. This is necessary
+    because the LoggingHandler requires a recording span to properly correlate
+    logs with the trace. A NonRecordingSpan (which is what TraceContextTextMapPropagator
+    creates) may not be correctly handled during log batching/export.
 
     Args:
         trace_parent (str): Trace parent ID from the incoming request
+        span_name (str): Name for the span. Defaults to "function_execution".
     """
     carrier = {"traceparent": trace_parent}
     ctx = TraceContextTextMapPropagator().extract(carrier=carrier)
 
-    # Get the span from the extracted context - this is a NonRecordingSpan
-    # created by the propagator with the correct trace_id and span_id
-    span = trace.get_current_span(ctx)
-
-    # Set this span as the current span in context
-    ctx = trace.set_span_in_context(span)
-
-    token = context.attach(ctx)
-    try:
-        yield
-    finally:
-        context.detach(token)
+    # Start a recording span as a child of the propagated context.
+    # This ensures logs are properly correlated with the trace.
+    tracer = trace.get_tracer(__name__)
+    with trace.use_span(trace.get_current_span(ctx), end_on_exit=False):
+        with tracer.start_as_current_span(span_name):
+            yield
 
 
 def get_tracer(name: str):

{warpzone_sdk-15.0.0.dev4.dist-info → warpzone_sdk-15.0.0.dev5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: warpzone-sdk
-Version: 15.0.0.dev4
+Version: 15.0.0.dev5
 Summary: The main objective of this package is to centralize logic used to interact with Azure Functions, Azure Service Bus and Azure Table Storage
 Author: Team Enigma
 Author-email: enigma@energinet.dk

{warpzone_sdk-15.0.0.dev4.dist-info → warpzone_sdk-15.0.0.dev5.dist-info}/RECORD

@@ -17,7 +17,7 @@ warpzone/function/__init__.py,sha256=rJOZBpWsUgjMc7YtXMJ1rLGm45KB1AhDJ_Y2ISiSISc
 warpzone/function/checks.py,sha256=B9YqThymf16ac_fVAYKilv20ru5v9nwXgHlbxYIaG98,1018
 warpzone/function/functionize.py,sha256=bSV0QvwKbD9Vo3a_8cc1rgV2rzTdMMvidinyXItBfvs,2128
 warpzone/function/integrations.py,sha256=sDt2BTx6a4mVc-33wTITP9XQVPustwj7rkX4urTyOqo,4018
-warpzone/function/monitor.py,sha256=qBvnSTSX--6_6qccTZrGPO1iS5nz2tNZlHdhE1XBrbs,2214
+warpzone/function/monitor.py,sha256=MsVMUQXn346pqY5l14NZdm4ELOTeHHQLX0_HxKTdx5Q,2293
 warpzone/function/process.py,sha256=nbUVywM8ChfUwuaqFisgaD98aNRgeZkK4g5sbtuBdRs,2339
 warpzone/function/processors/__init__.py,sha256=DhIdSWLBcIeSO8IJdxPqGIhgwwnkDN6_Xqwy93BCLeA,46
 warpzone/function/processors/dependencies.py,sha256=m17BwdKyQEvzCPxpQZpAW5l1uYRIHWmweDz3XJskmpA,1259
@@ -29,7 +29,7 @@ warpzone/healthchecks/__init__.py,sha256=9gc_Mt2szs8sDSwy0V4l3JZ6d9hX41xTpZCkDP2
 warpzone/healthchecks/model.py,sha256=mM7DnrirLbUpBPPfi82MUPP654D0eOR2_F65TmzsPD0,1187
 warpzone/monitor/__init__.py,sha256=ggI5fIUu-szgC44ICzuOmpYrIoVOKPbsMT3zza9ssD4,87
 warpzone/monitor/logs.py,sha256=fabjaB5SfHynvvfp2Js3IG-owqU5jZ3lTnnmTTjD6JM,1320
-warpzone/monitor/traces.py,sha256=B_1_fmUNUwHRKB-IYfhWSw5VRluVvn1NQqSYTqNV0wA,4158
+warpzone/monitor/traces.py,sha256=wmpA_6P8Zbb0O07NUg_6MV-Q4Gp0ULRYSNOaWGcM0hw,4087
 warpzone/servicebus/data/__init__.py,sha256=lnc0uiaGLF0qMi_rWhCpRSFvaj0CJEiMCAl6Yqn1ZiA,21
 warpzone/servicebus/data/client.py,sha256=zECS3JwedhYnDk8PntYgIYpBF_uu9YN38KzpPFK7CKs,6511
 warpzone/servicebus/events/__init__.py,sha256=lnc0uiaGLF0qMi_rWhCpRSFvaj0CJEiMCAl6Yqn1ZiA,21
@@ -52,6 +52,6 @@ warpzone/tools/copy.py,sha256=5fddotMZkXZO8avzUbGOhvs0cp8mce95pNpy0oPVjnQ,2596
 warpzone/transform/__init__.py,sha256=ruGa7tl-v4ndlWpULE1jSGU_a4_iRc3V6eyNr5xKP9E,27
 warpzone/transform/data.py,sha256=Abb8PcrgMbbNCJkkIUdtrTHdlY0OfXid387qw1nDpFY,2362
 warpzone/transform/schema.py,sha256=nbSQtDMvXkyqGKuwhuFCF0WsEDsaNyoPYpMKvbsKlv8,2423
-warpzone_sdk-15.0.0.dev4.dist-info/METADATA,sha256=jxPQ1HTdmdO0ePXvZAFg58OmIGw38qssEwnc6o9eTAI,7398
-warpzone_sdk-15.0.0.dev4.dist-info/WHEEL,sha256=3ny-bZhpXrU6vSQ1UPG34FoxZBp3lVcvK0LkgUz6VLk,88
-warpzone_sdk-15.0.0.dev4.dist-info/RECORD,,
+warpzone_sdk-15.0.0.dev5.dist-info/METADATA,sha256=0AcIiN1etcw_7hPsWEDb4fsaHm5MkWtwEgPq0HYu7GM,7398
+warpzone_sdk-15.0.0.dev5.dist-info/WHEEL,sha256=3ny-bZhpXrU6vSQ1UPG34FoxZBp3lVcvK0LkgUz6VLk,88
+warpzone_sdk-15.0.0.dev5.dist-info/RECORD,,