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

时间:2024-01-28 18:27:25
  1. 策略模式解决if - else 的代码

业务场景:

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

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

原来场景:

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

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

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

解决过程:

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

  2. 尝试策略模式

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

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

解决方法:

  1. 混合使用策略模式和反射来解决
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即可。

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