如何优化代码中大量的if/else,switch/case?

最新推荐文章于 2024-07-17 04:28:15 发布

qq_31102103

最新推荐文章于 2024-07-17 04:28:15 发布

阅读量309

点赞数

文章标签： JAVA

使用枚举
这里我们简单的定义一个表示状态的枚举。

public enum Status {
NEW(0),RUNNABLE(1),RUNNING(2),BLOCKED(3),DEAD(4);

public int statusCode;

Status(int statusCode){
this.statusCode = statusCode;
}
}

那么我们在使用的时候就可以直接通过枚举调用了。

int statusCode = Status.valueOf("NEW").statusCode;
优雅的解决了下面代码赋值的方式

if(param.equals("NEW")){
statusCode = 0;
}else if(param.equals("RUNNABLE")){
statusCode = 1;
}

善用 Optional
在项目中，总少不了一些非空的判断,可能大部分人还是如下的用法

if(null == user){
//action1
}else{
//action2
}

这时候该掏出Optional这个秘密武器了，它可以让非空校验更加优雅，间接的减少if操作。没了解过Optional的同学可自行Google，这里就不再赘述。

Optional<User> userOptional = Optional.ofNullable(user);
userOptional.map(action1).orElse(action2);
上面的代码跟第一段是等效的，通过一些新特性让代码更加紧凑。

表驱动法
来自Google的解释：表驱动法是一种编程模式，它的本质是，从表里查询信息来代替逻辑语句(if,case)。下面看一个案例，通过月份来获取当月的天数（仅作为案例演示，获取2月份的数据不严谨），普通做法：

int getMonthDays(int month){
   switch(month){
       case 1:return 31;break;
       case 2:return 29;break;
       case 3:return 31;break;
       case 4:return 30;break;
       case 5:return 31;break;
       case 6:return 30;break;
       case 7:return 31;break;
       case 8:return 31;break;
       case 9:return 30;break;
       case 10:return 31;break;
       case 11:return 30;break;
       case 12:return 31;break;
       default：return 0;
   }
}
表驱动法实现方式

int monthDays[12] = {31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
int getMonthDays(int month){
return monthDays[--month];
}
其实这里的表就是数组而已，通过直接查询数组来获得需要的数据，那么同理，Map之类的容器也可以成为我们编程概念中的表。

Map<?, Function<?> action> actionsMap = new HashMap<>();

// 初试配置对应动作
actionsMap.put(value1, (someParams) -> { doAction1(someParams)});
actionsMap.put(value2, (someParams) -> { doAction2(someParams)});
actionsMap.put(value3, (someParams) -> { doAction3(someParams)});

// 省略 null 判断
actionsMap.get(param).apply(someParams);
通过Java8的lambda表达式，我们把需要执行东西存进value中，调用的时候通过匹配key的方式进行。

提前判断返回
在之前的文章《优化代码中的“坏味道”》里也有提过，如下语句
if(condition){
//dost
}else{
return ;
}
改为

if(!condition){
return ;
}
//dost
1
避免一些不必要的分支，让代码更精炼。

其他方法
除了上面提到的方法，我们还可以通过一些设计模式，例如策略模式，责任链模式等来优化存在大量if,case的情况，其原理会和表驱动的模式比较相似，大家可以自己动手实现一下，例如我们在Netty的使用过程中，可能会出现需要大量判断不同的命令去执行对应动作的场景。

ServerHandler.java

if(command.equals("login")){
//执行登录
}else if(command.equals("chat")){
//聊天
}else if(command.equals("broadcast")){
//广播信息
}
..
最后要明确一点，不是所有的if/else,switch/case都需要优化，当我们发现有“痛点”或者“闻到代码有坏味道”再来优化才是最好的，不然你可能会写了一个从不扩展的可扩展代码，所有的优化都是为了更好的迭代项目，更好的服务于业务，而不是为了优化而优化。