add x-request-id

This commit is contained in:
朱潮 2025-12-16 10:49:32 +08:00
parent 402e2ea5c3
commit 73b87bd2eb

View File

@ -15,10 +15,10 @@ def add_request_routes(app: FastAPI):
@app.middleware("http") @app.middleware("http")
async def before_request(request: Request, call_next: Callable[[Request], Awaitable[JSONResponse]]): async def before_request(request: Request, call_next: Callable[[Request], Awaitable[JSONResponse]]):
# 先从header中获取X-Trace-Id如果没有则生成新的 # 先从header中获取x-request-id如果没有则生成新的
trace_id = request.headers.get('X-Trace-Id') trace_id = request.headers.get('X-Request-ID')
if not trace_id: if not trace_id:
trace_id = "generate_" + str(uuid.uuid4()) trace_id = "agent_" + str(uuid.uuid4())
# user_id = "未知的 user_id" # user_id = "未知的 user_id"
@ -26,7 +26,7 @@ def add_request_routes(app: FastAPI):
# g.user_id = user_id # g.user_id = user_id
response = await call_next(request) response = await call_next(request)
response.headers['X-Trace-Id'] = g.trace_id response.headers['X-Request-ID'] = g.trace_id
return response return response