feat(common): 建立公共归属与日志基础能力

This commit is contained in:
bruce
2026-09-09 13:01:36 +08:00
parent 57a20ff477
commit f4a92699f0
6 changed files with 116 additions and 0 deletions
+27
View File
@@ -0,0 +1,27 @@
"""请求级关联标识中间件。"""
import re
import uuid
from .logging import request_id_context
REQUEST_ID_PATTERN = re.compile(r"^[A-Za-z0-9._-]{1,64}$")
class RequestContextMiddleware:
"""复用合法入站标识或生成新标识,并确保上下文在请求结束后复位。"""
def __init__(self, get_response):
self.get_response = get_response
def __call__(self, request):
incoming = request.headers.get("X-Request-ID", "")
request_id = incoming if REQUEST_ID_PATTERN.fullmatch(incoming) else uuid.uuid4().hex
token = request_id_context.set(request_id)
request.request_id = request_id
try:
response = self.get_response(request)
response["X-Request-ID"] = request_id
return response
finally:
request_id_context.reset(token)