Spring boot 优雅实现全局自定义异常

一、前言：

　　SpringBoot的项目已经对有一定的异常处理了，但是对于我们开发者而言可能就不太合适了，因此我们需要对这些异常进行统一的捕获并处理。SpringBoot中有一个ControllerAdvice的注解，使用该注解表示开启了全局异常的捕获，我们只需在自定义一个方法使用ExceptionHandler注解然后定义捕获异常的类型即可对这些捕获的异常进行统一的处理。

二、目的

　　解决错误异常清晰、异常扩展性强、统一管理异常枚举类。

　　看了网上一些自定义异常，觉得不是太适用项目，异常展示不是太清晰，自己写了一套全局异常，只是为了解决开发者易于观察，方便扩展，如有不足之处还望大佬赐教。

三、目录截图

 四、开始编码

　　1.配置application.yml，方便开发者自定义开发模式，用于项目获取模式，展示异常。

server:
  port: 80
#spring相关配置
spring:
  profiles:
    active: @spring.active@

　　2.异常枚举基类，多个实现，用于统一管理异常枚举类。

/**
 * 异常基类
 */
public interface BaseExceptionEnum {

  public  String getMessage();

}

　　3.鉴权模块异常枚举类、只定义了一个异常类，如果有其他模块、可以定义多个实现BaseExceptionEnum ，用于异常管理，好找。

/**
 * 鉴权模块异常----按照模块划分
 */
public enum AuthExceptionEnum implements BaseExceptionEnum {
    /**
     * 账号或密码为空
     */
    USERNAME_PASS_NULL("账号或密码为空"),
    /**
     * 登录验证码为空或过期
     */
    USER_VERIFICATION_CODE_NULL("登录验证码为空或已过期"),
    /**
     * 登录验证码错误
     */
    USER_VERIFICATION_CODE("登录验证码错误"),
    /**
     * 登录密码错误
     */
    USERNAME_PASS_ERROR("登录密码错误"),
    /**
     * 登录失败
     */
    USER_LOGIN__ERROR("登录失败"),
    /**
     * 登录用户不存在
     */
    USER_NON_EXISTENT("登录用户不存在"),
    /**
     * 登录用户不存在
     */
    USER_HAS_BEEN_DELETE("登录用户不存在！"),
    /**
     * 登录用户已冻结
     */
    USER_FROZEN( "登录用户已冻结！");



    private final String message;

    AuthExceptionEnum(String message) {
        this.message = message;
    }
    AuthExceptionEnum(BaseExceptionEnum baseExceptionEnum) {
        this.message =baseExceptionEnum.getMessage();
    }

    @Override
    public String getMessage() {
        return message;
    }
}

　　4.定义统一异常类

/**
 * 异常统一
 */
public class BaseException extends RuntimeException  {

    private static final long serialVersionUID = 1L;

    private final int code;
    private final String value;

    protected BaseException(int code, String message) {
        super(message);
        this.code = code;
        this.value = message;
    }
    public int getCode() {
        return this.code;
    }
}

　　5.自定义响应用户异常类

/**
 * 用户请求错误
 * 向用户展示的异常错误对象 400 {"code":"1","message":"exception.message"}
 */
public class UserAlertException extends BaseException {

  public UserAlertException(String message) {
    super(1, message);
  }
  public UserAlertException(int code, String message) {
    super(code, message);
  }
  public UserAlertException(BaseExceptionEnum exceptionEnum) {
    super(1, exceptionEnum.getMessage());
  }
}

　　4.编写开发模式工厂类获取，方便扩展、易于维护。

/**
 * @ClassName: ExceptionPattern
 * @Description: 异常模式
 * @Author: fuzongle
 * @create: 2020-11-24 13:43
 **/
public class ExceptionPatternFactory {

    public Map<String, Function<Exception, ResponseResult>> map = new HashMap<>();

    {
        map.put("dev", this::dev);
        map.put("prod", this::prod);
    }

    //编写生产环境的业务错误异常逻辑
    public ResponseResult prod(Exception e) {
        return ResponseResult.builder().build();
    }

    //编写开发环境的业务错误异常逻辑
    public ResponseResult dev(Exception e) {
       return new ExceptionFactory().map.get(e.getClass().getName()).apply(e);
    }


}

　　5.编写异常统一返回格式、打印日志工厂类、本身想把这个用在开发模式、线上模式在建一个类维护线上的异常、现在项目上线大部分异常看日志的话、阿里云不是太方便、定位问题太慢，想集成存储异常日志，提供ui展示、也是再次基础上扩展生成模式即可。

/**
 * @ClassName: ExceptionPattern
 * @Description: 异常模式
 * @Author: fuzongle
 * @create: 2020-11-24 13:43
 **/
@Log4j2
public class ExceptionFactory {

  public Map<Object, Function<Object, ResponseResult>> map = new HashMap<>();

    {
        map.put(UserAlertException.class.getName(), this::UserAlertException);
        map.put(ArithmeticException.class.getName(), this::ArithmeticException);
    }


    /**
     * 用户请求错误 像用户展示
     */
    public  ResponseResult UserAlertException(Object e) {
        UserAlertException userAlertException = (UserAlertException) e;

        PrintlnErrorLog.printlnErrorLog("->>>自定义异常-给用户展示错误对象",
                ErrorUtils.getStackTrace(userAlertException),
                userAlertException.getStackTrace()[0]);

        return  ResponseResult.builder().code(userAlertException.getCode()).message(userAlertException.getMessage()).build();
    }

    /**
     * 算术运算异常
     */
    public  ResponseResult ArithmeticException(Object e) {
        ArithmeticException userAlertException = (ArithmeticException) e;

        PrintlnErrorLog.printlnErrorLog("->>>算术运算异常",
                ErrorUtils.getStackTrace(userAlertException),
                userAlertException.getStackTrace()[0]);

        return  ResponseResult.builder().code(HttpStatus.BAD_REQUEST.value()).message(userAlertException.getMessage()).build();
    }



}

　　5.打印异常日志、方法、可以提供开发者快速定位问题。

    /**
     * 异常日志
     */
    public static void printlnErrorLog(String title,String errInfo,StackTraceElement stackTraceElement) {

        log.info("
-----------------------异常分析 start-----------------------------------
	" +
                        "异常标题:{}
	" +
                        "文件名:{}
	" +
                        "类名:{}
	" +
                        "方法名:{}
	" +
                        "抛出异常行号:{}
	" +
                        "异常栈信息:{}
	" +
                        "解决方案: {}
" +
                        "------------------异常分析 end ---------------------------------------",
                title,
                stackTraceElement.getFileName(),
                stackTraceElement.getClassName(),
                stackTraceElement.getMethodName(),
                stackTraceElement.getLineNumber(),
                errInfo,
                "暂无");
    }

　　6.实现全局控制器通知、全局捕获异常、统一返回方式。@ExceptionHandler 拦截了异常，我们可以通过该注解实现自定义异常处理。其中，@ExceptionHandler 配置的 value 指定需要拦截的异常类型，上面拦截了 Exception.class 这种异常。

/**
 * @ClassName: ExceptionPattern
 * @Description: 全局异常配置
 * @Author: fuzongle
 * @create: 2020-11-25 12:43
 **/
@Slf4j
@RestControllerAdvice
public class GlobalExceptionHandler {


    /**
     * 获取开发模式、生产模式、默认dev
     */
    @Value("${spring.profiles.active}")
    private String var;

    /**
     * 全局异常.
     */
    @ExceptionHandler(Exception.class)
    public ResponseEntity<ResponseResult> handleGlobalException(Exception e) {
        ResponseResult apply = new ExceptionPatternFactory().map.get(var).apply(e);
        int httpStatus = Objects.equals(HttpStatus.INTERNAL_SERVER_ERROR.value(), apply.getCode()) ? HttpStatus.INTERNAL_SERVER_ERROR.value() : HttpStatus.BAD_REQUEST.value();
        return new ResponseEntity<>(apply, HttpStatus.valueOf(httpStatus));
    }


}

五、测试异常

　　1.直接编写API浏览器请求看结果。

/**
 * @Description: 测试
 **/
@RestController
public class ErrorController {

    /**
     * 自定义异常测试
     * @return
     */
    @RequestMapping("/test")
    public String test() {
        throw new UserAlertException(AuthExceptionEnum.USER_FROZEN);
    }

    /**
     * 系统捕获的异常
     */
    @RequestMapping("/ArithmeticException")
    public void ArithmeticException() {
        System.out.println(1/0);
    }
}

　　2.响应结果,结果非常清晰，可以在开发的地方遇见异常记录异常维护、ExceptionFactory维护异常，百度找到的解决方案可以再次记录，方便下次不在忘记错误，提高了开发效率。

六、温馨提示

1.如果有任何不懂的地方可以咨询我，随时欢迎互相帮助，如果写的不好的地方请大佬指教。

2.技术交流群QQ：422167709。

3.如果希望学习更多，希望微信扫码，长按扫码，帮忙关注一下，举手之劳，当您无助的时候真的能帮你。

4.获取源代码:扫码关注。回复任意字符、加作者微信秒发源码。