forgot to add the files

Author: extreme4all

src/core/fastapi/__init__.py

@@ -0,0 +1,4 @@
+from .logging import LoggingMiddleware
+from .metrics import PrometheusMiddleware
+
+__all__ = ["LoggingMiddleware", "PrometheusMiddleware"]

src/core/fastapi/middleware/__init__.py

@@ -0,0 +1,28 @@
+import logging
+import time
+
+from fastapi import Request
+from starlette.middleware.base import BaseHTTPMiddleware
+
+logger = logging.getLogger(__name__)
+
+
+class LoggingMiddleware(BaseHTTPMiddleware):
+    async def dispatch(self, request: Request, call_next):
+        start_time = time.perf_counter()
+        response = await call_next(request)
+        process_time = time.perf_counter() - start_time
+
+        query_params_list = [
+            (key, value if key != "token" else "***")
+            for key, value in request.query_params.items()
+        ]
+
+        logger.info(
+            {
+                "url": request.url.path,
+                "params": query_params_list,
+                "process_time": f"{process_time:.4f}",
+            }
+        )
+        return response

src/core/fastapi/middleware/logging.py

@@ -0,0 +1,38 @@
+from prometheus_client.metrics import Counter, Histogram
+
+import time
+from starlette.middleware.base import BaseHTTPMiddleware
+from fastapi import Request
+
+# Define Prometheus metrics
+REQUEST_COUNT = Counter(
+    "request_count", "Total number of requests", ["method", "endpoint", "http_status"]
+)
+REQUEST_LATENCY = Histogram(
+    "request_latency_seconds", "Latency of requests in seconds", ["method", "endpoint"]
+)
+
+
+# Middleware for Prometheus metrics logging
+class PrometheusMiddleware(BaseHTTPMiddleware):
+    async def dispatch(self, request: Request, call_next):
+        REQUEST_COUNT.labels(
+            method=request.method,
+            endpoint=request.url.path
+        ).inc()
+
+        # Start timer for request latency
+        start_time = time.perf_counter()
+
+        # Process request
+        response = await call_next(request)
+
+        # Calculate request latency
+        latency = time.perf_counter() - start_time
+
+        REQUEST_LATENCY.labels(
+            method=request.method,
+            endpoint=request.url.path,
+        ).observe(latency)
+
+        return response