快捷導(dǎo)航

SpringMVC @ControllerAdvice使用場(chǎng)景

更新時(shí)間：2019年11月01日 09:06:18 作者：江南一點(diǎn)雨

這篇文章主要介紹了SpringMVC @ControllerAdvice使用場(chǎng)景,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下

@ControllerAdvice ，很多初學(xué)者可能都沒(méi)有聽(tīng)說(shuō)過(guò)這個(gè)注解，實(shí)際上，這是一個(gè)非常有用的注解，顧名思義，這是一個(gè)增強(qiáng)的 Controller。使用這個(gè) Controller ，可以實(shí)現(xiàn)三個(gè)方面的功能：

全局異常處理
全局?jǐn)?shù)據(jù)綁定
全局?jǐn)?shù)據(jù)預(yù)處理

靈活使用這三個(gè)功能，可以幫助我們簡(jiǎn)化很多工作，需要注意的是，這是 SpringMVC 提供的功能，在 Spring Boot 中可以直接使用，下面分別來(lái)看。

全局異常處理

使用 @ControllerAdvice 實(shí)現(xiàn)全局異常處理，只需要定義類(lèi)，添加該注解即可定義方式如下：

@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;
  }
}

在該類(lèi)中，可以定義多個(gè)方法，不同的方法處理不同的異常，例如專門(mén)處理空指針的方法、專門(mén)處理數(shù)組越界的方法…，也可以直接向上面代碼一樣，在一個(gè)方法中處理所有的異常信息。

@ExceptionHandler 注解用來(lái)指明異常的處理類(lèi)型，即如果這里指定為 NullpointerException，則數(shù)組越界異常就不會(huì)進(jìn)到這個(gè)方法中來(lái)。

全局?jǐn)?shù)據(jù)綁定

全局?jǐn)?shù)據(jù)綁定功能可以用來(lái)做一些初始化的數(shù)據(jù)操作，我們可以將一些公共的數(shù)據(jù)定義在添加了 @ControllerAdvice 注解的類(lèi)中，這樣，在每一個(gè) Controller 的接口中，就都能夠訪問(wèn)導(dǎo)致這些數(shù)據(jù)。

使用步驟，首先定義全局?jǐn)?shù)據(jù)，如下：

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

使用 @ModelAttribute 注解標(biāo)記該方法的返回?cái)?shù)據(jù)是一個(gè)全局?jǐn)?shù)據(jù)，默認(rèn)情況下，這個(gè)全局?jǐn)?shù)據(jù)的 key 就是返回的變量名，value 就是方法返回值，當(dāng)然開(kāi)發(fā)者可以通過(guò) @ModelAttribute 注解的 name 屬性去重新指定 key。

定義完成后，在任何一個(gè)Controller 的接口中，都可以獲取到這里定義的數(shù)據(jù)：

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

全局?jǐn)?shù)據(jù)預(yù)處理

考慮我有兩個(gè)實(shí)體類(lèi)，Book 和 Author，分別定義如下：

public class Book {
  private String name;
  private Long price;
  //getter/setter
}
public class Author {
  private String name;
  private Integer age;
  //getter/setter
}

此時(shí)，如果我定義一個(gè)數(shù)據(jù)添加接口，如下：

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

這個(gè)時(shí)候，添加操作就會(huì)有問(wèn)題，因?yàn)閮蓚€(gè)實(shí)體類(lèi)都有一個(gè) name 屬性，從前端傳遞時(shí) ，無(wú)法區(qū)分。此時(shí)，通過(guò) @ControllerAdvice 的全局?jǐn)?shù)據(jù)預(yù)處理可以解決這個(gè)問(wèn)題

解決步驟如下:

1.給接口中的變量取別名

@PostMapping("/book")
public void addBook(@ModelAttribute("b") Book book, @ModelAttribute("a") Author author) {
  System.out.println(book);
  System.out.println(author);
}

2.進(jìn)行請(qǐng)求數(shù)據(jù)預(yù)處理

在 @ControllerAdvice 標(biāo)記的類(lèi)中添加如下代碼:

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

@InitBinder(“b”) 注解表示該方法用來(lái)處理和Book和相關(guān)的參數(shù),在方法中,給參數(shù)添加一個(gè) b 前綴,即請(qǐng)求參數(shù)要有b前綴.

3.發(fā)送請(qǐng)求

請(qǐng)求發(fā)送時(shí),通過(guò)給不同對(duì)象的參數(shù)添加不同的前綴,可以實(shí)現(xiàn)參數(shù)的區(qū)分.