30 lines
787 B
Python
30 lines
787 B
Python
from fastapi import FastAPI, Request
|
|
from starlette.middleware.base import BaseHTTPMiddleware, RequestResponseEndpoint
|
|
from starlette.responses import Response
|
|
|
|
from common.context import RequestContext
|
|
|
|
|
|
class ContextCleanupMiddleware(BaseHTTPMiddleware):
|
|
"""
|
|
上下文清理中间件
|
|
"""
|
|
|
|
async def dispatch(self, request: Request, call_next: RequestResponseEndpoint) -> Response:
|
|
"""
|
|
在每个请求处理完成后清理上下文信息
|
|
"""
|
|
try:
|
|
return await call_next(request)
|
|
finally:
|
|
RequestContext.clear_all()
|
|
|
|
|
|
def add_context_cleanup_middleware(app: FastAPI) -> None:
|
|
"""
|
|
添加上下文清理中间件
|
|
|
|
:param app: FastAPI对象
|
|
"""
|
|
app.add_middleware(ContextCleanupMiddleware)
|