JiuContinent
dc8605ce3f
refactor(core): 重构核心模块结构并更新导入路径 - 将配置相关的设置从 app.core.config 移除 - 将常量定义从 app.core.constants 移除 - 将数据库相关功能从 app.core.database 移除 - 将基础数据库模型从 app.core.db_base 移除 - 将敏感信息掩码功能从 app.core.masking 移除 - 将中间件定义从 app.core.middleware 移除 - 将操作保护功能从 app.core.operation_guard 移除 - 将分页工具从 app.core.pagination 移除 - 将请求上下文管理从 app.core.request_context 移除 - 将调度器功能从 app.core.scheduler 移除 - 将安全认证逻辑从 app.core.security 移除 - 将任务队列相关功能从 app.core.task_queue 移除 - 将时间工具从 app.core.time 移除 - 更新 alembic 配置中的 Base 模型导入路径 - 更新各模块中对重构后组件的引用路径 ```
19 lines
598 B
Python
19 lines
598 B
Python
import uuid
|
|
from collections.abc import Callable
|
|
|
|
from fastapi import Request, Response
|
|
|
|
from app.core.constants import HttpHeader
|
|
from app.core.http.request_context import reset_request_id, set_request_id
|
|
|
|
|
|
async def request_id_middleware(request: Request, call_next: Callable) -> Response:
|
|
request_id = request.headers.get(HttpHeader.X_REQUEST_ID) or uuid.uuid4().hex
|
|
token = set_request_id(request_id)
|
|
try:
|
|
response = await call_next(request)
|
|
response.headers[HttpHeader.X_REQUEST_ID] = request_id
|
|
return response
|
|
finally:
|
|
reset_request_id(token)
|