analyze performance upgrade

django_project/IoTAnalyzer/middleware.py

@@ -0,0 +1,36 @@
+"""Request logging for the API."""
+
+from __future__ import annotations
+
+import logging
+import time
+import uuid
+
+logger = logging.getLogger("IoTAnalyzer.request")
+
+
+class RequestLogMiddleware:
+    """Log one line per API request with duration, status, user and a request id."""
+
+    def __init__(self, get_response):
+        self.get_response = get_response
+
+    def __call__(self, request):
+        started = time.time()
+        request_id = request.headers.get("X-Request-ID") or uuid.uuid4().hex[:12]
+        request.request_id = request_id
+        try:
+            response = self.get_response(request)
+        except Exception:
+            logger.exception("%s %s failed after %.0f ms [rid=%s]", request.method,
+                             request.get_full_path(), (time.time() - started) * 1000, request_id)
+            raise
+        response["X-Request-ID"] = request_id
+        if request.path.startswith("/api/") and request.path != "/api/health":
+            user = getattr(getattr(request, "user", None), "username", None) or "-"
+            level = logging.WARNING if response.status_code >= 500 else \
+                logging.INFO if response.status_code >= 400 else logging.DEBUG
+            logger.log(level, "%s %s -> %s in %.0f ms user=%s [rid=%s]", request.method,
+                       request.get_full_path(), response.status_code,
+                       (time.time() - started) * 1000, user, request_id)
+        return response