Create utils/monitoring.py

utils/monitoring.py

@@ -0,0 +1,46 @@
+import logging
+from prometheus_client import Counter, Histogram, start_http_server
+from functools import wraps
+import time
+from typing import Callable, Any
+
+class MetricsManager:
+    def __init__(self):
+        self.requests_total = Counter(
+            'requests_total',
+            'Total requests processed',
+            ['endpoint']
+        )
+        self.request_duration = Histogram(
+            'request_duration_seconds',
+            'Request duration in seconds',
+            ['endpoint']
+        )
+        self.errors_total = Counter(
+            'errors_total',
+            'Total errors encountered',
+            ['type']
+        )
+        
+        # Start Prometheus metrics server
+        start_http_server(8000)
+
+def monitor_performance(endpoint: str) -> Callable:
+    def decorator(func: Callable) -> Callable:
+        @wraps(func)
+        def wrapper(*args, **kwargs) -> Any:
+            start_time = time.time()
+            try:
+                result = func(*args, **kwargs)
+                metrics_manager.requests_total.labels(endpoint=endpoint).inc()
+                return result
+            except Exception as e:
+                metrics_manager.errors_total.labels(type=type(e).__name__).inc()
+                raise
+            finally:
+                duration = time.time() - start_time
+                metrics_manager.request_duration.labels(endpoint=endpoint).observe(duration)
+        return wrapper
+    return decorator
+
+metrics_manager = MetricsManager()