袁聪的第二次提交,项目已完整
This commit is contained in:
@@ -0,0 +1,32 @@
|
||||
"""HTTP 层横切关注点。
|
||||
|
||||
目前只有一项:统一回传 `X-Trace-ID`。接口文档把它列为所有响应的必备头,
|
||||
但此前**完全没有实现**——客户端拿到错误也无法把请求与服务端日志对上。
|
||||
|
||||
放在独立模块而不是写在 `create_app()` 里,是为了让它可被单独测试:
|
||||
中间件函数不需要起整个应用就能验证三条分支(有请求上下文 / 只有请求头 / 两者都没有)。
|
||||
"""
|
||||
|
||||
from collections.abc import Awaitable, Callable
|
||||
|
||||
from starlette.requests import Request
|
||||
from starlette.responses import Response
|
||||
|
||||
TRACE_ID_HEADER = "X-Trace-ID"
|
||||
|
||||
RequestHandler = Callable[[Request], Awaitable[Response]]
|
||||
|
||||
|
||||
async def attach_trace_id(request: Request, call_next: RequestHandler) -> Response:
|
||||
"""给响应附加 `X-Trace-ID`。
|
||||
|
||||
优先级:请求上下文里的 `trace_id`(服务端生成或从令牌解析)→ 请求头里客户端带来的
|
||||
`X-Trace-ID`(透传,便于把客户端日志与服务端串起来)。两者都没有就不加头:
|
||||
认证失败等场景还没建立上下文,此时凭空造一个 id 只会误导排障。
|
||||
"""
|
||||
response = await call_next(request)
|
||||
context = getattr(request.state, "request_context", None)
|
||||
trace_id = getattr(context, "trace_id", None) or request.headers.get(TRACE_ID_HEADER)
|
||||
if trace_id:
|
||||
response.headers[TRACE_ID_HEADER] = str(trace_id)
|
||||
return response
|
||||
Reference in New Issue
Block a user