用优雅的方式去除过多的if else
在项目中在一个方法中过多的使用if else,一个复杂的业务可能会使用到6,7个if else,会影响代码的可读性,后期不利于维护,修改起来也很麻烦,例如:
private int ifResult(String value, int param) {
if(StringUtils.equals(value, "testA")){
return param + 1;
}else if (StringUtils.equals(value, "testB")){
return param + 2;
}else if (StringUtils.equals(value, "testC")) {
return param + 3;
}else {
return 0;
}
}
lambda方式去除if else
使用注解@FunctionalInterface自定义函数式接口:
@FunctionalInterface
public interface SumDemo {
int sum(int x);
}
lambda表达式返回SumDemo存进map中:
//这里的代码建立了对三个分支的业务逻辑的映射
Map<String, SumDemo> map = new HashMap<>();
map.put("testA", param -> param + 1);
map.put("testB", param -> param + 2);
map.put("testC", param -> param + 3);
输出:
int lambdaResult =map.get("testA").sum(2)
System.out.println("lamdaResult : " + lambdaResult);
//输出结果为 lamdaResult : 3
完整代码示例:
import org.apache.commons.lang3.StringUtils;
import java.util.HashMap;
import java.util.Map;
public class Test {
Map<String, SumDemo> map = new HashMap<>();
public Test(){
map.put("testA", param -> param + 1);
map.put("testB", param -> param + 2);
map.put("testC", param -> param + 3);
}
public static void main(String[] args) {
Test test = new Test();
String value = "testA";
int param = 2;
int lambdaResult = test.lambdaSum(value,param);
System.out.println("lamdaResult : " + lambdaResult);
int ifResult = test.ifResult(value,param);
System.out.println("ifResult : " + ifResult);
}
private int lambdaSum(String value, int param) {
return map.get(value).sum(param);
}
private int ifResult(String value, int param) {
if(StringUtils.equals(value, "testA")){
return param + 1;
}else if (StringUtils.equals(value, "testB")){
return param + 2;
}else if (StringUtils.equals(value, "testC")) {
return param + 3;
}else {
return 0;
}
}
}
输出为:
lamdaResult : 3
ifResult : 3