File size: 937 Bytes
9b5157d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | """
Correlation ID middleware.
Reads X-Request-ID from incoming headers (or generates one) and makes it
available via a ContextVar so it can be injected into logs and error responses
without threading issues.
"""
import uuid
from contextvars import ContextVar
from starlette.middleware.base import BaseHTTPMiddleware
from starlette.requests import Request
_request_id_ctx: ContextVar[str] = ContextVar("request_id", default="")
def get_request_id() -> str:
return _request_id_ctx.get()
class CorrelationIdMiddleware(BaseHTTPMiddleware):
async def dispatch(self, request: Request, call_next):
request_id = request.headers.get("X-Request-ID") or str(uuid.uuid4())
token = _request_id_ctx.set(request_id)
try:
response = await call_next(request)
response.headers["X-Request-ID"] = request_id
return response
finally:
_request_id_ctx.reset(token)
|