Global exception handling

Global exception handling

1、 Response result set encapsulation

public class ResponseResult<T> implements Serializable {
    
    private Integer code;
    private String msg;
    private T data;

    public ResponseResult() {
    
        this.code = AppHttpCodeEnum.SUCCESS.getCode();
        this.msg = AppHttpCodeEnum.SUCCESS.getMsg();
    }

    public ResponseResult(Integer code, T data) {
    
        this.code = code;
        this.data = data;
    }

    public ResponseResult(Integer code, String msg, T data) {
    
        this.code = code;
        this.msg = msg;
        this.data = data;
    }

    public ResponseResult(Integer code, String msg) {
    
        this.code = code;
        this.msg = msg;
    }

    public static ResponseResult errorResult(int code, String msg) {
    
        ResponseResult result = new ResponseResult();
        return result.error(code, msg);
    }

    public static ResponseResult okResult() {
    
        ResponseResult result = new ResponseResult();
        return result;
    }

    public static ResponseResult okResult(int code, String msg) {
    
        ResponseResult result = new ResponseResult();
        return result.ok(code, null, msg);
    }

    public static ResponseResult okResult(Object data) {
    
        ResponseResult result = setAppHttpCodeEnum(AppHttpCodeEnum.SUCCESS, AppHttpCodeEnum.SUCCESS.getMsg());
        if (data != null) {
    
            result.setData(data);
        }
        return result;
    }

    public static ResponseResult errorResult(AppHttpCodeEnum enums) {
    
        return setAppHttpCodeEnum(enums, enums.getMsg());
    }

    public static ResponseResult errorResult(AppHttpCodeEnum enums, String msg) {
    
        return setAppHttpCodeEnum(enums, msg);
    }

    public static ResponseResult setAppHttpCodeEnum(AppHttpCodeEnum enums) {
    
        return okResult(enums.getCode(), enums.getMsg());
    }

    private static ResponseResult setAppHttpCodeEnum(AppHttpCodeEnum enums, String msg) {
    
        return okResult(enums.getCode(), msg);
    }

    public ResponseResult<?> error(Integer code, String msg) {
    
        this.code = code;
        this.msg = msg;
        return this;
    }

    public ResponseResult<?> ok(Integer code, T data) {
    
        this.code = code;
        this.data = data;
        return this;
    }

    public ResponseResult<?> ok(Integer code, T data, String msg) {
    
        this.code = code;
        this.data = data;
        this.msg = msg;
        return this;
    }

    public ResponseResult<?> ok(T data) {
    
        this.data = data;
        return this;
    }

    public Integer getCode() {
    
        return code;
    }

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

    public String getMsg() {
    
        return msg;
    }

    public void setMsg(String msg) {
    
        this.msg = msg;
    }

    public T getData() {
    
        return data;
    }

    public void setData(T data) {
    
        this.data = data;
    }
}

2、 Exception response state enumeration class

public enum AppHttpCodeEnum {
    
    /** *  Successful operation  */
    SUCCESS(200," Successful operation "),
    SYSTEM_ERROR(504," Unknown system error "),
    LOGIN_ERROR(505," The divisor cannot be zero 0");
    final int code;
    final String msg;

    AppHttpCodeEnum(int code, String errorMessage){
    
        this.code = code;
        this.msg = errorMessage;
    }

    public int getCode() {
    
        return code;
    }

    public String getMsg() {
    
        return msg;
    }
}

3、 Custom global exception class

public class SystemException extends RuntimeException{
    

    private int code;

    private String msg;

    public int getCode() {
    
        return code;
    }

    public String getMsg() {
    
        return msg;
    }

    public SystemException(AppHttpCodeEnum httpCodeEnum) {
    
        super(httpCodeEnum.getMsg());
        this.code = httpCodeEnum.getCode();
        this.msg = httpCodeEnum.getMsg();
    }

}

4、 Configure global exception handling

/** *  Global exception handling  * @author YLi_Jing */
@RestControllerAdvice
public class GlobalExceptionHandler {
    

    @ExceptionHandler(SystemException.class)
    public ResponseResult systemExceptionHandler(SystemException e){
    
        // Get the prompt information from the exception object to encapsulate the return 
        return ResponseResult.errorResult(e.getCode(),e.getMsg());
    }


    @ExceptionHandler(Exception.class)
    public ResponseResult exceptionHandler(Exception e){
    
        // Get the prompt information from the exception object to encapsulate the return 
        return ResponseResult.errorResult(AppHttpCodeEnum.SYSTEM_ERROR.getCode(),e.getMessage());
    }
}