过多if

策略模式解决if - else 的代码

业务场景：

外包企业的审批人需要审批打卡的场景；

审批人分为多种不同的级别，多种级别中具有方式相同但是内容不同的操作：审批。

原来场景：

有前端传来审批人参数，使用if-else 来判断该审批人的级别属于哪一个级别，执行相应的审批方法。每一个审批方法写在了业务类底，命名采用1级审批，2级审批---等等的命名方式来命名。

问题：if - else 逻辑复杂，不易阅读，函数中审批能抽象的地方未抽象，

两个核心问题：test测试非常麻烦，新增审批人员需要对代码进行修改，违背开闭原则。

解决过程：

抽象审批中的原子操作的代码，比如查询对应级别所审批的同学名单。代码变得好看一些，但是任然未解决，if - else 和多个审批方法的实现。

尝试策略模式

使用策略模式将每个级别的用户抽象起来

package strategy;



public class Main {
    public static void main(String[] args) {
        if (LevelEnum.LEVEL1.equals(1)) {
            SupervisorControl supervisorControl = new SupervisorControl(new Supervisor1());
            supervisorControl.execute(1);
        }
        if (LevelEnum.LEVEL2.equals(2)) {
            SupervisorControl supervisorControl = new SupervisorControl(new Supervisor2());
            supervisorControl.execute(2);
        }
        if (LevelEnum.LEVEL3.equals(3)) {
            SupervisorControl supervisorControl = new SupervisorControl(new Supervisor3());
            supervisorControl.execute(3);
        }
    }
}

package strategy;

public enum LevelEnum {
    LEVEL1, LEVEL2, LEVEL3
}

package strategy;

public interface Supervisor {
    void examine(int supervisorId);
}

package strategy;

public class Supervisor1 implements Supervisor {
    public void examine(int supervisorId) {
        System.out.println("去做一些和1级supervisor相关的工作");
    }
}

package strategy;

public class Supervisor2 implements Supervisor {
    public void examine(int supervisorId) {
        System.out.println("去做一些和2级supervisor相关的工作");
    }
}

package strategy;

public class Supervisor3 implements Supervisor {
    public void examine(int supervisorId) {
        System.out.println("去做一些和3级supervisor相关的工作");
    }
}

package strategy;

public class SupervisorControl {
    Supervisor supervisor;
    public SupervisorControl(Supervisor supervisor) {
        this.supervisor = supervisor;
    }
    public void execute(int id) {
        supervisor.examine(id);
    }
}

解决方法：

混合使用策略模式和反射来解决

package strategy;



public class Main {
    public static void main(String[] args) {
//        if (LevelEnum.LEVEL1.equals(1)) {
//            SupervisorControl supervisorControl = new SupervisorControl(new Supervisor1());
//            supervisorControl.execute(1);
//        }
//        if (LevelEnum.LEVEL2.equals(2)) {
//            SupervisorControl supervisorControl = new SupervisorControl(new Supervisor2());
//            supervisorControl.execute(2);
//        }
//        if (LevelEnum.LEVEL3.equals(3)) {
//            SupervisorControl supervisorControl = new SupervisorControl(new Supervisor3());
//            supervisorControl.execute(3);
//        }
        String packageName = Supervisor1.class.getPackage().getName();
        String supervisorName = packageName + "." + "Supervisor" + "2";
        try {
            Class<?> clazz = Class.forName(supervisorName);
            Supervisor supervisor = (Supervisor) clazz.newInstance();
            supervisor.examine(1);
        } catch (Exception e) {
            System.out.println(e);
        }

    }
}

注意

class.forName() 需要传入的格式是"包名.类名"。如果找不到包名则会报ClassNotFoundException

可以通过class.getPackage来获取包名。

到此我们解决了if - else 很多的问题，并且解决了需要侵入代码修改的问题，如果新增supervisor的级别，只需要和前端达成一致，后台继续写一个supervisor4对象即可。

为什么解决了test难的问题

对于test来说，每一个If-else 都需要我们去验证，这其实就是两个test, if 一个 else一个，如果有很多if - else，想要保证高的test覆盖率，就会非常头痛。

然而我们用策略模式和反射来解决，只需要，对主逻辑一个test，每一个实现方法做一个test即可。

至此，所有问题都解决了，代码一片光明。