Kris
56022a9199
- 修复 scheduler store 末尾多余逗号 - 将智能体名称从“语析”改为“Kris” - 新增获取运行记录UID的权限校验接口 - 重构操作日志写入逻辑，使用独立会话避免事务污染 - 注册外部系统调度处理器 - 优化全局错误处理器，统一响应格式与序列化处理 - 新增调度任务运行日志的handler_name字段与索引 - 重构任务执行函数，新增payload覆盖与操作人审计参数 - 重写外部系统store，新增侧边栏折叠状态与持久化 - 优化会话、消息等模型的索引、约束与字段类型 - 新增多项配置项与环境变量支持 - 重构渠道相关模型，新增索引、约束与字段优化 - 新增渠道插件模型与内容审核模型的完善
102 lines
3.9 KiB
Python
102 lines
3.9 KiB
Python
"""全局异常处理器集中实现。
|
||
|
||
3 个 handler：
|
||
- unified_error_handler：处理三模块异常（满足 UnifiedError 协议）
|
||
- request_validation_error_handler：处理 Pydantic 请求体校验失败
|
||
- unhandled_exception_handler：兜底原生 Exception
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
import logging
|
||
|
||
from fastapi import Request
|
||
from fastapi.exceptions import RequestValidationError
|
||
from fastapi.responses import JSONResponse
|
||
from yuxi.channels.contract.errors import Error as ChannelError
|
||
from yuxi.external_systems.exceptions import ExternalSystemError
|
||
from yuxi.scheduler.exceptions import SchedulerError
|
||
|
||
from server.utils.error_response import build_error_response, build_error_response_from_exc
|
||
|
||
|
||
async def unified_error_handler(
|
||
_request: Request,
|
||
exc: ChannelError | ExternalSystemError | SchedulerError,
|
||
) -> JSONResponse:
|
||
"""三模块异常统一处理器。
|
||
|
||
三模块异常基类都满足 UnifiedError 协议（status_code/error_code/message/
|
||
details/trace_id），通过 build_error_response_from_exc() 统一构建响应。
|
||
不再有 isinstance 特判，不再有查表/类属性分支，逻辑完全相同。
|
||
|
||
Args:
|
||
_request: FastAPI 请求对象（未使用）。
|
||
exc: 三模块异常之一。
|
||
|
||
Returns:
|
||
统一格式的错误响应。
|
||
"""
|
||
return build_error_response_from_exc(exc)
|
||
|
||
|
||
async def unhandled_exception_handler(_request: Request, exc: Exception) -> JSONResponse:
|
||
"""原生异常兜底处理器。
|
||
|
||
所有未被特定异常处理器捕获的 Exception 在此统一兜底，
|
||
返回结构化 500 响应。trace_id 自动从 ContextVar 获取。
|
||
|
||
Args:
|
||
_request: FastAPI 请求对象（未使用）。
|
||
exc: 未被捕获的原生异常。
|
||
|
||
Returns:
|
||
500 内部服务器错误响应。
|
||
"""
|
||
logging.getLogger("uvicorn.error").exception("unhandled exception", exc_info=exc)
|
||
return build_error_response(
|
||
status_code=500,
|
||
code="INTERNAL_SERVER_ERROR",
|
||
message="服务器内部错误，请稍后重试",
|
||
details={"hint": "如果问题持续，请联系管理员并提供 trace_id"},
|
||
)
|
||
|
||
|
||
async def request_validation_error_handler(
|
||
_request: Request,
|
||
exc: RequestValidationError,
|
||
) -> JSONResponse:
|
||
"""Pydantic 请求体校验失败统一处理器。
|
||
|
||
将 FastAPI 默认的 422 + ``{"detail": [...]}`` 转换为项目统一格式
|
||
422 + ``{"success": false, "error": {...}}``，确保响应格式一致。
|
||
|
||
Pydantic v2 的 ``model_validator`` / ``field_validator`` 中抛出的
|
||
``ValueError`` 子类（含项目的 ``ValidationError``）会被包装为
|
||
``pydantic.ValidationError``，再由 FastAPI 转为 ``RequestValidationError``，
|
||
本处理器统一捕获并转换为项目格式。状态码保持 422（FastAPI 标准），
|
||
仅统一响应结构。
|
||
|
||
``exc.errors()`` 返回的 error dict 中 ``ctx`` 字段可能携带非 JSON 可序列化
|
||
对象（如 ``field_validator`` 抛出 ``ValueError`` 时 ctx 为
|
||
``{"error": <ValueError 实例>}``）。直接放入 ``JSONResponse`` 会触发
|
||
``TypeError``，被 ``unhandled_exception_handler`` 兜底为 500。此处通过
|
||
``json.dumps(..., default=str)`` 将非序列化对象降级为字符串，确保响应
|
||
始终可序列化。
|
||
"""
|
||
errors = json.loads(json.dumps(exc.errors(), default=str))
|
||
first = errors[0] if errors else {}
|
||
loc = first.get("loc", [])
|
||
field = ".".join(str(part) for part in loc) if loc else ""
|
||
message = first.get("msg", "request validation failed")
|
||
# Pydantic 对 ValueError 子类的包装会加 "Value error, " 前缀，去除以还原原始 message
|
||
if message.startswith("Value error, "):
|
||
message = message[len("Value error, ") :]
|
||
return build_error_response(
|
||
status_code=422,
|
||
code="VALIDATION_ERROR",
|
||
message=message,
|
||
details={"field": field, "errors": errors},
|
||
)
|