Update middleware code in main.py

rest-api
CIzz22 2 months ago
parent da2ce29c45
commit bbfa010b39

@ -31,7 +31,7 @@ from src.rate_limiter import limiter
from src.api import api_router
from src.database.core import engine, async_session, collector_async_session
from src.exceptions import handle_exception
from src.middleware import RequestValidationMiddleware
log = logging.getLogger(__name__)
@ -58,6 +58,8 @@ _request_id_ctx_var: ContextVar[Optional[str]] = ContextVar(
def get_request_id() -> Optional[str]:
return _request_id_ctx_var.get()
app.add_middleware(RequestValidationMiddleware)
@app.middleware("http")
async def db_session_middleware(request: Request, call_next):

Loading…
Cancel
Save