arsam
9082465a93
Import the existing local project snapshot. This commit organizes existing files; it does not represent newly developed functionality.
25 lines
804 B
Python
25 lines
804 B
Python
import time
|
|
import uuid
|
|
|
|
from starlette.middleware.base import BaseHTTPMiddleware
|
|
from starlette.requests import Request
|
|
|
|
from axon.services.metrics import metrics
|
|
|
|
|
|
class RequestContextMiddleware(BaseHTTPMiddleware):
|
|
async def dispatch(self, request: Request, call_next):
|
|
request_id = request.headers.get("X-Request-ID") or str(uuid.uuid4())
|
|
start = time.monotonic()
|
|
status_code = 500
|
|
|
|
try:
|
|
response = await call_next(request)
|
|
status_code = response.status_code
|
|
return response
|
|
finally:
|
|
latency_ms = (time.monotonic() - start) * 1000
|
|
await metrics.record(request.url.path, status_code, latency_ms)
|
|
if "response" in locals():
|
|
response.headers["X-Request-ID"] = request_id
|