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")
|
@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
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user