Spring Boot 系列（八）@ControllerAdvice 攔截異常并統(tǒng)一處理

Spring Boot 系列（八）@ControllerAdvice 攔截異常并統(tǒng)一處理

在spring 3.2中，新增了@ControllerAdvice 注解，可以用于定義@ExceptionHandler、@InitBinder、@ModelAttribute，并應(yīng)用到所有@RequestMapping中。參考：@ControllerAdvice 文檔

一、介紹

創(chuàng)建 MyControllerAdvice，并添加 @ControllerAdvice注解。

package com.sam.demo.controller; import org.springframework.ui.Model; import org.springframework.web.bind.WebDataBinder; import org.springframework.web.bind.annotation.*; import java.util.HashMap; import java.util.Map; /** * controller 增強(qiáng)器 * @author sam * @since 2017/7/17 */ @ControllerAdvice public class MyControllerAdvice { /** * 應(yīng)用到所有@RequestMapping注解方法，在其執(zhí)行之前初始化數(shù)據(jù)綁定器 * @param binder */ @InitBinder public void initBinder(WebDataBinder binder) {} /** * 把值綁定到Model中，使全局@RequestMapping可以獲取到該值 * @param model */ @ModelAttribute public void addAttributes(Model model) { model.addAttribute("author", "Magical Sam"); } /** * 全局異常捕捉處理 * @param ex * @return */ @ResponseBody @ExceptionHandler(value = Exception.class) public Map errorHandler(Exception ex) { Map map = new HashMap(); map.put("code", 100); map.put("msg", ex.getMessage()); return map; } }

啟動(dòng)應(yīng)用后，被 @ExceptionHandler、@InitBinder、@ModelAttribute 注解的方法，都會(huì)作用在 被 @RequestMapping 注解的方法上。

@ModelAttribute：在Model上設(shè)置的值，對(duì)于所有被 @RequestMapping 注解的方法中，都可以通過(guò) ModelMap 獲取，如下：

@RequestMapping("/home") public String home(ModelMap modelMap) { System.out.println(modelMap.get("author")); } //或者 通過(guò)@ModelAttribute獲取 @RequestMapping("/home") public String home(@ModelAttribute("author") String author) { System.out.println(author); }

@ExceptionHandler 攔截了異常，我們可以通過(guò)該注解實(shí)現(xiàn)自定義異常處理。其中，@ExceptionHandler 配置的 value 指定需要攔截的異常類型，上面攔截了 Exception.class 這種異常。

二、自定義異常處理（全局異常處理）

spring boot 默認(rèn)情況下會(huì)映射到 /error 進(jìn)行異常處理，但是提示并不十分友好，下面自定義異常處理，提供友好展示。
1、編寫(xiě)自定義異常類：

package com.sam.demo.custom; /** * @author sam * @since 2017/7/17 */ public class MyException extends RuntimeException { public MyException(String code, String msg) { this.code = code; this.msg = msg; } private String code; private String msg; // getter & setter }

注：spring 對(duì)于 RuntimeException 異常才會(huì)進(jìn)行事務(wù)回滾。

2、編寫(xiě)全局異常處理類

創(chuàng)建 MyControllerAdvice.java，如下：

package com.sam.demo.controller; import org.springframework.ui.Model; import org.springframework.web.bind.WebDataBinder; import org.springframework.web.bind.annotation.*; import java.util.HashMap; import java.util.Map; /** * controller 增強(qiáng)器 * * @author sam * @since 2017/7/17 */ @ControllerAdvice public class MyControllerAdvice { /** * 全局異常捕捉處理 * @param ex * @return */ @ResponseBody @ExceptionHandler(value = Exception.class) public Map errorHandler(Exception ex) { Map map = new HashMap(); map.put("code", 100); map.put("msg", ex.getMessage()); return map; } /** * 攔截捕捉自定義異常 MyException.class * @param ex * @return */ @ResponseBody @ExceptionHandler(value = MyException.class) public Map myErrorHandler(MyException ex) { Map map = new HashMap(); map.put("code", ex.getCode()); map.put("msg", ex.getMsg()); return map; } }