断言+异常处理类,代码更简洁了

2022-07-01 13:10:40 浏览数 (1)

点击上方“芋道源码”,选择“设为星标”

管她前浪,还是后浪?

能浪的浪,才是好浪!

每天 10:33 更新文章,每天掉亿点点头发...

源码精品专栏

  • 原创 | Java 2021 超神之路,很肝~
  • 中文详细注释的开源项目
  • RPC 框架 Dubbo 源码解析
  • 网络应用框架 Netty 源码解析
  • 消息中间件 RocketMQ 源码解析
  • 数据库中间件 Sharding-JDBC 和 MyCAT 源码解析
  • 作业调度中间件 Elastic-Job 源码解析
  • 分布式事务中间件 TCC-Transaction 源码解析
  • Eureka 和 Hystrix 源码解析
  • Java 并发源码

来源:c1n.cn/E6fZj

  • 背景
  • 业务异常处理示例
  • 附上代码

背景

软件开发过程中,不可避免的是需要处理各种异常,所以代码中就会出现大量的 try {...} catch {...} finally {...} 代码块,不仅有大量的冗余代码,而且还影响代码的可读性。

另一个就是面对业务异常的情况,我们经常需要将业务异常结果组装成统一的信息返回给前端进行提示。

假如我们在每个接口中都去包装异常信息进行返回就会让代码变得很冗余且混乱。在我司的实际项目开发过程中,我们会巧用断言去简化代码。

基于 Spring Boot MyBatis Plus Vue & Element 实现的后台管理系统 用户小程序,支持 RBAC 动态权限、多租户、数据权限、工作流、三方登录、支付、短信、商城等功能。 项目地址:https://github.com/YunaiV/ruoyi-vue-pro

业务异常处理示例

假设我们定义的标准接口响应实体为 ApiResult:

代码语言:javascript复制
 @Data
 @Builder
 @NoArgsConstructor
 @AllArgsConstructor
 public class ApiResult<T> implements Serializable {
     private static final long serialVersionUID = 411731814484355577L;
     private int responseCode;
     private String responseMsg;
     private boolean isSuccess;
     private T data;
 
     public String toString() {
         return "ApiResult(responseCode="   this.getResponseCode()   ", responseMsg="   this.getResponseMsg()   ", isSuccess="   this.isSuccess()   ", data="   this.getData()   ")";
     }
 }

那么我们接口处理业务逻辑时代码就会变成这样,看起来非常多代码:

代码语言:javascript复制
 public ApiResult cancelService(@PathVariable Long serviceOrderId){
     ServiceOrder serviceOrder = serviceOrderMapper.selectByPrimaryKey(serviceOrderId);
     ApiResult result = new ApiResult<>();
     if (ObjectUtil.isNull(serviceOrder)) {
         result.setSuccess(false);
         result.setResponseCode(ErrorCodeEnum.FAIL.getCode());
         result.setResponseMsg("查无此服务单");
         return result;
     }
     if(serviceOrder.getOrderStatus().equals(cancelOrderStatus)){
         result.setSuccess(false);
         result.setResponseCode(ErrorCodeEnum.FAIL.getCode());
         result.setResponseMsg("已取消的服务单不允许再次取消");
         return result;
     }
     if(serviceOrder.getSortOrderId() != null){
         result.setSuccess(false);
         result.setResponseCode(ErrorCodeEnum.FAIL.getCode());
         result.setResponseMsg("已配置物料的服务单不允许取消");
         return result;
     }
     // ...other check
 
     // ...do something
     return result;
 }

然后在上面这个代码基础上,我们可以观察到,里面其实有非常多的重复代码,完全可以把它们装到 ApiResult 里面。

这也是我看到很多开源框架的处理方式(PS:所以我第一个自己写的框架也是这么处理的)

在原 ApiResult 实体中增加一些公用的处理方法:

代码语言:javascript复制
 public static ApiResult<String> success() {
     return success("success");
 }
 
 public static <T> ApiResult<T> success(T data) {
     return (new ApiResult()).setResponseCode(0).setResponseMsg("操作成功").setSuccess(true).setData(data);
 }
 
 public static ApiResult<String> fail() {
     return fail(-1);
 }
 
 public static ApiResult<String> fail(int code) {
     return fail(code, "fail");
 }
 
 public static <T> ApiResult<T> fail(T data) {
     return fail(-1, data);
 }
 
 public static <T> ApiResult<T> fail(int code, T data) {
     return (new ApiResult()).setResponseCode(code).setResponseMsg("操作失败").setSuccess(false).setData(data);
 }
 
 public static <T> ApiResult<T> success(int code, String message, T data) {
     return (new ApiResult()).setResponseCode(code).setResponseMsg(message).setSuccess(true).setData(data);
 }
 
 public static <T> ApiResult<T> fail(int code, String message, T data) {
     return (new ApiResult()).setResponseCode(code).setResponseMsg(message).setSuccess(false).setData(data);
 }

然后业务逻辑处理就变成这样了,看起来还不错是不是:

代码语言:javascript复制
 /**
  * 取消服务单(不用断言)
  */
 public ApiResult cancelService(Long serviceOrderId){
     ServiceOrder serviceOrder = serviceOrderMapper.selectByPrimaryKey(serviceOrderId);
     ApiResult result = new ApiResult<>();
     if (ObjectUtil.isNull(serviceOrder)) {
         result = ApiResult.fail(ErrorCodeEnum.FAIL.getCode(), "查无此服务单");
         return result;
     }
     if(serviceOrder.getOrderStatus().equals(cancelOrderStatus)){
         result = ApiResult.fail(ErrorCodeEnum.FAIL.getCode(), "已取消的服务单不允许再次取消");
         return result;
     }
     if(serviceOrder.getSortOrderId() != null){
         result = ApiResult.fail(ErrorCodeEnum.FAIL.getCode(), "已配置物料的服务单不允许取消");
         return result;
     }
     // ...other check
 
     // ...do something
     return result;
 }

但是我们可以用异常处理类 断言处理得更加简化。

增加异常处理类:

代码语言:javascript复制
 @Slf4j
 @ControllerAdvice
 public class GlobalExceptionHandler {
     @ExceptionHandler(value = BusinessException.class)
     @ResponseBody
     public ResponseBean businessExceptionHandler(BusinessException e) {
         log.info("business error : {}",e.getMessage(),e);
         if (e.getCode() == -1) {
             return ResponseBean.error(ApiCode.SERVICE_ERROR.getValue(), ApiCode.SERVICE_ERROR.getMessage());
         }
         return ResponseBean.error(e.getCode(), e.getMessage());
     }
 }

增加异常类 BusinessException:

代码语言:javascript复制
 /**
  * 业务异常,异常信息会返回到前端展示给用户
  *
  * @date 2020/12/15 14:18
  */
 public class BusinessException extends RuntimeException {
     private static final long serialVersionUID = -5770538329754222306L;
 
     private int code = 1;
     private Level level;
 
     public BusinessException(int code, String message, Throwable cause) {
         super(message, cause);
         this.code = code;
     }
 
     public BusinessException(String message) {
         super(message);
     }
 
     public BusinessException(Level level, String message) {
         super(message);
         this.level = level;
     }
 
     public BusinessException(Throwable cause) {
         super(cause);
     }
 
     public BusinessException(int code, String message) {
         super(message);
         this.code = code;
     }
 
     public int getCode() {
         return this.code;
     }
 
     public final Level getLevel() {
         return this.level;
     }
 }

增加断言工具类 AssertUtil:

代码语言:javascript复制
 public class AssertUtil extends cn.com.bluemoon.common.web.exception.AssertUtil  {
     public AssertUtil() {
     }
 
     /**
      * 服务调用异常
      * @param expression
      * @param message
      */
     public static void isTrueServiceInvoke(boolean expression, String message) {
         if (!expression) {
             throw new ServiceInvokeException(message);
         }
     }
 
     /**
      * 抛出异常(默认错误1000)
      * @param message
      */
     public static void businessInvalid(String message) {
         throw new BusinessException(ApiCode.SERVICE_ERROR.getValue(), message);
     }
 
     /**
      * 表达式为真即抛出异常(默认错误1000)
      *
      * @param expression
      * @param message
      */
     public static void businessInvalid(boolean expression, String message) {
         if (expression) {
             throw new BusinessException(ApiCode.SERVICE_ERROR.getValue(), message);
         }
     }
 
     /**
      * 表达式为真即抛出异常
      *
      * @param expression
      * @param message
      */
     public static void businessInvalid(boolean expression, int code, String message) {
         if (expression) {
             throw new BusinessException(code, message);
         }
     }
 }

最后优化的结果:

代码语言:javascript复制
 /**
  * 取消服务单
  */
 public ApiResult cancelService(@PathVariable Long serviceOrderId){
     ServiceOrder serviceOrder = serviceOrderMapper.selectByPrimaryKey(serviceOrderId);
     AssertUtil.businessInvalid(ObjectUtil.isNull(serviceOrder),"查无此服务单");
     AssertUtil.businessInvalid(serviceOrder.getOrderStatus().equals(cancelOrderStatus),"查无此服务单");
     AssertUtil.businessInvalid(serviceOrder.getSortOrderId() != null,"查无此服务单");
     // ...other check
 
     // ...do something
     return ApiResult.success();
 }

最后,我们可以看到我们的接口由 19 行的业务检查代码简化到了 3 行。这只是单接口的情况下,在业务多且复杂的情况下能给我们节省更多的开发时间,把精力集中在核心业务上。

基于微服务的思想,构建在 B2C 电商场景下的项目实战。核心技术栈,是 Spring Boot Dubbo 。未来,会重构成 Spring Cloud Alibaba 。 项目地址:https://github.com/YunaiV/onemall

附上代码

统一异常处理类:

代码语言:javascript复制
 /**
  * 统一异常处理
  */
 @Slf4j
 @ControllerAdvice
 public class GlobalExceptionHandler {
 
     @ExceptionHandler(value = AssertException.class)
     @ResponseBody
     public ResponseBean bootExceptionHandler(AssertException e) {
         ApiCode apiCode = ApiCode.getObjectByValue(e.getCode());
         log.error("business error : {}", e.getMessage(), e);
         if (e.getCode() == -1) {
             return ResponseBean.error(ApiCode.SERVICE_ERROR.getValue(), ApiCode.SERVICE_ERROR.getMessage());
         }
         return ResponseBean.error(apiCode.getValue(), e.getMessage());
     }
 
     @ExceptionHandler(value = com.alibaba.fastjson.JSONException.class)
     public ResponseBean alibabaJsonExceptionHandler(com.alibaba.fastjson.JSONException e) {
         ResponseBean response = new ResponseBean(false, ApiCode.PARAM_FORMAT_INCORR.getValue(), ApiCode.PARAM_FORMAT_INCORR.getMessage()   e.getMessage(), null);
         log.error("1102", e);
         return response;
     }
 
     @ExceptionHandler(value = JSONException.class)
     @ResponseBody
     public ResponseBean jsonExceptionHandler(JSONException e) {
         ResponseBean response = new ResponseBean(false, ApiCode.PARAM_FORMAT_INCORR.getValue(), ApiCode.PARAM_FORMAT_INCORR.getMessage()   e.getMessage(), null);
         log.error(ApiCode.PARAM_FORMAT_INCORR.getValue()   "", e);
         return response;
     }
 
     @ExceptionHandler(value = JsonParseException.class)
     @ResponseBody
     public ResponseBean jsonParseExceptionHandler(JsonParseException e) {
         ResponseBean response = new ResponseBean(false, ApiCode.PARAM_FORMAT_INCORR.getValue(), String.format(ApiCode.PARAM_FORMAT_INCORR.getMessage()   ":%s", e.getMessage()), null);
         log.error(ApiCode.PARAM_FORMAT_INCORR.getValue()   "", e);
         return response;
     }
 
     @ExceptionHandler(value = Exception.class)
     @ResponseBody
     public ResponseBean exceptionHandler(Exception e) {
         ResponseBean response = new ResponseBean(false, ApiCode.SERVICE_ERROR.getValue(), ApiCode.SERVICE_ERROR.getMessage(), null);
         log.error(ApiCode.SERVICE_ERROR.getValue()   "", e);
         return response;
     }
 
     @ExceptionHandler(value = MethodArgumentTypeMismatchException.class)
     @ResponseBody
     public ResponseBean exceptionHandle(MethodArgumentTypeMismatchException e) {
         ResponseBean response = new ResponseBean(false, ApiCode.PARAM_FORMAT_INCORR.getValue(), String.format(ApiCode.PARAM_FORMAT_INCORR.getMessage()   ":%s", e.getMessage()), null);
         log.error(ApiCode.PARAM_FORMAT_INCORR.getValue()   "", e);
         return response;
     }
 
     @ExceptionHandler(value = WebException.class)
     @ResponseBody
     public ResponseBean exceptionHandler(WebException e) {
         ResponseBean response = new ResponseBean(e.getIsSuccess(), e.getResponseCode(), e.getResponseMsg(), null);
         log.error(e.getResponseCode()   "", e);
         return response;
     }
 
     @ExceptionHandler(value = IllegalArgumentException.class)
     @ResponseBody
     public ResponseBean exceptionHandler(IllegalArgumentException e) {
         log.error("illegal request : {}", e.getMessage(), e);
         return ResponseBean.error(ApiCode.PARAM_INVALID.getValue(), ApiCode.PARAM_INVALID.getMessage());
     }
 
     @ExceptionHandler(value = ServiceInvokeException.class)
     @ResponseBody
     public ResponseBean exceptionHandler(ServiceInvokeException e) {
         log.error("serviceInvoke error request : {}", e.getMessage(), e);
         return ResponseBean.error(ApiCode.SERVICE_ERROR.getValue(), ApiCode.SERVICE_ERROR.getMessage());
     }
 
     @ExceptionHandler(value = BusinessException.class)
     @ResponseBody
     public ResponseBean businessExceptionHandler(BusinessException e) {
         log.info("business error : {}",e.getMessage(),e);
         if (e.getCode() == -1) {
             return ResponseBean.error(ApiCode.SERVICE_ERROR.getValue(), ApiCode.SERVICE_ERROR.getMessage());
         }
         return ResponseBean.error(e.getCode(), e.getMessage());
     }
 
     @ResponseBody
     @ExceptionHandler(MethodArgumentNotValidException.class)
     public ResponseBean  exceptionHandler(MethodArgumentNotValidException e) {
         log.info("req params error", e);
         String message = e.getBindingResult().getFieldError().getDefaultMessage();
         if (StringUtils.isNotBlank(message) && !"不能为空".equals(message)) {
             return ResponseBean.error(ApiCode.PARAM_INVALID.getValue(), message);
         }
         return ResponseBean.error(ApiCode.PARAM_INVALID.getValue(), ApiCode.PARAM_INVALID.getMessage());
     }
 
     @ExceptionHandler(value = TokenErrorException.class)
     @ResponseBody
     public ResponseBean tokenErrorExceptionHandler(TokenErrorException e) {
         log.info("登录失效 : {}",e.getMessage(),e);
         return ResponseBean.error(ApiCode.SERVICE_ERROR.getValue(), "登录已失效,请重新登录!");
     }
 
     @ExceptionHandler(value = ServiceException.class)
     @ResponseBody
     public ResponseBean businessExceptionHandler(ServiceException e) {
         log.info("service error : {}",e.getMessage(),e);
         return ResponseBean.error(ApiCode.SERVICE_ERROR.getValue(), e.getMessage());
     }
 }

异常情况枚举,仅作参考:

代码语言:javascript复制
 public enum ErrorCodeEnum implements EnumBase{
 
     FAIL(-1, "网络异常,请稍后再试"),
 
     SUCCESS(0, "请求成功"),
 
     MAX_UPLOAD_SIZE_ERROR(1000, "上传文件不能超过20M"),
 
     SERVICE_BUSY_ERROR(1000, "服务器正在繁忙,请稍后再试哦~"),
 
     REQUEST_PARAMS_FAIL(1001, "参数错误"),
 
     USER_NOT_LOGIN(1002, "用户未登录,请重新登录"),
 
     USER_HAS_EXIST_LOGIN(1007, "用户已经存在,请检查!"),
 
     USER_CODE_NOT_EXIST(1008, "用户编码不存在,请检查!"),
 
     REQUEST_PARAMS_FORMAT_ERROR(1102, "请求参数格式异常"),
 
     PASSWORD_SAFETY_ERROE(2204, "密码不符合安全规则,请通过忘记密码重新设置8-18位数字 字母组合密码"),
 
     TOKEN_EXPIRED(2301, "token过期"),
 
     TOKEN_ERROR(2302, "token验证失败"),
 
     INTERFACE_ERROR(10000, "接口服务器异常");
 
     private final int code;
     private final String msg;
 
     ErrorCodeEnum(int code, String msg) {
         this.code = code;
         this.msg = msg;
     }
 
     @Override
     public int getCode() {
         return this.code;
     }
 
     @Override
     public String getMsg() {
         return this.msg;
     }
 }


欢迎加入我的知识星球,一起探讨架构,交流源码。加入方式,长按下方二维码噢

已在知识星球更新源码解析如下:

最近更新《芋道 SpringBoot 2.X 入门》系列,已经 101 余篇,覆盖了 MyBatis、Redis、MongoDB、ES、分库分表、读写分离、SpringMVC、Webflux、权限、WebSocket、Dubbo、RabbitMQ、RocketMQ、Kafka、性能测试等等内容。

提供近 3W 行代码的 SpringBoot 示例,以及超 4W 行代码的电商微服务项目。

获取方式:点“在看”,关注公众号并回复 666 领取,更多内容陆续奉上。

代码语言:javascript复制
文章有帮助的话,在看,转发吧。谢谢支持哟 (*^__^*)

0 人点赞