mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-10-03 08:49:14 +00:00
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
import time
|
|
|
|
from fastapi import Request, Response
|
|
from starlette.middleware.base import BaseHTTPMiddleware, RequestResponseEndpoint
|
|
|
|
from backend.common.context import ctx
|
|
from backend.common.log import log
|
|
from backend.common.observability.prometheus import (
|
|
PROMETHEUS_APP_NAME,
|
|
PROMETHEUS_REQUEST_COUNTER,
|
|
PROMETHEUS_REQUEST_IN_PROGRESS_GAUGE,
|
|
)
|
|
from backend.core.conf import settings
|
|
from backend.utils.timezone import timezone
|
|
|
|
|
|
class AccessMiddleware(BaseHTTPMiddleware):
|
|
"""访问日志中间件"""
|
|
|
|
async def dispatch(self, request: Request, call_next: RequestResponseEndpoint) -> Response:
|
|
"""
|
|
处理请求并记录访问日志
|
|
|
|
:param request: FastAPI 请求对象
|
|
:param call_next: 下一个中间件或路由处理函数
|
|
:return:
|
|
"""
|
|
perf_time = time.perf_counter()
|
|
ctx.perf_time = perf_time
|
|
|
|
start_time = timezone.now()
|
|
ctx.start_time = start_time
|
|
|
|
path = request.url.path
|
|
method = request.method
|
|
|
|
if method != 'OPTIONS':
|
|
log.debug(f'--> 请求开始[{path if not request.url.query else request.url.path + "/" + request.url.query}]')
|
|
|
|
if path.startswith(settings.FASTAPI_API_V1_PATH):
|
|
PROMETHEUS_REQUEST_IN_PROGRESS_GAUGE.labels(app_name=PROMETHEUS_APP_NAME, method=method, path=path).inc()
|
|
PROMETHEUS_REQUEST_COUNTER.labels(app_name=PROMETHEUS_APP_NAME, method=method, path=path).inc()
|
|
|
|
# 为每个请求上下文注入默认租户 ID，授权接口认证成功后会覆盖为真实值
|
|
ctx.tenant_id = settings.TENANT_DEFAULT_ID
|
|
|
|
response = await call_next(request)
|
|
|
|
return response
|