if-else优化实践总结

在实际开发中，我们经常会用到if-else的情况，但是这样代码写太过冗余，我们可以通过一些方式来避免总写if-else的代码；

public String getDay(String day){
    if("Monday".equals(day)){
       return "今天上英语课";
    }else if("Tuesday".equals(day)){
       return "今天上语文课";
    }else if("Wednesday".equals(day)){
       return "今天上数学课";
    }else if("Thursday".equals(day)){
       return "今天上音乐课";
    }else if("Sunday".equals(day)){
       return "今天上编程课";
    }else{
       ......
    }
}

 

优化策略1：三目运算符

如果是简单的判断赋值，可以使用三门运算符

String id="";
if(flag){
    id="a";
}else{
    id="b";
}

利用三目运算符，可以直接优化成一行代码：

id=flag?"a":"b";

 

如果是简单的判断调用方法，可以使用三门运算符

Set set1=new HashSet<>();
Set set2=new HashSet<>();

if(flag){
    set1.add(id);
}else{
    set2.add(id);
}    

 利用三目运算符，可以直接优化成：

Set set1=new HashSet<>();
Set set2=new HashSet<>();
(flag?set1:set2).add(id);

 

优化策略2：使用Stream优化

Jdk1.8新特性Stream流有三个这样API，anyMatch，allMatch，noneMatch，各自的作用如下：

• anyMatch：判断条件里任意一个满足条件，则返回true；

• allMatch：判断条件里所有都满足条件，则返回true；

• noneMatch：判断条件里所有都不满足条件，则返回true;

在日常实际开发当中，我们可能会看到过这样存在很多判断条件的代码：

 if(StringUtils.isEmpty(str1) || StringUtils.isEmpty(str2) ||
    StringUtils.isEmpty(str3) || StringUtils.isEmpty(str4) ||
    StringUtils.isEmpty(str5) || StringUtils.isEmpty(str6)
   ){
 .....
}

 这时，就可以考虑到，使用stream流来优化，优化后的代码如下：

 if(Stream.of(str1, str2, str3, str4,str5,str6).anyMatch(s->StringUtils.isEmpty(s))){
 .....
 }

 当然，这只是针对或条件的，若是遇到与条件时，同样可以用Stream来优化，例如：

if(StringUtils.isEmpty(str1) && StringUtils.isEmpty(str2) &&
   StringUtils.isEmpty(str3) && StringUtils.isEmpty(str4) &&
   StringUtils.isEmpty(str5) && StringUtils.isEmpty(str6)
){
   .....
}

 使用Stream优化后：

if(Stream.of(str1, str2, str3, str4,str5,str6).allMatch(s->StringUtils.isEmpty(s))){
  .....
}

还有一个判断任意都不为空的操作：

StringUtils.isNoneEmpty(str1,str2,str3)

 

优化策略3：使用Map优化

优化量比较多的面向过程的if-else语句，还可以考虑使用Map来优化，虽然在一定程度上，创建一个额外map会占用内存，但那丁点内存对于现阶段计算机而言，可以说不足挂齿。

public String getDay(String day){
    if("Monday".equals(day)){
       return "今天上英语课";
    }else if("Tuesday".equals(day)){
       return "今天上语文课";
    }else if("Wednesday".equals(day)){
       return "今天上数学课";
    }else if("Thursday".equals(day)){
       return "今天上音乐课";
    }else if("Sunday".equals(day)){
       return "今天上编程课";
    }else{
       ......
    }
}

这时，可以根据具体场景，来考虑是否可以利用Map优化，使用Map优化的方式，是先在该类中定义一个static的map，类似这样：

public static final Map dayMap=  ImmutableMap.builder()
    .put("Monday","今天上英语课")
    .put("Tuesday","今天上语文课")
    .put("Wednesday","今天上数学课")
    .put("Thursday","今天上音乐课")
    .put("Sunday","今天上编程课")
    .build();

定义完后，就直接在先前使用if-else的方法里，进行这样优化：

public String getDay(String day){
    return dayMap.get(day);
}

这样优化后，业务方法里的判断获取值的处理，是不是就清爽了很多，当然，这只是针对量比较多的if-else而言，若是比较少的判断语句，再额外定义一个map来搞，隐约有画蛇添足的嫌疑。

细心的读者可能会发现， 我在定义map的时候，使用到了一个ImmutableMap的类，这是Google Guava里的一个类，可生成一个不可变的Map对象，这就意味着，初始化定义后，后续就无法再put修改了，它的这个特性可以保证线程的安全。一般用来替换if-else的map，我们就是要求在初始化定义后，就不会再允许修改了，因此，这个ImmutableMap生成的map，可以很好地帮我们实现这一点。另外，最重要一点是，使用这个ImmutableMap，可以实现链式编程，就像上面定义的链式写法，若是用传统的map定义，就每次都要map.put()、map.put()地赋值。

优化策略4：使用枚举优化

前面提到过可使用策略枚举来优化大批量的if-else，当然，若只是判断获不同条件来取值的代码，可以考虑直接使用枚举来优化，其效果与map的处理效果类似。

还是用前面判断课程的if-else为案例来优化。

首先，先在类中定义一个枚举：

public enum dayEnum {
    Monday("今天上英语课"),
    Tuesday("今天上语文课"),
    Wednesday("今天上数学课"),
    Thursday("今天上音乐课"),
    Sunday("今天上编程课");

    public String value;

    dayEnum(String value){
      this.value=value;
    }
}

定义完后，就可以类似前面map的方式，直接将判断值去枚举里获取，然后直接返回获取到的值，这样写法是不是也比较优雅了。

public String getDay(String day){
    return dayEnum.valueOf(day).value;
}

优化策略5：使用Optional类优化

在实际工作中，我曾经遇到类似这样的代码，看起来像没什么问题，但如果其中某个属性值不幸为null，那么，恭喜你，你将会喜提一个NullPointerException异常。

String name=school.getGrades().getStuendt().getName();

若要处理这个可能出现的空指针异常，传统写法，可以写一堆if-else语句来处理，就像这样子

String name=null;
if(school!=null){
    Grades grade=school.getGrades();
    if(grade!=null){
        Student student=grade.getStuendt();
        if(student!=null){
          name = student.getName();
        }
    }
}

在遇到这种层层嵌套的if-else判断时，可以考虑使用jdk1.8新特性Optional 类来优化，优化后的效果如下，顿时又优雅了很多。

String name = Optional.ofNullable(school)
  .flatMap(School::getGrades)
  .flatMap(Grades::getStuendt)
  .map(Student::getName)
  .orElse(null);

优化策略6：使用设计模式——策略模式+枚举

使用策略模式代替繁琐的if else_策略模式代替if else_lucky_ljw的博客-CSDN博客