Antoni Nuñez Romeu
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
import time
|
|
import uuid
|
|
|
|
from starlette.middleware.base import BaseHTTPMiddleware, RequestResponseEndpoint
|
|
from starlette.requests import Request
|
|
from starlette.responses import Response
|
|
|
|
from src.core.logging import get_logger
|
|
|
|
logger = get_logger("request_logging")
|
|
|
|
|
|
class RequestLoggingMiddleware(BaseHTTPMiddleware):
|
|
async def dispatch(self, request: Request, call_next: RequestResponseEndpoint) -> Response:
|
|
request_id = request.headers.get("X-Request-ID", str(uuid.uuid4()))
|
|
request.state.request_id = request_id
|
|
|
|
start = time.perf_counter()
|
|
response = await call_next(request)
|
|
duration_ms = (time.perf_counter() - start) * 1000
|
|
|
|
logger.info(
|
|
"request_completed",
|
|
extra={
|
|
"request_id": request_id,
|
|
"method": request.method,
|
|
"path": request.url.path,
|
|
"status_code": response.status_code,
|
|
"duration_ms": round(duration_ms, 2),
|
|
"client_ip": request.client.host if request.client else None,
|
|
"user_agent": request.headers.get("user-agent", ""),
|
|
},
|
|
)
|
|
|
|
response.headers["X-Request-ID"] = request_id
|
|
return response
|