基于断言方式，减少业务逻辑中大量if判断

前言

使用过spring全家桶的都应该知道现如今大部分系统对于业务逻辑的异常处理，都是基于springmvc的统一异常处理器进行拦截包装后，返回给前端一个Result（包含code，业务异常msg等消息）结果对象，前端可以根据code以及msg进行友好提示。如下：

1. 自定义业务逻辑异常

/**
 * @author meilin.huang
 * @version 1.0
 * @date 2019-01-05 2:19 PM
 */
public class BusinessRuntimeException extends RuntimeException {

    private static final long serialVersionUID = -789021883759549647L;

    /**
     * 异常码
     */
    private Integer code;

    public BusinessRuntimeException(String msg) {
        super(msg);
    }

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

    public Integer getCode() {
        return code;
    }
}

2. 定义统一异常拦截器

/**
 * @author meilin.huang
 * @version 1.0
 * @date 2018-12-18 11:03 AM
 */
@RestControllerAdvice
public class GlobalExceptionHandler {

    private static Logger log = LoggerFactory.getLogger(GlobalExceptionHandler.class);

    @ExceptionHandler(Exception.class)
    public Result handleException(Exception e) {
        if (e instanceof BusinessException || e instanceof BusinessRuntimeException) {
            return Result.error(e.getMessage());
        }
        return Result.serverError();
    }
}

3. 就可以在业务逻辑中随意抛自定义的业务逻辑异常

public void updateUser(UserForm userForm) {
  User user = getById(userForm.getId);
  if (user == null) {
    throw new BusinessRuntimeException("不存在该用户");
  }
  // 校验用户状态枚举值是否存在，当然这也可以通过注解方式在controller层就校验了，这里只是举例
  if (!EnumUtils.isExist(StatusEnum.values(), userForm.getStatus())) {
    throw new BusinessRuntimeException("状态枚举值错误");
  }

  // ...........
  //余下可能还有一些其他的业务逻辑判断校验，并抛异常的代码
}

更优雅的解决方式

通过上面业务判断可以发现，其实这些业务逻辑的判断处理，基本都是基于if(...) {throw new BusinessRuntimeException(...)} 于是其实可以借鉴Spring中的org.springframework.util.Assert断言类进行处理，可以减少大量的if判断。具体可参考如下：

1. 首先定义一个业务断言类

/**
 * 业务断言
 * @author meilin.huang
 * @version 1.0
 * @date 2019-07-14 18:24
 */
public final class BusinessAssert {

    /**
     *  断言对象不为空
     * @param object    对象
     * @param msg       不满足断言的异常信息
     */
    public static void notNull(Object object, String msg) {
        state(object != null, msg);
    }

    public static void notNull(Object object, Supplier supplier) {
        state(object != null, supplier);
    }

    /**
     * 断言字符串不为空
     * @param str   字符串
     * @param msg   不满足断言的异常信息
     */
    public static void notEmpty(String str, String msg) {
        state(!StringUtils.isEmpty(str), msg);
    }

    /**
     * 断言集合不为空
     * @param collection  集合
     * @param msg         不满足断言的异常信息
     */
    public static void notEmpty(Collection collection, String msg) {
        state(!CollectionUtils.isEmpty(collection), msg);
    }

    /**
     * 断言一个boolean表达式
     * @param expression  boolean表达式
     * @param message     不满足断言的异常信息
     */
    public static void state(boolean expression, String message) {
        if (!expression) {
            throw new BusinessRuntimeException(message);
        }
    }

    /**
     * 断言一个boolean表达式，用于需要大量拼接字符串以及一些其他操作等
     * @param expression  boolean表达式
     * @param supplier    msg生产者
     */
    public static void state(boolean expression, Supplier supplier) {
        if (!expression) {
            throw new BusinessRuntimeException(supplier.get());
        }
    }
}

2. 于是乎就可以在业务逻辑代码中如下使用啦：

public void updateUser(UserForm userForm) {
  User user = getById(userForm.getId);
  BusinessAssert.notNull(user, "不存在该用户");
  // 校验用户状态枚举值是否存在，当然这也可以通过注解方式在controller层就校验了，这里只是举例
  BusinessAssert.state(EnumUtils.isExist(StatusEnum.values(), userForm.getStatus()), "状态枚举值错误");

  // ...........
  //余下可能还有一些其他的业务逻辑判断校验，并抛异常的代码
}

总结

使用了基于断言的方式进行业务逻辑判断，明显会比第一种方式优雅并简洁许多，也可以少写不少代码，大家可以试着用用看蛤，舒服不是一点点。