如何优化代码中的 if else

╰半夏微凉° 2024-04-17 18:52 171阅读 0赞

前段时间在阅读别人所写的代码的时候 , 发现其中一些业务相关的方法体内 , 出现了比较多的`if-else`语句多层嵌套的情况 . 首先我个人不是不提倡写`if-else`语句 , 不得不说 , 很多时候 , 在写某些逻辑 使用`if-else` 去做判断 , 代码看起来还是十分直观的 , 但是如果滥用`if-else` , 形成多层嵌套或者形成, 其中每个case 还包含了大量的逻辑 , 此时从可读性来说 , 使用`if-else`就有点得不偿失了 . 而且某些时候 , 可能并不需这么多的`if-else` , 或者是可以使用其他编码方式从而达到减少的`if-else` 的效果 .

减少`if-else` 的使用的方式有很多 , 例如设计模式层面的**策略模式**或者是**责任链模式** . 而这里跟大家分享一下一些个人在日常编码过程中经常用到的 , 比较简单的 、**从编码习惯层面上**的方式 , 去一些减少不必要`if-else`使用 . 由于本人只是一个小菜鸟 , 如果有写得不对的地方 , 恳请批评指正 .

# 一些减少if-else的编码方式 #

## 方式一 : 提前return ##

首先展示一段代码示例 :

if (condition1) {
        if (condition2) {
            return getSomething();
        } else {
            return 0;
        }
    } else {
        return 0;
    }
    复制代码

修改后的代码如下 :

//这里最好对这个flag所判断的逻辑补充注释进行描述
    boolean flag = !condition1 || (condition1 && !condition2)
    if(flag) {
    	return 0;
    }
    
    if (condition1 && condition2) {
        return getSomething();
    } 
    复制代码

如果存在已知在某些条件下 , 需要返回固定值的逻辑 , 可以将这部分逻辑抽取为一个独立的 `if-else block` , 并置于其他`if-else block`的前面 , 当符合该特定条件时 , 直接提前 `return` 固定值 . 这种方式最直接的效果就是降低`if-else`的嵌套数量 .

## 方式二 : 使用三目运算符 ##

先上例子 , 这里以一个业务场景为例 :

> 查询某条评论的图片URL列表 (如果有 , 评论的图片url列表以JSON数组字符串格式保存在评论表中)

修改前的代码如下 :

Comment comment = getById(commentId);
    
    if (Objects.isNull(comment)) {
        throw new RuntimeException("评论不存在或已被删除");
    }
    
    String imgListStr = comment.getImgList();
    if(StringUtils.isEmpty(imgListStr)) {
        return null;
    }
    
    return JSON.parseArray(imgListStr, String.class);
    复制代码

修改后 :

Comment comment = getById(commentId);
    
    if (Objects.isNull(comment)) {
        throw new RuntimeException("评论不存在或已被删除");
    }
    
    String imgListStr = comment.getImgList();
    return StringUtils.isEmpty(imgListStr)) ?
        null : JSON.parseArray(imgListStr, String.class);
    复制代码

## 方式三 : 使用Assert断言 ##

在编写业务代码过程中 , 如果需要对某些特定的条件进行判断 , 且当条件不满足时需要抛出异常 . 对于这种场景 , 除了使用上面三目运算符的示例当中的`if`方式 , 还可以通过使用Spring Framework 给我们提供的 `Assert` 工具类进行 .

其中常用的API 有 :

*  **isTrue(boolean expression , String message) :** 当`expressio == false`时 , 会抛出异常 , 异常的`message`则为第二个入参 ;
 *  **void notNull(@Nullable Object object, String message) :** 同上 , 当`object == null` 时 , 会抛出异常;
 *  **void notEmpty(@Nullable Collection<?> collection, String message) :** 同上 , 当集合对象为`null`或者集合元素为空时 , 会抛出异常 .
 *  .....

还有其他较多方法 , 可以直接看源码的解析 , 当然实际上`isTrue()` 已经够用了 , 如果需要更加的语义化 , 可以使用对应的API .

修改前代码 :

if (Objects.isNull(comment)) {
        throw new RuntimeException("评论不存在或已被删除");
    }
    复制代码

修改后代码 :

Assert.isTrue(Objects.nonNull(comment),"评论不存在或已被删除");
     Assert.notNull(comment,"评论不存在或已被删除");
    复制代码

目前Assert工具方法只能抛出单一一种异常 `java.lang.IllegalArgumentException` , 如果需要自定义所抛出的异常 , 则该方式不适用 .

## 方式四 : 使用Optional ##

Optional是 java8 的新特性 , 相当于一个对象的**容器** , 主要用于对象的`null`值校验 , 以及在进行校验后可链式地进行后续操作 , 如 : 抛出异常、`null`替换 等 .

其中我个人比较常用的几个方法为 :

*  **static <T> Optional<T> ofNullable(T value) :** 使用Optional 将对象进行包裹 ;
 *  **T orElse(T other) :** Optional中的对象为`null`时 ,返回入参的对象 .
 *  **T orElseGet(Supplier<? extends T> other) :** Optional中的对象为`null`时 , 返回`Supplier` 提供的值 ;
 *  **T orElseThrow(Supplier<? extends X> exceptionSupplier) :** Optional中的对象为`null`时 , 抛出`supplier`提供的自定义异常

代码示例 :

Message message1 = Optional.ofNullable(getById(messageId))
        .orElseThrow(() -> new RuntimeException("消息不存在!"));
    
    Message message2 = Optional.ofNullable(getById(messageId))
        .orElse(new Message());
    
    Message message3 = Optional.ofNullable(getById(messageId))
        .orElseGet(Message::new);
    复制代码

由于我日常需要的进行空值判断的比较多的场景是从数据库查询数据完毕时 , 需要查询结果进行空值判断 . 由于我所在的公司使用的持久层框架是mybatis , 不像Spirng Boot 2.x 默认版本的JPA 那样DAO层方法支持返回值为Optional , 所以这里如果需要使用Optional , 只能手动去使用上面列举的第一个方法对查询结果进行包装 .

当然 , IDEA其实已经给我们提供了该包装方式的热键了 , 如下图所示 :

![Animation.gif][]

# 结语 #

个人的一些减少`if-else` 编码习惯分享就这里了 , 这几种方式里面 , 我个人觉得效果最明显的还是第一种 **提前return** , 很多时候 , 提前return 也可以很好降低一段代码的复杂度 .

当然如果必须要使用大量的`if-else` 去控制逻辑时 , 在每个condition 标明一下注释还是一个挺不错的习惯 .

[Animation.gif]: https://imgconvert.csdnimg.cn/aHR0cHM6Ly91c2VyLWdvbGQtY2RuLnhpdHUuaW8vMjAxOS84LzI3LzE2Y2QyMWQ4MjY2MTliNmU_aW1hZ2VzbGlt?x-oss-process=image/format,png