add x-request-id
This commit is contained in:
parent
402e2ea5c3
commit
73b87bd2eb
@ -15,10 +15,10 @@ def add_request_routes(app: FastAPI):
|
||||
@app.middleware("http")
|
||||
async def before_request(request: Request, call_next: Callable[[Request], Awaitable[JSONResponse]]):
|
||||
|
||||
# 先从header中获取X-Trace-Id,如果没有则生成新的
|
||||
trace_id = request.headers.get('X-Trace-Id')
|
||||
# 先从header中获取x-request-id,如果没有则生成新的
|
||||
trace_id = request.headers.get('X-Request-ID')
|
||||
if not trace_id:
|
||||
trace_id = "generate_" + str(uuid.uuid4())
|
||||
trace_id = "agent_" + str(uuid.uuid4())
|
||||
|
||||
# user_id = "未知的 user_id"
|
||||
|
||||
@ -26,7 +26,7 @@ def add_request_routes(app: FastAPI):
|
||||
# g.user_id = user_id
|
||||
response = await call_next(request)
|
||||
|
||||
response.headers['X-Trace-Id'] = g.trace_id
|
||||
response.headers['X-Request-ID'] = g.trace_id
|
||||
return response
|
||||
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user