Spring Controller中的异常统一处理

Posted 2020-11-20 kong90hou

/**
 * 自定义异常类
 * @author ky01
 *
 */
@SuppressWarnings("serial")
public class ServiceException extends RuntimeException {
    private Integer code;

    public ServiceException(ErrorCode exceptionEnum) {
        super(exceptionEnum.getMsg());
        this.code = exceptionEnum.getCode();
    }

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

    public Integer getCode() {
        return code;
    }

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

/**
 * 统一异常处理类
 * @author ky01
 *
 */
@ControllerAdvice
public class ExceptionHandler {
    private static final Logger log = LoggerFactory.getLogger(ExceptionHandler.class);
    
    @org.springframework.web.bind.annotation.ExceptionHandler(value = Exception.class)
    @ResponseBody
    public String exceptionGet(Throwable e){
        if(e instanceof ServiceException){
            ServiceException myException = (ServiceException)e;
            return Result.error(myException.getCode(), myException.getMessage()).toJson();
        }
        
        if(e instanceof ValidationException){
            ValidationException myException = (ValidationException)e;
            return Result.error(ErrorCode.PARAM_INVALID.getCode(), myException.getMessage()).toJson();
        }
        
        //RuntimeException统一提示“未知错误”
        log.error("【系统异常】", e);
        return Result.error(ErrorCode.UNKNOWN_ERROR).toJson();
    }
    
}

使用：在Controller中抛出Exception时都会被ExceptionHandler捕获到并转为相应的字符串返回。