55 lines
2.0 KiB
Python
55 lines
2.0 KiB
Python
from fastapi import Request, HTTPException, status
|
|
from fastapi.responses import JSONResponse
|
|
from sqlalchemy.exc import SQLAlchemyError
|
|
from config.logging import logger
|
|
|
|
|
|
# 自定义异常类
|
|
class AppException(Exception):
|
|
def __init__(self, message: str, details=None):
|
|
self.message = message
|
|
self.details = details
|
|
|
|
|
|
# 全局异常处理中间件
|
|
async def global_exception_handler(request: Request, call_next):
|
|
try:
|
|
# 记录请求信息（可选）
|
|
logger.info(f"请求: {request.method} {request.url}")
|
|
if request.query_params:
|
|
logger.info(f"查询参数: {dict(request.query_params)}")
|
|
|
|
response = await call_next(request)
|
|
|
|
# 记录响应信息（可选）
|
|
if response.status_code >= 400:
|
|
logger.warning(f"响应: {response.status_code}")
|
|
|
|
return response
|
|
|
|
except AppException as e:
|
|
# 记录业务异常
|
|
logger.error(f"业务异常: {e.message} - 详情: {e.details}")
|
|
return JSONResponse(status_code=status.HTTP_400_BAD_REQUEST,
|
|
content={"message": e.message, "details": e.details})
|
|
|
|
except SQLAlchemyError as e:
|
|
# 记录数据库异常
|
|
#logger.critical(f"数据库异常: {str(e)}", exc_info=True)
|
|
return JSONResponse(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|
content={"message": "数据库操作失败", "details": str(e)})
|
|
|
|
except Exception as e:
|
|
# 记录未知异常（带堆栈信息）
|
|
#logger.critical(f"未知异常: {str(e)}", exc_info=True)
|
|
return JSONResponse(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|
content={"code": 500, "message": "服务器内部错误", "details": str(e)})
|
|
|
|
|
|
def get_credentials_exception() -> HTTPException:
|
|
return HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail="Could not validate credentials",
|
|
headers={"WWW-Authenticate": "Bearer"},
|
|
)
|