前言
if-else基本上是所有高级语言都有的语句。Java,Python,Go,C++。可以说if-else是编程语言种必须用的。基本上大大小小的需求都需要用到if-else。if-else不能滥用,不要让自己变成只会写if-else的程序员。if-else滥用会导致代码可读性,维护成本增高。我们需要使用好if-else来。
- 使用卫语句,提前return。
提前return减少不必要的判断,减少If-else嵌套层次。
优化前:
if(a != null){
b = doSomething(a);
if(b != null){
c = doSomething(b);
if(c != null){
d = doSomething(c);
if(d != null){
doSomething(d);
}
}
}
}
想想如果我们要在第二层,再增加一个同样的if-else判断会咋样,修改麻烦不?我们得把第二层之后的所有代码再包一层。再来几个怎么修改?
优化之后:
if(a == null){
return;
}
b = doSomething(a);
if(b == null){
return;
}
c = doSomething(b);
- 表驱动
对于固定的那种If-else多层次判断的。我们可以使用类似表格的方式。找到某个输入对应需要的结果函数。
优化前:
if(name.equals("A") || name.equals("B")){
status = 0;
}else if(name.equals("C") || name.equals("D")){
status = 1;
}else if(name.equals("E") || name.equals("F")){
status = 2;
}else {
status = 3;
}
优化后
Map<String,Integer> map = new HashMap<String,Integer>(){
{
put("A",0);
put("B",0);
put("C",1);
put("D",1);
}
};
status = map.getOrDefault("C",3);
如果复杂点的,可以使用Map与Function来解决
优化前
if(name.equals("A") || name.equals("B")){
result = doSomethingA(status);
}else if(name.equals("C") || name.equals("D")){
result = doSomethingC(status);
}else {
result = doSomethingOther(status);
}
优化后:
Map<String, Function<Integer,String>> map = new HashMap<String, Function<Integer,String>>(){
{
put("A", LearnDemo::doSomethingA);
put("B", LearnDemo::doSomethingA);
put("C", LearnDemo::doSomethingC);
}
};
result = map.getOrDefault(name,LearnDemo::doSomethingOther).apply(2);
- 使用策略模式
场景根据不同的参数走不同的逻辑,这种的一般可以使用策略模式。
优化前
if(name.equals("NORMAL")){
result = doSomething(status);
}else if(name.equals("VIP")){
result = doSomething(status);
}else if(name.equals("DEFAULT")){
result = doSomething(status);
}
优化后:
result = Strategy.valueOf(name).doSomething(status);
//使用枚举,这里也可以使用多态
public enum Strategy {
NORMAL {
@Override
String doSomething(Integer object) {
return "NORMAL";
}
},
VIP {
@Override
String doSomething(Integer object) {
return "VIP";
}
},
DEFAULT {
@Override
String doSomething(Integer object) {
return "DEFAULT";
}
},
;
abstract String doSomething(Integer object);
}
- 使用ObjectUtils.defaultIfNull Optional Lambda
优化前
List<Friend> userFriendList = userFriendMap.get(userId);
if(!CollectionUtils.isEmpty(userFriendList)){
userFriendList.stream().forEach(friend -> {
if(friend.getUserId().equals(friendId)){
user.setFriend(true);
return;
}
});
}
优化后
List<Friend> userFriendList = ObjectUtils.defaultIfNull(userFriendMap.get(userId),new ArrayList<>());
boolean isFriend = userFriendList.stream().anyMatch(friend -> friend.getUserId().equals(friendId));
总结
if-else过多可以使用以下方法优化
- 使用卫语句,提前return。减少不必要的嵌套判断。
- 采用表驱动,映射关系,减少判断。
- 采用设计模式,比如策略模式,责任链模式。
- 合理使用ObjectUtils.defaultIfNull ,Optional ,Lambda,Assert减少If判断