345cc73965
P0-1 SqlCredentialStore/save_draft 由自提交改 flush，端点/服务统一 commit （新增 CredentialStore.commit() 统一提交点；token 刷新落库显式提交）； 补多凭据一请求中途失败整体回滚集成测试。 P0-2 启动校验 _fernet(enc_key) 快速失败 + catch-all Exception → ErrorEnvelope； credential_enc_key 改 SecretStr。 P0-3 run_job 异常分类：AppError 存 code+message，其余存通用文案不泄 str(exc)。 P0-4 评审/正文 SSE 失败先发 error 事件，尾部 commit 包 try/except。 P1-4 max_version 加 FOR UPDATE 行锁消除 TOCTOU。 P1-5 scan_overdue 谓词下推 + 批量 UPDATE RETURNING。 P1-10 移除 OAuth user_code 日志。 P2 provider_deps 改调网关 build_adapter；accept_service Committable Protocol； CORS 白名单收窄；request_id 安全字符集白名单；stdlib 日志接管；读端点 404 校验； httpx timeout；测试用合法 Fernet key；类型化响应模型(JobResponse/DimensionEntry/ ReviewConflictView/selling_points)+路由 ErrorEnvelope responses（供 codegen）。
41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
"""request_id 关联中间件（ARCH §9.3 / CLAUDE.md）。
|
||
|
||
每请求生成/透传 request_id，绑入 structlog contextvars，并回写响应头，
|
||
错误信封也带上同一 id，便于端到端 grep。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import re
|
||
import uuid
|
||
from collections.abc import Awaitable, Callable
|
||
|
||
import structlog
|
||
from starlette.requests import Request
|
||
from starlette.responses import Response
|
||
|
||
REQUEST_ID_HEADER = "x-request-id"
|
||
|
||
# 放行安全字符集（字母数字 . _ -，1–128 位）：覆盖 uuid4().hex、常见 trace id 与客户端
|
||
# 自定义短 id；含空白/控制字符/换行/注入序列/超长的一律丢弃改生成新 uuid——防止把未经
|
||
# 校验的客户端值原样写进日志/响应头（日志注入防护）。
|
||
_REQUEST_ID_PATTERN = re.compile(r"^[A-Za-z0-9._-]{1,128}$")
|
||
|
||
|
||
def _sanitize_request_id(raw: str | None) -> str:
|
||
if raw is not None and _REQUEST_ID_PATTERN.match(raw):
|
||
return raw
|
||
return uuid.uuid4().hex
|
||
|
||
|
||
async def request_id_middleware(
|
||
request: Request, call_next: Callable[[Request], Awaitable[Response]]
|
||
) -> Response:
|
||
request_id = _sanitize_request_id(request.headers.get(REQUEST_ID_HEADER))
|
||
structlog.contextvars.clear_contextvars()
|
||
structlog.contextvars.bind_contextvars(request_id=request_id)
|
||
request.state.request_id = request_id
|
||
response = await call_next(request)
|
||
response.headers[REQUEST_ID_HEADER] = request_id
|
||
return response
|