说实话,其实我很讨厌在代码里大量使用if-else,一是因为该类代码执行方式属于面向过程的,二嘛,则是会显得代码过于冗余。这篇笔记,主要记录一些自己在工作实践当中针对if-else的优化心得,将会不定期地长期更新。
看到网上蛮多人推荐使用策略模式来优化if-else,但我总觉得,搞一堆策略类来优化大批量if-else,虽然想法很好,但无意之中很可能又会创造出很多类对象,就显得过于繁重了。若想使用策略模式来优化大批量if-else,其实有一种更好的方式,这是策略模式+枚举方式的改良。
1、根据if-else条件来判断赋值的,如:
String id = "";
if (flag) {
id = "a";
} else {
id = "b";
}
利用三目运算符,可以直接优化成一行代码:
id = flag ? "a" : "b";
2、利用if-else条件来判断调用方法,如:
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);
Jdk1.8新特性Stream流有三个这样API,anyMatch,allMatch,noneMatch,各自的作用如下:
● anyMatch:判断条件里任意一个满足条件,则返回true;
● allMatch:判断条件里所有都满足条件,则返回true;
● noneMatch:判断条件里所有都不满足条件,则返回true;
它们的使用方式其实很简单:
List list = Arrays.asList("a", "b", "c","d", "");
//任意一个字符串判断不为空则为true
boolean anyMatch = list.stream().anyMatch( s->StringUtils.isEmpty(s));
//所有字符串判断都不为空则为true
boolean allMatch = list.stream().allMatch( s->StringUtils.isEmpty(s));
//没有一个字符判断为空则为true
boolean noneMatch = list.stream().noneMatch( s->StringUtils.isEmpty(s));
可见,根据以上三种实现方式,可以在某种程度上优化if里判断条件过多的情况,那么,在哪种场景里比较合适利用其优化呢?
在日常实际开发当中,我们可能会看到过这样存在很多判断条件的代码:
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))) {
.....
}
这样优化后,是不是就比那堆if里堆积到一块的条件更为优雅了?
当然,这只是针对或条件的,若是遇到与条件时,同样可以用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)
优化量比较多的面向过程的if-else语句,还可以考虑使用Map来优化,虽然在一定程度上,创建一个额外map会占用内存,但那丁点内存对于现阶段计算机而言,可以说不足挂齿。 下面使用一个案例来介绍下————
在一些祖传老代码当中,可能遇到过类似这样又臭又冗余的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{
......
}
}
这时,可以根据具体场景,来考虑是否可以利用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()地赋值
前面提到过可使用策略枚举来优化大批量的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;
}
在实际工作中,我曾经遇到类似这样的代码,看起来像没什么问题,但如果其中某个属性值不幸为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的人士,怎么能容忍这一堆层层嵌套的代码存在呢!
在遇到这种层层嵌套的if-else判断时,可以考虑使用jdk1.8新特性Optional 类来优化,优化后的效果如下,顿时又优雅了很多。
String name = Optional.ofNullable(school)
.flatMap(School::getGrades)
.flatMap(Grades::getStuendt)
.map(Student::getName)
.orElse(null);