ControllerAdvice用法
@ControllerAdvice
是一个组件注解,它允许你在一个地方处理整个应用程序控制器的异常、绑定数据和预处理请求。这意味着你不需要在每个控制器中重复相同的异常处理代码,从而使得代码更加简洁、易于管理。
主要特性
- 全局异常处理:通过
@ExceptionHandler
注解捕获全局异常,减少重复的错误处理代码。 - 数据绑定:使用
@InitBinder
注解自定义请求参数的格式化和转换。 - 全局数据预处理:通过
@ModelAttribute
注解添加全局模型属性,这些属性对所有的@RequestMapping
方法都是可见的。
全局异常处理
在开发过程中,统一异常处理是非常重要的一环。使用@ControllerAdvice
,你可以定义一个或多个方法来处理整个应用中的异常。这样,当应用抛出异常时,Spring会自动寻找带有@ExceptionHandler
的方法来处理它。
返回 ResponseEntity
使用@ControllerAdvice
进行全局异常处理的示例代码:
@ControllerAdvice
public class GlobalExceptionHandler {
@ExceptionHandler(Exception.class)
public ResponseEntity<String> handleException(Exception ex) {
// 可以记录日志
return new ResponseEntity<>("发生异常: " + ex.getMessage(), HttpStatus.INTERNAL_SERVER_ERROR);
}
}
在这个例子中,我们定义了一个GlobalExceptionHandler
类,并使用@ControllerAdvice
注解标记。其中,@ExceptionHandler(Exception.class)
注解表示这个方法可以处理所有类型的异常。当应用中发生异常时,这个方法会被调用,并返回一个包含错误信息的ResponseEntity
对象。
直接使用HttpServletResponse返回json
在@ControllerAdvice
使用中,除了返回ResponseEntity
对象外,还可以通过HttpServletResponse
直接写入响应来返回JSON格式的错误信息。这种方式在需要更灵活控制响应内容和格式时非常有用,特别是在构建RESTful API时。
以下是一个示例,展示了如何在拦截到异常后通过HttpServletResponse
返回JSON格式的错误信息:
@ControllerAdvice
public class GlobalExceptionHandlerWithResponse {
@ExceptionHandler(Exception.class)
public void handleException(Exception ex, HttpServletResponse response) throws IOException {
response.setStatus(HttpStatus.INTERNAL_SERVER_ERROR.value());
response.setContentType("application/json");
// 使用Jackson或其他JSON处理库将错误信息转换为JSON格式
String jsonResponse = "{\"error\": \"" + ex.getMessage() + "\"}";
response.getWriter().write(jsonResponse);
}
}
在这个例子中,handleException
方法除了接收异常对象Exception
作为参数外,还接收了HttpServletResponse
对象。这允许我们直接向响应输出流写入内容。首先,我们通过setStatus
方法设置了HTTP状态码,然后通过setContentType
方法指定了响应的内容类型为application/json
。最后,我们构造了一个包含错误信息的JSON字符串,并通过response.getWriter().write(jsonResponse)
将其写入响应体中。
返回自定义通用返回类
handleException
方法可以直接返回一个自定义的通用返回对象,Spring框架会自动将该对象序列化为JSON格式返回给客户端。这是通过Spring MVC的消息转换器(如MappingJackson2HttpMessageConverter
)自动完成的,前提是你已经在项目中加入了JSON处理库(如Jackson)。
以下是一个示例,展示如何返回一个自定义的通用返回对象,并让Spring自动将其序列化为JSON格式:
@ControllerAdvice
public class GlobalExceptionHandler {
@ExceptionHandler(Exception.class)
@ResponseBody // 确保返回的对象被序列化为JSON
public ErrorResponse handleException(Exception ex) {
return new ErrorResponse(HttpStatus.INTERNAL_SERVER_ERROR.value(), ex.getMessage());
}
// 自定义的通用返回对象类
public static class ErrorResponse {
private int status;
private String message;
public ErrorResponse(int status, String message) {
this.status = status;
this.message = message;
}
// getter和setter省略
}
}
在这个例子中,handleException
方法返回了一个ErrorResponse
对象,该对象包含了错误状态码和错误消息。我们通过@ResponseBody
注解告知Spring,这个方法的返回值应该被视为响应体,并且需要被转换为JSON格式。
Spring MVC在内部使用配置的HTTP消息转换器来将返回对象序列化为JSON。如果你的项目依赖于Jackson库,Spring会使用MappingJackson2HttpMessageConverter
自动将ErrorResponse
对象转换为JSON格式,然后发送给客户端。
这种方式简化了异常处理的代码,同时也提供了返回结构化错误信息的灵活性,非常适合构建RESTful API和微服务架构。
自定义参数绑定
当你需要对进入控制器的请求参数进行自定义处理时,比如日期格式的转换,@ControllerAdvice
配合@InitBinder
可以全局定义转换规则,避免在每个控制器中重复定义。
使用@InitBinder
自定义日期格式绑定的示例:
@ControllerAdvice
public class GlobalBindingInitializer {
@InitBinder
public void initBinder(WebDataBinder binder) {
SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd");
binder.registerCustomEditor(Date.class, new CustomDateEditor(dateFormat, true));
}
}
在这个例子中,我们通过@InitBinder
注解标记的方法initBinder
中,注册了一个自定义的编辑器CustomDateEditor
,用于将字符串转换为Date
类型。这意味着,当Spring MVC接收到日期格式的字符串参数时,会自动使用这个编辑器进行转换。
添加全局数据模型
有时候,你可能希望某些数据对所有的控制器都是可用的,比如用户登录信息、全局配置信息等。@ControllerAdvice
允许你通过@ModelAttribute
注解轻松实现这一点。
使用@ModelAttribute
添加全局数据模型的例子:
@ControllerAdvice
public class GlobalModelAttributes {
@ModelAttribute("globalMessage")
public String globalMessage() {
return "这是一个全局消息";
}
}
在这个例子中,我们使用@ModelAttribute
注解定义了一个方法globalMessage
,该方法返回一个字符串。这个字符串会被添加到所有控制器方法的模型中,可以在视图中使用${globalMessage}
来访问它。