33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
"""日志中间件：记录 RequestID/AgentID 链路与访问日志，避免敏感载荷与日志刷屏。"""
|
||
import logging
|
||
import time
|
||
|
||
from starlette.middleware.base import BaseHTTPMiddleware
|
||
from starlette.requests import Request
|
||
|
||
logger = logging.getLogger("access")
|
||
|
||
|
||
class RequestLogMiddleware(BaseHTTPMiddleware):
|
||
async def dispatch(self, request: Request, call_next):
|
||
start = time.perf_counter()
|
||
# 心跳类接口降级为 debug，避免刷屏
|
||
is_heartbeat = request.url.path.endswith("/heartbeat")
|
||
if logger.isEnabledFor(logging.DEBUG) or not is_heartbeat:
|
||
logger.info(
|
||
"%s %s path=%s",
|
||
request.method,
|
||
request.client.host if request.client else "-",
|
||
request.url.path,
|
||
)
|
||
response = await call_next(request)
|
||
cost_ms = (time.perf_counter() - start) * 1000
|
||
if not is_heartbeat or logger.isEnabledFor(logging.DEBUG):
|
||
logger.info(
|
||
"%s %s -> %s (%dms)",
|
||
request.method,
|
||
request.url.path,
|
||
response.status_code,
|
||
round(cost_ms, 2),
|
||
)
|
||
return response |