优雅替代if else判断逻辑

参考内容： 去除多余的if else, 5种方式你知道几种， 优化if else的几种方式

最初的逻辑

// 根据员工类型，输出不同的工资
public void getSalary(String person){
if(StringUtils.equals("BOSS", person)){
System.out.println("Boss salary 10000000");
}else if(StringUtils.equals("MANAGER", person)){
System.out.println("Manager salary 10000");
}else if(StringUtils.equals("STAFF", person)){
System.out.println("Staff salary 100");
}
}

工厂模式解决方案

1. 新建抽象类，有获取工资的抽象方法

   public abstract class Salary {
   abstract void getSalary();
   }

2. 三个不同的类 实现该方法

   public class Boss extends Salary {
   @Override
   public void getSalary() {
   System.out.println("Boss salary 10000000");
   }
   }

   public class Manager extends Salary {
   @Override
   public void getSalary() {
   System.out.println("Manager salary 10000");
   }
   }

   public class Staff extends Salary {
   @Override
   public void getSalary() {
   System.out.println("Staff salary 100");
   }
   }

3. 静态工厂方式，提供获取工资方法调用

   public class TestSalary {
   static Map<String, Salary> data = new HashMap<>();
   static {
   data.put("BOSS", new Boss());
   data.put("MANAGER", new Manager());
   data.put("STAFF", new Staff());
   }

   public static void main(String[] args) {
   getSalary("MANAGER");
   }

   static void getSalary(String person){
   data.get(person).getSalary();
   }
   }

枚举解决方案

1. 枚举类 内有抽象方法以及不同枚举值的实现

   public enum Salary {
   BOSS{
   @Override
   void getSalary() {
   System.out.println("Boss salary 10000000");
   }
   },
   MANAGER{
   @Override
   void getSalary() {
   System.out.println("Manager salary 10000");
   }
   },
   STAFF{
   @Override
   void getSalary() {
   System.out.println("Staff salary 100");
   }
   };

   abstract void getSalary();
   }

2. 直接调用该枚举的抽象方法即可

   public class TestSalary {

   public static void main(String[] args) {
   getSalary(Salary.valueOf("BOSS"));
   }

   public static void getSalary(Salary salary){
   salary.getSalary();
   }
   }

springboot 策略模式注解方式

1. 新建抽象类

   public abstract class Salary {
   abstract void getSalary();
   }

2. 三个不同实现，实现该方法

   @Service("BOSS")
   public class Boss extends Salary {
   @Override
   public void getSalary() {
   System.out.println("Boss salary 10000000");
   }
   }

   @Service("MANAGER")
   public class Manager extends Salary {
   @Override
   public void getSalary() {
   System.out.println("Manager salary 10000");
   }
   }

   @Service("STAFF")
   public class Staff extends Salary {
   @Override
   public void getSalary() {
   System.out.println("Staff salary 100");
   }
   }

3. 上下文切换类

   @Component("salaryContext")
   public class SalaryContext {
   @Autowired
   private final Map<String, Salary> salaryMap = new ConcurrentHashMap<>(3);
   public void getSalary(String serviceName) {
   salaryMap.get(serviceName).getSalary();
   }
   }

4. 测试类

   @RunWith(SpringRunner.class)
   @SpringBootTest(classes = Application.class)
   public class ApplicationTests {
   @Resource
   private SalaryContext salaryContext;

   @Test
   public void contextLoads() {
   }

   @Test
   public void getSalary() {
   salaryContext.getSalary("BOSS");
   }
   }

注意：调试中，出现bean注入失败的原因，可以从以下顺序找问题(参考内容1， 参考内容2)：

1. 首先检测测试类包名和启动类包名是否一致
2. @SpringBootTest(classes = Application.class) application为该项目的正常启动类
3. @Test导包的是org.junit.Test
4. Application记得配置扫描目录 @SpringBootApplication(scanBasePackages = {"com.xxx"})