如何解决 if-else 过多的问题

前言

if-else基本上是所有高级语言都有的语句。Java,Python,Go,C++。可以说if-else是编程语言种必须用的。基本上大大小小的需求都需要用到if-else。if-else不能滥用，不要让自己变成只会写if-else的程序员。if-else滥用会导致代码可读性，维护成本增高。我们需要使用好if-else来。

1. 使用卫语句,提前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);

2. 表驱动
   对于固定的那种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);

3. 使用策略模式
   场景根据不同的参数走不同的逻辑，这种的一般可以使用策略模式。
   优化前

        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);
    }

4. 使用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过多可以使用以下方法优化

1. 使用卫语句,提前return。减少不必要的嵌套判断。
2. 采用表驱动，映射关系，减少判断。
3. 采用设计模式，比如策略模式，责任链模式。
4. 合理使用ObjectUtils.defaultIfNull ,Optional ,Lambda,Assert减少If判断