FastAPI中如何区分处理RequestValidationError与中间件逻辑

核心问题 #

当前代码存在三个关键问题：

• 自定义RequestValidationError异常处理器的响应会被HTTP中间件二次包装，违背“仅让参数错误触发指定处理器”的需求
• 中间件捕获所有异常后，即使无异常处理器也返回200 OK，掩盖了真实错误状态
• 中间件无法直接捕获RequestValidationError，因为call_next执行时，FastAPI内部已用异常处理器处理该异常，不会将其抛出到中间层

解决方案 #

利用FastAPI的请求流程（中间件→路由/异常处理器→中间件），通过标记特殊响应+中间件识别跳过的方式实现需求：

1. 给参数错误响应添加唯一标识 #

修改RequestValidationError处理器，在响应头或内容中加入识别标记，同时保留正确的状态码：

from fastapi import Request, status
from fastapi.exceptions import RequestValidationError
from fastapi.responses import JSONResponse

@app.exception_handler(RequestValidationError)
async def validation_exception_handler(request: Request, exc: RequestValidationError):
    response_content = prepare_response({}, g_ERROR__INCORRECT_PARAMS)
    # 用自定义响应头标记参数错误响应
    return JSONResponse(
        content=response_content,
        headers={"X-Validation-Error": "true"},
        status_code=status.HTTP_422_UNPROCESSABLE_ENTITY
    )

2. 修改中间件逻辑，识别并跳过参数错误响应 #

在中间件中检查响应标记，若为参数错误则直接返回；仅对正常响应和未被捕获的异常进行处理：

from fastapi import Request
from fastapi.responses import JSONResponse

@app.middleware("http")
async def response_middleware(request: Request, call_next):
    try:
        result = await call_next(request)
        
        # 识别参数错误响应，直接返回不包装
        if result.headers.get("X-Validation-Error") == "true":
            return result
        
        # 正常响应的包装逻辑
        res_body = b''
        async for chunk in result.body_iterator:
            res_body += chunk
        
        response = prepare_response(res_body.decode(), g_ERROR__ALL_OK)
        return JSONResponse(content=response)
    
    except Exception as e:
        # 仅处理未被上层异常处理器捕获的未知异常
        response = prepare_response({}, g_ERROR__UNKNOWN_ERROR)
        return JSONResponse(content=response)

替代方案：通过响应内容字段识别 #

如果不想用自定义响应头，可在prepare_response生成的内容中加入特定错误码，中间件解析后判断：

# 中间件内修改响应处理逻辑
try:
    result = await call_next(request)
    res_body = b''
    async for chunk in result.body_iterator:
        res_body += chunk
    
    # 尝试解析响应体，判断是否为参数错误
    try:
        res_json = json.loads(res_body)
        if res_json.get("error_code") == g_ERROR__INCORRECT_PARAMS:
            return JSONResponse(content=res_json, status_code=result.status_code)
    except json.JSONDecodeError:
        pass
    
    # 正常响应包装
    response = prepare_response(res_body.decode(), g_ERROR__ALL_OK)
    return JSONResponse(content=response)
except Exception as e:
    response = prepare_response({}, g_ERROR__UNKNOWN_ERROR)
    return JSONResponse(content=response)

关键原理说明 #

FastAPI在执行call_next时，会优先调用注册的异常处理器处理RequestValidationError，因此该异常不会被中间件的try-except捕获，而是直接返回处理器生成的响应。必须通过响应的特征（头/内容字段）来识别该情况，避免二次包装。

内容的提问来源于stack exchange，提问作者Zhihar