Merge pull request #128 from NHSDigital/reinstate-opentelemetry

Reinstate opentelemetry

Author: steventux

src/mwl_main.py

@@ -6,6 +6,7 @@
 from dotenv import load_dotenv
 
 from server import MWLServer
+from telemetry import configure_telemetry
 
 load_dotenv()
 
@@ -30,6 +31,8 @@ def main():
 
     mwl_server = MWLServer(mwl_aet, mwl_port, mwl_db_path, block=True)
 
+    configure_telemetry(service_name="mwl-server")
+
     try:
         mwl_server.start()
     except KeyboardInterrupt:

src/pacs_main.py

@@ -6,6 +6,7 @@
 from dotenv import load_dotenv
 
 from server import PACSServer
+from telemetry import configure_telemetry
 
 load_dotenv()
 
@@ -33,6 +34,8 @@ def main():
 
     pacs_server = PACSServer(pacs_aet, pacs_port, pacs_storage_path, pacs_db_path, block=True, mwl_db_path=mwl_db_path)
 
+    configure_telemetry(service_name="pacs-server")
+
     try:
         pacs_server.start()
     except KeyboardInterrupt:

src/telemetry.py

@@ -3,6 +3,14 @@
 
 logger = logging.getLogger(__name__)
 
+# Suppress verbose logging from OpenTelemetry and Azure Monitor libraries
+for package_name in [
+    "azure.monitor.opentelemetry",
+    "azure.core.pipeline.policies.http_logging_policy",
+]:
+    package_logger = logging.getLogger(package_name)
+    package_logger.setLevel(logging.WARNING)
+
 
 def configure_telemetry(service_name: str | None = None) -> None:
     """Configure OpenTelemetry with Azure Monitor.

src/upload_main.py

@@ -9,6 +9,7 @@
 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()
 
@@ -67,6 +68,8 @@ def main():
     logging.info(f"API endpoint: {uploader.api_endpoint}")
     logging.info("=" * 60)
 
+    configure_telemetry(service_name="upload-listener")
+
     try:
         listener.start()
     except KeyboardInterrupt: