SpringBoot優雅的全域性異常處理

虛無境發表於2019-05-27

前言

本篇文章主要介紹的是SpringBoot專案進行全域性異常的處理。

SpringBoot全域性異常準備

說明:如果想直接獲取工程那麼可以直接跳到底部,通過連結下載工程程式碼。

開發準備

環境要求
JDK:1.8
SpringBoot:1.5.17.RELEASE

首先還是Maven的相關依賴:


  <properties>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
        <java.version>1.8</java.version>
        <maven.compiler.source>1.8</maven.compiler.source>
        <maven.compiler.target>1.8</maven.compiler.target>
    </properties>
    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>1.5.17.RELEASE</version>
        <relativePath />
    </parent>
    <dependencies>
        <!-- Spring Boot Web 依賴 核心 -->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <!-- Spring Boot Test 依賴 -->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>

        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>fastjson</artifactId>
            <version>1.2.41</version>
        </dependency>
    </dependencies>
    

配置檔案這塊基本不需要更改,全域性異常的處理只需在程式碼中實現即可。

程式碼編寫

SpringBoot的專案已經對有一定的異常處理了,但是對於我們開發者而言可能就不太合適了,因此我們需要對這些異常進行統一的捕獲並處理。SpringBoot中有一個ControllerAdvice的註解,使用該註解表示開啟了全域性異常的捕獲,我們只需在自定義一個方法使用ExceptionHandler註解然後定義捕獲異常的型別即可對這些捕獲的異常進行統一的處理。

我們根據下面的這個示例來看該註解是如何使用吧。

示例程式碼:


@ControllerAdvice
public class MyExceptionHandler {

    @ExceptionHandler(value =Exception.class)
    public String exceptionHandler(Exception e){
        System.out.println("未知異常!原因是:"+e);
        return e.getMessage();
    }
}

上述的示例中,我們對捕獲的異常進行簡單的二次處理,返回異常的資訊,雖然這種能夠讓我們知道異常的原因,但是在很多的情況下來說,可能還是不夠人性化,不符合我們的要求。
那麼我們這裡可以通過自定義的異常類以及列舉類來實現我們想要的那種資料吧。

自定義基礎介面類

首先定義一個基礎的介面類,自定義的錯誤描述列舉類需實現該介面。
程式碼如下:


public interface BaseErrorInfoInterface {
    /** 錯誤碼*/
     String getResultCode();
    
    /** 錯誤描述*/
     String getResultMsg();
}

自定義列舉類

然後我們這裡在自定義一個列舉類,並實現該介面。
程式碼如下:


public enum CommonEnum implements BaseErrorInfoInterface {
    // 資料操作錯誤定義
    SUCCESS("200", "成功!"), 
    BODY_NOT_MATCH("400","請求的資料格式不符!"),
    SIGNATURE_NOT_MATCH("401","請求的數字簽名不匹配!"),
    NOT_FOUND("404", "未找到該資源!"), 
    INTERNAL_SERVER_ERROR("500", "伺服器內部錯誤!"),
    SERVER_BUSY("503","伺服器正忙,請稍後再試!")
    ;

    /** 錯誤碼 */
    private String resultCode;

    /** 錯誤描述 */
    private String resultMsg;

    CommonEnum(String resultCode, String resultMsg) {
        this.resultCode = resultCode;
        this.resultMsg = resultMsg;
    }

    @Override
    public String getResultCode() {
        return resultCode;
    }

    @Override
    public String getResultMsg() {
        return resultMsg;
    }

}

自定義異常類

然後我們在來自定義一個異常類,用於處理我們發生的業務異常。
程式碼如下:


public class BizException extends RuntimeException {

    private static final long serialVersionUID = 1L;

    /**
     * 錯誤碼
     */
    protected String errorCode;
    /**
     * 錯誤資訊
     */
    protected String errorMsg;

    public BizException() {
        super();
    }

    public BizException(BaseErrorInfoInterface errorInfoInterface) {
        super(errorInfoInterface.getResultCode());
        this.errorCode = errorInfoInterface.getResultCode();
        this.errorMsg = errorInfoInterface.getResultMsg();
    }
    
    public BizException(BaseErrorInfoInterface errorInfoInterface, Throwable cause) {
        super(errorInfoInterface.getResultCode(), cause);
        this.errorCode = errorInfoInterface.getResultCode();
        this.errorMsg = errorInfoInterface.getResultMsg();
    }
    
    public BizException(String errorMsg) {
        super(errorMsg);
        this.errorMsg = errorMsg;
    }
    
    public BizException(String errorCode, String errorMsg) {
        super(errorCode);
        this.errorCode = errorCode;
        this.errorMsg = errorMsg;
    }

    public BizException(String errorCode, String errorMsg, Throwable cause) {
        super(errorCode, cause);
        this.errorCode = errorCode;
        this.errorMsg = errorMsg;
    }
    

    public String getErrorCode() {
        return errorCode;
    }

    public void setErrorCode(String errorCode) {
        this.errorCode = errorCode;
    }

    public String getErrorMsg() {
        return errorMsg;
    }

    public void setErrorMsg(String errorMsg) {
        this.errorMsg = errorMsg;
    }

    public String getMessage() {
        return errorMsg;
    }

    @Override
    public Throwable fillInStackTrace() {
        return this;
    }

}

自定義資料格式

順便這裡我們定義一下資料的傳輸格式。
程式碼如下:


public class ResultBody {
    /**
     * 響應程式碼
     */
    private String code;

    /**
     * 響應訊息
     */
    private String message;

    /**
     * 響應結果
     */
    private Object result;

    public ResultBody() {
    }

    public ResultBody(BaseErrorInfoInterface errorInfo) {
        this.code = errorInfo.getResultCode();
        this.message = errorInfo.getResultMsg();
    }

    public String getCode() {
        return code;
    }

    public void setCode(String code) {
        this.code = code;
    }

    public String getMessage() {
        return message;
    }

    public void setMessage(String message) {
        this.message = message;
    }

    public Object getResult() {
        return result;
    }

    public void setResult(Object result) {
        this.result = result;
    }

    /**
     * 成功
     * 
     * @return
     */
    public static ResultBody success() {
        return success(null);
    }

    /**
     * 成功
     * @param data
     * @return
     */
    public static ResultBody success(Object data) {
        ResultBody rb = new ResultBody();
        rb.setCode(CommonEnum.SUCCESS.getResultCode());
        rb.setMessage(CommonEnum.SUCCESS.getResultMsg());
        rb.setResult(data);
        return rb;
    }

    /**
     * 失敗
     */
    public static ResultBody error(BaseErrorInfoInterface errorInfo) {
        ResultBody rb = new ResultBody();
        rb.setCode(errorInfo.getResultCode());
        rb.setMessage(errorInfo.getResultMsg());
        rb.setResult(null);
        return rb;
    }

    /**
     * 失敗
     */
    public static ResultBody error(String code, String message) {
        ResultBody rb = new ResultBody();
        rb.setCode(code);
        rb.setMessage(message);
        rb.setResult(null);
        return rb;
    }

    /**
     * 失敗
     */
    public static ResultBody error( String message) {
        ResultBody rb = new ResultBody();
        rb.setCode("-1");
        rb.setMessage(message);
        rb.setResult(null);
        return rb;
    }

    @Override
    public String toString() {
        return JSONObject.toJSONString(this);
    }

}

自定義全域性異常處理類

最後我們在來編寫一個自定義全域性異常處理的類。
程式碼如下:


@ControllerAdvice
public class GlobalExceptionHandler {
    private static final Logger logger = LoggerFactory.getLogger(GlobalExceptionHandler.class);
    
    /**
     * 處理自定義的業務異常
     * @param req
     * @param e
     * @return
     */
    @ExceptionHandler(value = BizException.class)  
    @ResponseBody  
    public  ResultBody bizExceptionHandler(HttpServletRequest req, BizException e){
        logger.error("發生業務異常!原因是:{}",e.getErrorMsg());
        return ResultBody.error(e.getErrorCode(),e.getErrorMsg());
    }

    /**
     * 處理空指標的異常
     * @param req
     * @param e
     * @return
     */
    @ExceptionHandler(value =NullPointerException.class)
    @ResponseBody
    public ResultBody exceptionHandler(HttpServletRequest req, NullPointerException e){
        logger.error("發生空指標異常!原因是:",e);
        return ResultBody.error(CommonEnum.BODY_NOT_MATCH);
    }


    /**
        * 處理其他異常
     * @param req
     * @param e
     * @return
     */
    @ExceptionHandler(value =Exception.class)
    @ResponseBody
    public ResultBody exceptionHandler(HttpServletRequest req, Exception e){
        logger.error("未知異常!原因是:",e);
        return ResultBody.error(CommonEnum.INTERNAL_SERVER_ERROR);
    }
}

因為這裡我們只是用於做全域性異常處理的功能實現以及測試,所以這裡我們只需在新增一個實體類和一個控制層類即可。

實體類

又是萬能的使用者表 (^▽^)

程式碼如下:


public class User implements Serializable{
    private static final long serialVersionUID = 1L;
    /** 編號 */
     private int id;
     /** 姓名 */
     private String name;
     /** 年齡 */
     private int age;
     
     public User(){
     }

    public int getId() {
        return id;
    }
    
    public void setId(int id) {
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public int getAge() {
        return age;
    }

    public void setAge(int age) {
        this.age = age;
    }

    public String toString() {
        return JSONObject.toJSONString(this);
    }
}

Controller 控制層

控制層這邊也比較簡單,使用Restful風格實現的CRUD功能,不同的是這裡我故意弄出了一些異常,好讓這些異常被捕獲到然後處理。這些異常中,有自定義的異常丟擲,也有空指標的異常丟擲,當然也有不可預知的異常丟擲(這裡我用型別轉換異常代替),那麼我們在完成程式碼編寫之後,看看這些異常是否能夠被捕獲處理成功吧!

程式碼如下:


@RestController
@RequestMapping(value = "/api")
public class UserRestController {

    @PostMapping("/user")
    public boolean insert(@RequestBody User user) {
        System.out.println("開始新增...");
        //如果姓名為空就手動丟擲一個自定義的異常!
        if(user.getName()==null){
            throw  new BizException("-1","使用者姓名不能為空!");
        }
        return true;
    }

    @PutMapping("/user")
    public boolean update(@RequestBody User user) {
        System.out.println("開始更新...");
       //這裡故意造成一個空指標的異常,並且不進行處理
        String str=null;
        str.equals("111");
        return true;
    }

    @DeleteMapping("/user")
    public boolean delete(@RequestBody User user)  {
        System.out.println("開始刪除...");
        //這裡故意造成一個異常,並且不進行處理
        Integer.parseInt("abc123");
        return true;
    }

    @GetMapping("/user")
    public List<User> findByUser(User user) {
        System.out.println("開始查詢...");
        List<User> userList =new ArrayList<>();
        User user2=new User();
        user2.setId(1L);
        user2.setName("xuwujing");
        user2.setAge(18);
        userList.add(user2);
        return userList;
    }
    
}

App 入口

和普通的SpringBoot專案基本一樣。

程式碼如下:


@SpringBootApplication
public class App 
{
    public static void main( String[] args )
    {
        SpringApplication.run(App.class, args);
        System.out.println("程式正在執行...");
    }
}

功能測試

我們成功啟動該程式之後,使用Postman工具來進行介面測試。

首先進行查詢,檢視程式正常執行是否ok,使用GET 方式進行請求。

GET http://localhost:8181/api/user

返回引數為:

{"id":1,"name":"xuwujing","age":18}

示例圖:
在這裡插入圖片描述
可以看到程式正常返回,並沒有因自定義的全域性異常而影響。

然後我們再來測試下自定義的異常是否能夠被正確的捕獲並處理。

使用POST方式進行請求

POST http://localhost:8181/api/user

Body引數為:

{"id":1,"age":18}

返回引數為:

{"code":"-1","message":"使用者姓名不能為空!","result":null}

示例圖:
在這裡插入圖片描述
可以看出將我們丟擲的異常進行資料封裝,然後將異常返回出來。

然後我們再來測試下空指標異常是否能夠被正確的捕獲並處理。在自定義全域性異常中,我們除了定義空指標的異常處理,也定義最高階別之一的Exception異常,那麼這裡發生了空指標異常之後,它是回優先使用哪一個呢?這裡我們來測試下。

使用PUT方式進行請求。

PUT http://localhost:8181/api/user

Body引數為:

{"id":1,"age":18}

返回引數為:

{"code":"400","message":"請求的資料格式不符!","result":null}

示例圖:
SpringBoot優雅的全域性異常處理
我們可以看到這裡的的確是返回空指標的異常護理,可以得出全域性異常處理優先處理子類的異常。

那麼我們在來試試未指定其異常的處理,看該異常是否能夠被捕獲。

使用DELETE方式進行請求。

DELETE http://localhost:8181/api/user

Body引數為:

{"id":1}

返回引數為:

{"code":"500","message":"伺服器內部錯誤!","result":null}

在這裡插入圖片描述

這裡可以看到它使用了我們在自定義全域性異常處理類中的Exception異常處理的方法。
到這裡,測試就結束了。順便再說一下,自義定全域性異常處理除了可以處理上述的資料格式之外,也可以處理頁面的跳轉,只需在新增的異常方法的返回處理上填寫該跳轉的路徑並不使用ResponseBody 註解即可。 細心的同學也許發現了在GlobalExceptionHandler類中使用的是ControllerAdvice註解,而非RestControllerAdvice註解,如果是用的RestControllerAdvice註解,它會將資料自動轉換成JSON格式,這種於ControllerRestController類似,所以我們在使用全域性異常處理的之後可以進行靈活的選擇處理。

其它

關於SpringBoot優雅的全域性異常處理的文章就講解到這裡了,如有不妥,歡迎指正!

專案地址

SpringBoot全域性異常的處理專案工程地址:
https://github.com/xuwujing/springBoot-study/tree/master/springboot-exceptionHandler

SpringBoot整個集合的地址:
https://github.com/xuwujing/springBoot-study

SpringBoot整合系列的文章

音樂推薦

原創不易,如果感覺不錯,希望給個推薦!您的支援是我寫作的最大動力!
版權宣告:
作者:虛無境
部落格園出處:http://www.cnblogs.com/xuwujing
CSDN出處:http://blog.csdn.net/qazwsxpcm    
個人部落格出處:http://www.panchengming.com

相關文章