在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注解的方法朵你，都會作用在 被@RequestMapping注解的方法上各聘。
@ModelAttribute：在Model上設(shè)置的值，對于所有被@RequestMapping注解的方法中撬呢，都可以通過 ModelMap 獲取伦吠，如下：
@RequestMapping("/home")
public String home(ModelMap modelMap) {
System.out.println(modelMap.get("author"));
}
//或者 通過@ModelAttribute獲取
@RequestMapping("/home")
public String home(@ModelAttribute("author") String author) {
System.out.println(author);
}
@ExceptionHandler 攔截了異常妆兑，我們可以通過該注解實(shí)現(xiàn)自定義異常處理魂拦。其中，@ExceptionHandler 配置的 value 指定需要攔截的異常類型搁嗓，上面攔截了 Exception.class 這種異常芯勘。
二、自定義異常處理（全局異常處理）
spring boot 默認(rèn)情況下會映射到 /error 進(jìn)行異常處理腺逛，但是提示并不十分友好荷愕，下面自定義異常處理，提供友好展示棍矛。
1安疗、編寫自定義異常類：
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 對于 RuntimeException 異常才會進(jìn)行事務(wù)回滾。
2够委、編寫全局異常處理類
創(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;
}
}
3、controller中拋出異常進(jìn)行測試茁帽。
@RequestMapping("/home")
public String home() throws Exception {
// throw new Exception("Sam 錯(cuò)誤");
throw new MyException("101", "Sam 錯(cuò)誤");
}
啟動(dòng)應(yīng)用玉罐，訪問：http://localhost:8080/home ，正常顯示以下json內(nèi)容潘拨，證明自定義異常已經(jīng)成功被攔截吊输。
{"msg":"Sam 錯(cuò)誤","code":"101"}
* 如果不需要返回json數(shù)據(jù)，而要渲染某個(gè)頁面模板返回給瀏覽器铁追，那么MyControllerAdvice中可以這么實(shí)現(xiàn)：
@ExceptionHandler(value = MyException.class)
public ModelAndView myErrorHandler(MyException ex) {
ModelAndView modelAndView = new ModelAndView();
modelAndView.setViewName("error");
modelAndView.addObject("code", ex.getCode());
modelAndView.addObject("msg", ex.getMsg());
return modelAndView;
}
在 templates 目錄下，添加 error.ftl（這里使用freemarker） 進(jìn)行渲染：
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>錯(cuò)誤頁面</title>
</head>
<body>
<h1>${code}</h1>
<h1>${msg}</h1>
</body>
</html>
重啟應(yīng)用琅束，http://localhost:8080/home 顯示自定的錯(cuò)誤頁面內(nèi)容。
補(bǔ)充：如果全部異常處理返回json狰闪，那么可以使用 @RestControllerAdvice 代替 @ControllerAdvice ，這樣在方法上就可以不需要添加 @ResponseBody埋泵。
版權(quán)聲明：本文為博主原創(chuàng)文章幔欧，轉(zhuǎn)載請注明出處。
