37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
"""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
|