warpzone-sdk 15.0.0.dev7__tar.gz → 15.0.0.dev8__tar.gz

{warpzone_sdk-15.0.0.dev7 → warpzone_sdk-15.0.0.dev8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: warpzone-sdk
-Version: 15.0.0.dev7
+Version: 15.0.0.dev8
 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.dev7 → warpzone_sdk-15.0.0.dev8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "warpzone-sdk"
-version = "15.0.0.dev7"
+version = "15.0.0.dev8"
 description = "The main objective of this package is to centralize logic used to interact with Azure Functions, Azure Service Bus and Azure Table Storage"
 authors = [{ name = "Team Enigma", email = "enigma@energinet.dk" }]
 requires-python = ">=3.10"

warpzone_sdk-15.0.0.dev8/warpzone/function/monitor.py

@@ -0,0 +1,53 @@
+import inspect
+from contextlib import contextmanager
+from typing import Callable
+
+import azure.functions as func
+
+from warpzone.function.types import SingleArgumentCallable
+from warpzone.monitor import traces
+
+SUBJECT_IDENTIFIER = "<Subject>"
+
+
+@contextmanager
+def run_in_trace_context(context: func.Context):
+    trace_context = context.trace_context
+    span_name = context.function_name or "azure_function"
+    with traces.set_trace_context(trace_context.trace_parent, span_name=span_name):
+        yield
+
+
+def monitor(main: SingleArgumentCallable) -> Callable:
+    """Wrap Azure function with logging and tracing
+    configured for monitoring in App Insights.
+
+    Args:
+        f (SingleArgumentCallable): Azure function to be wrapped
+
+    Returns:
+        Callable: Azure function with
+            - argument
+                name: "arg"
+                description: argument of the original function
+            - argument
+                name: "context"
+                description: Azure function context
+            - return value
+                description: return value of original function
+    """
+
+    async def wrapper_async(arg, context: func.Context):
+        with run_in_trace_context(context):
+            result = await main(arg)
+            return result
+
+    def wrapper(arg, context: func.Context):
+        with run_in_trace_context(context):
+            result = main(arg)
+            return result
+
+    if inspect.iscoroutinefunction(main):
+        return wrapper_async
+    else:
+        return wrapper

warpzone_sdk-15.0.0.dev7/warpzone/function/monitor.py

@@ -1,93 +0,0 @@
-import inspect
-import threading
-from contextlib import contextmanager
-from typing import Callable
-
-import azure.functions as func
-from azure.monitor.opentelemetry import configure_azure_monitor
-from opentelemetry import trace
-
-from warpzone.function.types import SingleArgumentCallable
-from warpzone.monitor import traces
-
-# Thread-safe lazy initialization of Azure Monitor
-_azure_monitor_configured = False
-_azure_monitor_lock = threading.Lock()
-
-
-def _ensure_azure_monitor_configured():
-    """Ensure Azure Monitor is configured exactly once, thread-safely."""
-    global _azure_monitor_configured
-    if _azure_monitor_configured:
-        return
-
-    with _azure_monitor_lock:
-        # Double-check after acquiring lock
-        if _azure_monitor_configured:
-            return
-
-        # Disable OpenTelemetry logging instrumentation - let Azure Functions
-        # handle log correlation via its native thread-local storage mechanism.
-        # This prevents conflicts between OpenTelemetry's context-based correlation
-        # and Azure Functions' invocation_id-based correlation which can cause
-        # logs to leak between concurrent function invocations.
-        configure_azure_monitor(
-            disable_logging=True,
-        )
-
-        # Apply trace filtering to suppress all Azure SDK traces except Service Bus
-        tracer_provider = trace.get_tracer_provider()
-        if hasattr(tracer_provider, "_active_span_processor"):
-            original_processor = tracer_provider._active_span_processor
-            filtered_processor = traces.AzureSDKTraceFilter(original_processor)
-            tracer_provider._active_span_processor = filtered_processor
-
-        _azure_monitor_configured = True
-
-
-SUBJECT_IDENTIFIER = "<Subject>"
-
-
-@contextmanager
-def run_in_trace_context(context: func.Context):
-    trace_context = context.trace_context
-    span_name = context.function_name or "azure_function"
-    with traces.set_trace_context(trace_context.trace_parent, span_name=span_name):
-        yield
-
-
-def monitor(main: SingleArgumentCallable) -> Callable:
-    """Wrap Azure function with logging and tracing
-    configured for monitoring in App Insights.
-
-    Args:
-        f (SingleArgumentCallable): Azure function to be wrapped
-
-    Returns:
-        Callable: Azure function with
-            - argument
-                name: "arg"
-                description: argument of the original function
-            - argument
-                name: "context"
-                description: Azure function context
-            - return value
-                description: return value of original function
-    """
-    # Ensure Azure Monitor is configured before any function runs
-    _ensure_azure_monitor_configured()
-
-    async def wrapper_async(arg, context: func.Context):
-        with run_in_trace_context(context):
-            result = await main(arg)
-            return result
-
-    def wrapper(arg, context: func.Context):
-        with run_in_trace_context(context):
-            result = main(arg)
-            return result
-
-    if inspect.iscoroutinefunction(main):
-        return wrapper_async
-    else:
-        return wrapper