文章目錄

• Pre
• 演進(jìn)過(guò)程
• 版本V1
• 版本2
• • Step1 約定統(tǒng)一返回格式
  • Step2 開發(fā)統(tǒng)一返回對(duì)象
  • Step3 約定接口狀態(tài)碼
  • Step4 驗(yàn)證
  • Step5 完善全局異常處理 @RestControllerAdvice + @ExceptionHandler
  • 全局異常處理器的必要行
• 版本3 （ResponseBodyAdvice）
• • Step1 自定義ResponseBodyAdvice接口實(shí)現(xiàn)類
  • Step2 全局異常整合到返回的標(biāo)準(zhǔn)格式
• 源碼

---

Pre

Spring Boot2.x-11 使用@ControllerAdvice和@ExceptionHandler實(shí)現(xiàn)自定義全局異常

---

演進(jìn)過(guò)程

我們搞個(gè)boot工程 ，來(lái)看下為什么以及如何來(lái)實(shí)現(xiàn)統(tǒng)一格式封裝及高階全局異常處理

<dependencies><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-web</artifactId></dependency><dependency><groupId>org.projectlombok</groupId><artifactId>lombok</artifactId></dependency><dependency><groupId>jakarta.validation</groupId><artifactId>jakarta.validation-api</artifactId></dependency></dependencies>

---

版本V1

@RestController // 返回JSON @RequestMapping("/v1") public class ArtisanV1Controller {/*** 返回字符串** @return*/@GetMapping("/getString")public String getStr() {return "OOOOOOOK";}/*** 返回自定義對(duì)象** @return*/@GetMapping("/getArtisan")public Artisan getArt() {Artisan artisan = new Artisan();artisan.setJob("ArtisanJob");artisan.setAge(18);return artisan;}/*** 接口異常** @return*/@GetMapping("/getMockError")public int getMockError() {int i = 1 / 0;return i;}}

分別測(cè)試下

這混亂的格式， 前端同學(xué)怎么想

---

版本2

Step1 約定統(tǒng)一返回格式

一個(gè)合格的標(biāo)準(zhǔn)的返回格式至少包含3部分：

• status 狀態(tài)值

  由后端統(tǒng)一定義各種返回結(jié)果的狀態(tài)碼

• message 描述

  本次接口調(diào)用的結(jié)果描述

• data 數(shù)據(jù)

  本次接口返回的數(shù)據(jù)

如果需要可以加入其他節(jié)點(diǎn)，比如在返回對(duì)象中添加了接口調(diào)用時(shí)間 （timestamp: 接口調(diào)用時(shí)間）

Step2 開發(fā)統(tǒng)一返回對(duì)象

package com.artisan.resp;import lombok.Data;/*** @author 小工匠* @version 1.0* @description: 公共結(jié)果 * @mark: show me the code , change the world*/ @Data public class ResponseData<T> {/*** 結(jié)果狀態(tài) ,具體狀態(tài)碼參見ResponseCode*/private int status;/*** 響應(yīng)消息**/private String message;/*** 響應(yīng)數(shù)據(jù)**/private T data;/*** 接口請(qǐng)求時(shí)間**/private long timestamp;/*** 初始化，增加接口請(qǐng)求事件*/public ResponseData() {this.timestamp = System.currentTimeMillis();}/*** 成功** @param <T>* @return*/public static <T> ResponseData<T> success() {ResponseData<T> resultData = new ResponseData<>();resultData.setStatus(ResponseCode.RC100.getCode());resultData.setMessage(ResponseCode.RC100.getMessage());return resultData;}/*** 成功** @param message* @param <T>* @return*/public static <T> ResponseData<T> success(String message) {ResponseData<T> resultData = new ResponseData<>();resultData.setStatus(ResponseCode.RC100.getCode());resultData.setMessage(message);return resultData;}/*** 成功** @param data* @param <T>* @return*/public static <T> ResponseData<T> success(T data) {ResponseData<T> resultData = new ResponseData<>();resultData.setStatus(ResponseCode.RC100.getCode());resultData.setMessage(ResponseCode.RC100.getMessage());resultData.setData(data);return resultData;}/*** 失敗** @param message* @param <T>* @return*/public static <T> ResponseData<T> fail(String message) {ResponseData<T> resultData = new ResponseData<>();resultData.setStatus(ResponseCode.RC999.getCode());resultData.setMessage(message);return resultData;}/*** 失敗** @param code* @param message* @param <T>* @return*/public static <T> ResponseData<T> fail(int code, String message) {ResponseData<T> resultData = new ResponseData<>();resultData.setStatus(code);resultData.setMessage(message);return resultData;}/*** 失敗** @param <T>* @return*/public static <T> ResponseData<T> fail() {ResponseData<T> resultData = new ResponseData<>();resultData.setStatus(ResponseCode.RC999.getCode());resultData.setMessage(ResponseCode.RC999.getMessage());return resultData;}}

Step3 約定接口狀態(tài)碼

package com.artisan.resp;import lombok.Getter;/*** @author 小工匠* @version 1.0* @description: 狀態(tài)碼集合* @mark: show me the code , change the world*/ public enum ResponseCode {/*** 操作成功**/RC100(100, "操作成功"),/*** 操作失敗**/RC999(999, "操作失敗"),/*** access_denied**/RC403(403, "無(wú)訪問(wèn)權(quán)限,請(qǐng)聯(lián)系管理員授予權(quán)限"),/*** access_denied**/RC401(401, "匿名用戶訪問(wèn)無(wú)權(quán)限資源時(shí)的異常"),/*** 服務(wù)異常**/RC500(500, "系統(tǒng)異常，請(qǐng)稍后重試"),ILLEGAL_ARGUMENT(3001, "非法參數(shù)"),INVALID_TOKEN(2001, "訪問(wèn)令牌不合法"),ACCESS_DENIED(2003, "沒(méi)有權(quán)限訪問(wèn)該資源"),CLIENT_AUTHENTICATION_FAILED(1001, "客戶端認(rèn)證失敗"),USERNAME_OR_PASSWORD_NOTMATCH(1002, "用戶名或密碼錯(cuò)誤"); /*** 自定義狀態(tài)碼**/@Getterprivate final int code;/*** 自定義描述**/@Getterprivate final String message;ResponseCode(int code, String message) {this.code = code;this.message = message;}}

Step4 驗(yàn)證

package com.artisan.controller;import com.artisan.entity.Artisan; import com.artisan.resp.ResponseData; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RestController;/*** @author 小工匠* @version 1.0* @description: 版本2* @mark: show me the code , change the world*/@RestController @RequestMapping("/v2") public class ArtisanV2Controller {@GetMapping("/getString")public ResponseData<String> getStr() {return ResponseData.success("OOOOOOK");}@GetMapping("/getArtisan")public ResponseData<Artisan> getArt() {Artisan artisan = new Artisan();artisan.setJob("CodeMonkey");artisan.setAge(18);return ResponseData.success(artisan);}@GetMapping("/getMockError")public ResponseData<Integer> getMockError() {int i = 1 / 0;return ResponseData.success(i);}}

好像部分實(shí)現(xiàn)了統(tǒng)一格式返回，確實(shí)也是有很多項(xiàng)目在Controller層通過(guò)ResponseData.success()對(duì)返回結(jié)果進(jìn)行包裝后返回給前端。

但是這個(gè)拋異常的這么玩還是不行呀？ ------------------------> 全局異常處理

Step5 完善全局異常處理 @RestControllerAdvice + @ExceptionHandler

package com.artisan.resp;import com.artisan.exception.BaseException; import lombok.extern.slf4j.Slf4j; import org.springframework.http.HttpStatus; import org.springframework.http.ResponseEntity; import org.springframework.validation.BindException; import org.springframework.validation.ObjectError; import org.springframework.web.bind.MethodArgumentNotValidException; import org.springframework.web.bind.annotation.ExceptionHandler; import org.springframework.web.bind.annotation.ResponseStatus; import org.springframework.web.bind.annotation.RestControllerAdvice;import javax.validation.ConstraintViolation; import javax.validation.ConstraintViolationException; import javax.validation.ValidationException; import java.util.stream.Collectors;/*** @author 小工匠* @version 1.0* @description: 全局異常處理* @mark: show me the code , change the world*/ @Slf4j @RestControllerAdvice public class GlobalExceptionHandler {/*** 默認(rèn)全局異常處理。** @param e e* @return ResponseData*/@ExceptionHandler(Exception.class)@ResponseStatus(HttpStatus.INTERNAL_SERVER_ERROR)public ResponseData<String> exception(Exception e) {log.error("兜底異常信息 ex={}", e.getMessage());return ResponseData.fail(ResponseCode.RC500.getCode(), e.getMessage());}/*** Assert異常*/@ExceptionHandler({IllegalArgumentException.class, IllegalStateException.class})@ResponseStatus(HttpStatus.INTERNAL_SERVER_ERROR)public ResponseData<String> exception(IllegalArgumentException e) {return ResponseData.fail(ResponseCode.ILLEGAL_ARGUMENT.getCode(), e.getMessage());}/*** 抓取自定義異常 BaseException*/@ExceptionHandler(BaseException.class)@ResponseStatus(HttpStatus.INTERNAL_SERVER_ERROR)public ResponseData<String> exception(BaseException e) {return ResponseData.fail(e.getErrorCode(), e.getMessage());}}

• @ExceptionHandler,統(tǒng)一處理某一類異常， 減少代碼重復(fù)率和復(fù)雜度，比如要捕獲自定義異常可以@ExceptionHandler(BusinessException.class)

• @ResponseStatus指定客戶端收到的http狀態(tài)碼

重新驗(yàn)證下

---

全局異常處理器的必要行

避免try…catch，由全局異常處理器統(tǒng)一捕獲

自定義異常，只能通過(guò)全局異常處理器來(lái)處理

Validator參數(shù)校驗(yàn)器的時(shí)候，參數(shù)校驗(yàn)不通過(guò)會(huì)拋出異常，無(wú)法用try…catch捕獲，只能使用全局異常處理器。

…

。

---

版本3 （ResponseBodyAdvice）

V2版本有缺陷么？

我們不難發(fā)現(xiàn)每寫一個(gè)接口都需要調(diào)用ResponseData.success()對(duì)結(jié)果進(jìn)行包裝 ，程序猿懶啊, 能不寫嗎

---

Step1 自定義ResponseBodyAdvice接口實(shí)現(xiàn)類

ResponseBodyAdvice的作用一般是用于攔截Controller方法的返回值，統(tǒng)一處理返回值/響應(yīng)體， 加解密，簽名等

package com.artisan.resp.v3;import com.artisan.resp.ResponseData; import com.fasterxml.jackson.databind.ObjectMapper; import lombok.SneakyThrows; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.core.MethodParameter; import org.springframework.http.MediaType; import org.springframework.http.converter.HttpMessageConverter; import org.springframework.http.server.ServerHttpRequest; import org.springframework.http.server.ServerHttpResponse; import org.springframework.web.bind.annotation.RestControllerAdvice; import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;/*** @author 小工匠* @version 1.0* @description: 攔截Controller方法的返回值，統(tǒng)一處理返回值/響應(yīng)體* @mark: show me the code , change the world*/@RestControllerAdvice public class CustomResponseAdvice implements ResponseBodyAdvice<Object> {@Autowiredprivate ObjectMapper objectMapper;@Overridepublic boolean supports(MethodParameter methodParameter, Class<? extends HttpMessageConverter<?>> aClass) {return true;}@SneakyThrows@Overridepublic Object beforeBodyWrite(Object o, MethodParameter methodParameter, MediaType mediaType, Class<? extends HttpMessageConverter<?>> aClass, ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse) {// 處理String類型 if (o instanceof String) {return objectMapper.writeValueAsString(ResponseData.success(o));}// 若是統(tǒng)一返回類型，則不用再此封裝 // if (o instanceof ResponseData) { // return o; // }return ResponseData.success(o);} }

• @RestControllerAdvice，RestController的增強(qiáng)類，可用于實(shí)現(xiàn)全局異常處理器

• 有一個(gè)地方對(duì)String做了特殊處理，因?yàn)槿绻鸆ontroller直接返回String ，SpringBoot是直接返回，所以我們需要手動(dòng)轉(zhuǎn)換成json

---

接入@RestControllerAdvice后， Controller就正常寫就可以了，不用統(tǒng)一格式去包裝了，如下

@RestController @RequestMapping("/v3") public class ArtisanV3Controller {@GetMapping("/getString")public String getStr() {return "OOOOOOK";}@GetMapping("/getArtisan")public Artisan getArt() {Artisan artisan = new Artisan();artisan.setJob("CodeMonkey");artisan.setAge(18);return artisan;}@GetMapping("/getMockError")public int getMockError() {int i = 1 / 0;return i;}}

測(cè)試下

看到問(wèn)題了吧，當(dāng)我們同時(shí)啟用統(tǒng)一標(biāo)準(zhǔn)格式封裝功能ResponseAdvice和RestExceptionHandler全局異常處理器時(shí)，統(tǒng)一格式增強(qiáng)功能會(huì)給返回的異常結(jié)果再次封裝，所以跟前端的接口響應(yīng)又迷糊了

---

Step2 全局異常整合到返回的標(biāo)準(zhǔn)格式

因?yàn)槿之惓Ｌ幚砥饕呀?jīng)幫我們封裝好了標(biāo)準(zhǔn)格式，我們只需要直接返回給客戶端即可。

// 若是統(tǒng)一返回類型，則不用再此封裝if (o instanceof ResponseData) {return o;}

如果返回的結(jié)果是ResponseData對(duì)象，直接返回即可。

重新測(cè)試 ，

---

源碼

https://github.com/yangshangwei/boot2

總結(jié)

以上是生活随笔為你收集整理的SpringBoot - 统一格式封装及高阶全局异常处理的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。