增强的@Controller:@ControllerAdvice

一、@ControllerAdvice 增强的@Controller

@ControllerAdvice,可以实现三个方面的功能:

  1. 全局异常处理
  2. 全局数据绑定
  3. 全局数据预处理

灵活使用这三个功能,可以简化很多工作。需要注意的是,这是 SpringMVC 提供的功能,在 Spring Boot 中可以直接使用,下面分别来看。

二、全局异常处理

使用 @ControllerAdvice 实现全局异常处理。只需要定义类,添加该注解即可定义,方式如下:

@ControllerAdvice
public class MyGlobalExceptionHandler {
  @ExceptionHandler(Exception.class)
  public ModelAndView customException(Exception e) {
   ModelAndView mv = new ModelAndView();
    mv.addObject("message", e.getMessage());
    mv.setViewName("myerror");
   return mv;
  }
}

在该类中,可以定义多个方法,不同的方法处理不同的异常,例如专门处理空指针的方法、专门处理数组越界的方法等等。也可以直接像上面代码一样,在一个方法中处理所有的异常信息。@ExceptionHandler 用来指明异常的处理类型,如果这里指定为 NullpointerException,则数组越界异常就不会进到这个方法中来。

三、全局数据绑定

全局数据绑定功能可以用来做一些初始化的数据操作,可以将一些公共的数据定义在添加了 @ControllerAdvice 的类中,这样在每一个 Controller 的接口中,就都能够访问到这些数据。使用步骤,首先定义全局数据,如下:

@ControllerAdvice
public class MyGlobalExceptionHandler {
  @ModelAttribute(name = "md")
  public Map mydata() {
   HashMap map = new HashMap<>();
    map.put("age", 99);
    map.put("gender", "男");
   return map;
  }
}

使用 @ModelAttribute 标记该方法的返回数据是一个全局数据。默认情况下,这个全局数据的 key 就是返回的变量名,value 就是方法返回值。当然可以通过 @ModelAttribute 的 name 属性去重新指定 key。定义完成后,在任何一个Controller 的接口中,都可以获取到这里定义的数据:

@RestController
public class HelloController {
  @GetMapping("/hello")
  public String hello(Model model) {
   Map map = model.asMap();
   System.out.println(map);
   int i = 1 / 0;
   return "hello controller advice";
  }
}

四、全局数据预处理

有两个实体类 Book 和 Author,分别定义如下:

public class Book {
 private String name;
 private Long price;
 public String getName() {
  return name;
 }
 public void setName(String name) {
  this.name = name;
 }
 public Long getPrice() {
  return price;
 }
 public void setPrice(Long price) {
   this.price = price;
 }
}
public class Author {
 private String name;
 private Integer age;
 public String getName() {
  return name;
 }
 public void setName(String name) {
  this.name = name;
 }
 public Integer getAge() {
  return age;
 }
 public void setAge(Integer age) {
  this.age = age;
 }
}

此时,定义一个数据添加接口@PostMapping("/addBook"),如下:

public void addBook(Book book, Author author) {
     System.out.println(book);
     System.out.println(author);
}

这个时候,添加操作就会有问题,因为两个实体类都有一个 name 属性,从前端传递时无法区分。此时,通过 @ControllerAdvice 的全局数据预处理可以解决这个问题。

解决步骤如下:

  1. 给接口中的变量取别名:
@PostMapping("/book")
public void addBook(@ModelAttribute("b") Book book, @ModelAttribute("a") Author author) {
 System.out.println(book);
 System.out.println(author);
}
  1. 进行请求数据预处理。在 @ControllerAdvice 标记的类中添加如下代码:
@InitBinder("b")
public void b(WebDataBinder binder) {
 binder.setFieldDefaultPrefix("b.");
}

@InitBinder("a")
public void a(WebDataBinder binder) {
 binder.setFieldDefaultPrefix("a.");
}

@InitBinder("b")表示该方法用来处理和Book相关的参数。在方法中,给参数添加一个 b 前缀,即请求参数要有b前缀。

  1. 发送请求。请求发送时,通过给不同对象的参数添加不同的前缀,可以实现参数的区分。

你可能感兴趣的:(增强的@Controller:@ControllerAdvice)