Implement OpenTelemetry

.env.development

@@ -17,5 +17,8 @@ PACS_PORT=4244
 PACS_STORAGE_PATH=/var/lib/pacs/storage
 PACS_DB_PATH=/var/lib/pacs/pacs.db
 
+# Azure Application Insights (leave unset to disable telemetry locally)
+# APPLICATIONINSIGHTS_CONNECTION_STRING=InstrumentationKey=...
+
 # General Configuration
 LOG_LEVEL=INFO

pyproject.toml

@@ -15,6 +15,7 @@ dependencies = [
   "websockets==16.0",
   "requests>=2.32.0",
   "python-dotenv>=1.2.1",
+  "azure-monitor-opentelemetry>=1.8.6",
 ]
 
 [dependency-groups]

src/mwl_main.py

@@ -6,8 +6,10 @@
 from dotenv import load_dotenv
 
 from server import MWLServer
+from telemetry import configure_telemetry
 
 load_dotenv()
+configure_telemetry(service_name="mwl")
 
 
 def main():

src/pacs_main.py

@@ -6,8 +6,10 @@
 from dotenv import load_dotenv
 
 from server import PACSServer
+from telemetry import configure_telemetry
 
 load_dotenv()
+configure_telemetry(service_name="pacs")
 
 
 def main():

src/relay_listener.py

@@ -16,11 +16,12 @@
 
 from dotenv import load_dotenv
 from websockets.asyncio.client import connect
-from websockets.frames import CloseCode
 from websockets.exceptions import ConnectionClosedError
+from websockets.frames import CloseCode
 
 from services.mwl.create_worklist_item import CreateWorklistItem
 from services.storage import MWLStorage
+from telemetry import configure_telemetry
 
 load_dotenv()
 
@@ -130,6 +131,7 @@ async def main():
         level=os.getenv("LOG_LEVEL", "INFO").upper(),
         format=os.getenv("LOG_FORMAT", "%(asctime)s - %(name)s - %(levelname)s - %(message)s"),
     )
+    configure_telemetry(service_name="relay-listener")
 
     logger.info("Socket Listener Starting...")
     storage = MWLStorage(db_path=DB_PATH)

src/telemetry.py

@@ -0,0 +1,25 @@
+import logging
+import os
+
+logger = logging.getLogger(__name__)
+
+
+def configure_telemetry(service_name: str | None = None) -> None:
+    """Configure OpenTelemetry with Azure Monitor.
+
+    If APPLICATIONINSIGHTS_CONNECTION_STRING is not set, this is a no-op,
+    so local development works without any Azure configuration.
+
+    Args:
+        service_name: Identifies this service in Application Insights.
+    """
+    if not os.getenv("APPLICATIONINSIGHTS_CONNECTION_STRING"):
+        return
+
+    if service_name:
+        os.environ.setdefault("OTEL_SERVICE_NAME", service_name)
+
+    from azure.monitor.opentelemetry import configure_azure_monitor
+
+    configure_azure_monitor()
+    logger.info("Azure Monitor telemetry configured")

src/upload_main.py

@@ -9,8 +9,10 @@
 from services.dicom.upload_listener import UploadListener
 from services.dicom.upload_processor import UploadProcessor
 from services.storage import MWLStorage, PACSStorage
+from telemetry import configure_telemetry
 
 load_dotenv()
+configure_telemetry(service_name="upload")
 
 
 def main():