@ControllerAdvice + @ExceptionHandler 全域性處理 Controller 層異常
零、前言
對於與資料庫相關的 Spring MVC 專案,我們通常會把 事務 配置在 Service層,當資料庫操作失敗時讓 Service 層丟擲執行時異常,Spring 事物管理器就會進行回滾。
如此一來,我們的 Controller 層就不得不進行 try-catch Service 層的異常,否則會返回一些不友好的錯誤資訊到客戶端。但是,Controller 層每個方法體都寫一些模板化的 try-catch 的程式碼,很難看也難維護,特別是還需要對 Service 層的不同異常進行不同處理的時候。例如以下 Controller 方法程式碼(非常難看且冗餘):
/**
* 手動處理 Service 層異常和資料校驗異常的示例
* @param dog
* @param errors
* @return
*/
@PostMapping(value = "")
AppResponse add(@Validated(Add.class) @RequestBody Dog dog, Errors errors){
AppResponse resp = new AppResponse();
try {
// 資料校驗
BSUtil.controllerValidate(errors);
// 執行業務
Dog newDog = dogService.save(dog);
// 返回資料
resp.setData(newDog);
}catch (BusinessException e){
LOGGER.error(e.getMessage(), e);
resp.setFail(e.getMessage());
}catch (Exception e){
LOGGER.error(e.getMessage(), e);
resp.setFail("操作失敗!");
}
return resp;
}
本文講解使用 @ControllerAdvice + @ExceptionHandler 進行全域性的 Controller 層異常處理,只要設計得當,就再也不用在 Controller 層進行 try-catch 了!而且,@Validated 校驗器註解的異常,也可以一起處理,無需手動判斷繫結校驗結果 BindingResult/Errors 了!
一、優缺點
- 優點:將 Controller 層的異常和資料校驗的異常進行統一處理,減少模板程式碼,減少編碼量,提升擴充套件性和可維護性。
- 缺點:只能處理 Controller 層未捕獲(往外拋)的異常,對於 Interceptor(攔截器)層的異常,Spring 框架層的異常,就無能為力了。
二、基本使用示例
2.1 @ControllerAdvice 註解定義全域性異常處理類
@ControllerAdvice
public class GlobalExceptionHandler {
}
請確保此 GlobalExceptionHandler 類能被掃描到並裝載進 Spring 容器中。
2.2 @ExceptionHandler 註解宣告異常處理方法
@ControllerAdvice
public class GlobalExceptionHandler {
@ExceptionHandler(Exception.class)
@ResponseBody
String handleException(){
return "Exception Deal!";
}
}
方法 handleException() 就會處理所有 Controller 層丟擲的 Exception 及其子類的異常,這是最基本的用法了。
被 @ExceptionHandler 註解的方法的引數列表裡,還可以宣告很多種型別的引數,詳見文件。其原型如下:
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface ExceptionHandler {
/**
* Exceptions handled by the annotated method. If empty, will default to any
* exceptions listed in the method argument list.
*/
Class<? extends Throwable>[] value() default {};
}
如果 @ExceptionHandler 註解中未宣告要處理的異常型別,則預設為引數列表中的異常型別。所以上面的寫法,還可以寫成這樣:
@ControllerAdvice
public class GlobalExceptionHandler {
@ExceptionHandler()
@ResponseBody
String handleException(Exception e){
return "Exception Deal! " + e.getMessage();
}
}
引數物件就是 Controller 層丟擲的異常物件!
三、處理 Service 層上拋的業務異常
有時我們會在複雜的帶有資料庫事務的業務中,當出現不和預期的資料時,直接丟擲封裝後的業務級執行時異常,進行資料庫事務回滾,並希望該異常資訊能被返回顯示給使用者。
3.1 程式碼示例
封裝的業務異常類:
public class BusinessException extends RuntimeException {
public BusinessException(String message){
super(message);
}
}
Service 實現類:
@Service
public class DogService {
@Transactional
public Dog update(Dog dog){
// some database options
// 模擬狗狗新名字與其他狗狗的名字衝突
BSUtil.isTrue(false, "狗狗名字已經被使用了...");
// update database dog info
return dog;
}
}
其中輔助工具類 BSUtil
public static void isTrue(boolean expression, String error){
if(!expression) {
throw new BusinessException(error);
}
}
那麼,我們應該在 GlobalExceptionHandler 類中宣告該業務異常類,並進行相應的處理,然後返回給使用者。更貼近真實專案的程式碼,應該長這樣子:
/**
* Created by kinginblue on 2017/4/10.
* @ControllerAdvice + @ExceptionHandler 實現全域性的 Controller 層的異常處理
*/
@ControllerAdvice
public class GlobalExceptionHandler {
private static final Logger LOGGER = LoggerFactory.getLogger(GlobalExceptionHandler.class);
/**
* 處理所有不可知的異常
* @param e
* @return
*/
@ExceptionHandler(Exception.class)
@ResponseBody
AppResponse handleException(Exception e){
LOGGER.error(e.getMessage(), e);
AppResponse response = new AppResponse();
response.setFail("操作失敗!");
return response;
}
/**
* 處理所有業務異常
* @param e
* @return
*/
@ExceptionHandler(BusinessException.class)
@ResponseBody
AppResponse handleBusinessException(BusinessException e){
LOGGER.error(e.getMessage(), e);
AppResponse response = new AppResponse();
response.setFail(e.getMessage());
return response;
}
}
Controller 層的程式碼,就不需要進行異常處理了:
@RestController
@RequestMapping(value = "/dogs", consumes = {MediaType.APPLICATION_JSON_UTF8_VALUE})
public class DogController {
@Autowired
private DogService dogService;
@PatchMapping(value = "")
Dog update(@Validated(Update.class) @RequestBody Dog dog){
return dogService.update(dog);
}
}
3.2 程式碼說明
Logger 進行所有的異常日誌記錄。
@ExceptionHandler(BusinessException.class) 聲明瞭對 BusinessException 業務異常的處理,並獲取該業務異常中的錯誤提示,構造後返回給客戶端。
@ExceptionHandler(Exception.class) 聲明瞭對 Exception 異常的處理,起到兜底作用,不管 Controller 層執行的程式碼出現了什麼未能考慮到的異常,都返回統一的錯誤提示給客戶端。
備註:以上 GlobalExceptionHandler 只是返回 Json 給客戶端,更大的發揮空間需要按需求情況來做。
四、處理 Controller 資料繫結、資料校驗的異常
在 Dog 類中的欄位上的註解資料校驗規則:
@Data
public class Dog {
@NotNull(message = "{Dog.id.non}", groups = {Update.class})
@Min(value = 1, message = "{Dog.age.lt1}", groups = {Update.class})
private Long id;
@NotBlank(message = "{Dog.name.non}", groups = {Add.class, Update.class})
private String name;
@Min(value = 1, message = "{Dog.age.lt1}", groups = {Add.class, Update.class})
private Integer age;
}
說明:@NotNull、@Min、@NotBlank 這些註解的使用方法,不在本文範圍內。如果不熟悉,請查詢資料學習即可。
其他說明:
@Data 註解是 **Lombok** 專案的註解,可以使我們不用再在程式碼裡手動加 getter & setter。
在 Eclipse 和 IntelliJ IDEA 中使用時,還需要安裝相關外掛,這個步驟自行Google/Baidu 吧!
SpringMVC 中對於 RESTFUL 的 Json 介面來說,資料繫結和校驗,是這樣的:
/**
* 使用 GlobalExceptionHandler 全域性處理 Controller 層異常的示例
* @param dog
* @return
*/
@PatchMapping(value = "")
AppResponse update(@Validated(Update.class) @RequestBody Dog dog){
AppResponse resp = new AppResponse();
// 執行業務
Dog newDog = dogService.update(dog);
// 返回資料
resp.setData(newDog);
return resp;
}
使用 @Validated + @RequestBody
註解實現。
當使用了 @Validated + @RequestBody
註解但是沒有在繫結的資料物件後面跟上 Errors
型別的引數宣告的話,Spring MVC 框架會丟擲 MethodArgumentNotValidException
異常。
所以,在 GlobalExceptionHandler 中加上對 MethodArgumentNotValidException
異常的宣告和處理,就可以全域性處理資料校驗的異常了!加完後的程式碼如下:
/**
* Created by kinginblue on 2017/4/10.
* @ControllerAdvice + @ExceptionHandler 實現全域性的 Controller 層的異常處理
*/
@ControllerAdvice
public class GlobalExceptionHandler {
private static final Logger LOGGER = LoggerFactory.getLogger(GlobalExceptionHandler.class);
/**
* 處理所有不可知的異常
* @param e
* @return
*/
@ExceptionHandler(Exception.class)
@ResponseBody
AppResponse handleException(Exception e){
LOGGER.error(e.getMessage(), e);
AppResponse response = new AppResponse();
response.setFail("操作失敗!");
return response;
}
/**
* 處理所有業務異常
* @param e
* @return
*/
@ExceptionHandler(BusinessException.class)
@ResponseBody
AppResponse handleBusinessException(BusinessException e){
LOGGER.error(e.getMessage(), e);
AppResponse response = new AppResponse();
response.setFail(e.getMessage());
return response;
}
/**
* 處理所有介面資料驗證異常
* @param e
* @return
*/
@ExceptionHandler(MethodArgumentNotValidException.class)
@ResponseBody
AppResponse handleMethodArgumentNotValidException(MethodArgumentNotValidException e){
LOGGER.error(e.getMessage(), e);
AppResponse response = new AppResponse();
response.setFail(e.getBindingResult().getAllErrors().get(0).getDefaultMessage());
return response;
}
}
注意到了嗎,所有的 Controller 層的異常的日誌記錄,都是在這個 GlobalExceptionHandler 中進行記錄。也就是說,Controller 層也不需要在手動記錄錯誤日誌了。
五、總結
本文主要講 @ControllerAdvice + @ExceptionHandler 組合進行的 Controller 層上拋的異常全域性統一處理。
@ControllerAdvice 也還可以結合 @InitBinder、@ModelAttribute 等註解一起使用,應用在所有被 @RequestMapping 註解的方法上,詳見搜尋引擎。
六、附錄
本文示例程式碼已放到 Github。