过多if - else 的问题, 以及策略模式 + 反射解决方法

策略模式解决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即可。

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