可以通过重写 ExceptionFilterAttribute 类的 OnException 方法来实现在 API 异常时返回正确的 HTTP 响应代码。
以下是代码示例:
public class CustomExceptionFilterAttribute : ExceptionFilterAttribute
{
public override void OnException(HttpActionExecutedContext context)
{
if (context.Exception != null)
{
// 根据异常类型设置要返回的 HTTP 响应代码
var responseCode = HttpStatusCode.InternalServerError;
if (context.Exception is AccessDeniedException)
{
responseCode = HttpStatusCode.Forbidden;
}
else if (context.Exception is NotFoundException)
{
responseCode = HttpStatusCode.NotFound;
}
// 创建 HttpResponseMessage 对象并设置要返回的 HTTP 响应代码
var response = context.Request.CreateResponse(responseCode);
// 将错误信息写入响应体中
response.Content = new StringContent(context.Exception.Message);
// 替换执行结果为新的 HttpResponseMessage 对象
context.Response = response;
}
}
}
使用方法:
可以将 CustomExceptionFilterAttribute 属性添加到 API 控制器中的某个方法上,以使该方法调用后产生的异常能够被过滤器捕获并返回正确的 HTTP 响应代码。
例如:
[CustomExceptionFilter]
public IHttpActionResult Get(int id)
{
var result = service.GetData(id);
if (result == null)
{
throw new NotFoundException("Data not found.");
}
return Ok(result);
}