自定义异常类

/**
* 自定制异常类
*
* @author MoCha
* @date 2019/5/25
*/
@Getter
public class CustomException extends RuntimeException {
private int code;
private String message;

public CustomException(int code, String message) {
this.code = code;
this.message = message;
}

public CustomException(ResultStatusEnum resultStatusEnum) {
this.code = resultStatusEnum.getCode();
this.message = resultStatusEnum.getMessage();
}
}

/**
* 全局异常处理
*
* @author MoCha
* @date 2019/5/25
*/
@ControllerAdvice
public class GlobalExceptionHandler {
@ResponseBody
@ExceptionHandler(CustomException.class)
public Map<String, Object> handleCustomException(CustomException customException) {
Map<String, Object> errorResultMap = new HashMap<>(16);
errorResultMap.put("code", customException.getCode());
errorResultMap.put("message", customException.getMessage());
return errorResultMap;
}
}