JAVA策略模式demo【设计模式系列】
策略模式用在统一的入口,但需要根据某个类型判断后续执行逻辑,例如我最近遇到的场景:我需要对接一个设备,前端请求我这边,我再去和设备交互,但设备种类很多,各自有自己的接入规则!传统代码会变成if{}else{}面条代码,而且扩展性不佳,如果后续新增了设备,需要去修改这里的代码,非常不优雅~
闲话少说上代码!
1.创建一个枚举,包含所有策略
public enum StrategyEnums {STRATEGY_ONE("strategyOne", "策略1"),STRATEGY_TWO("strategyTwo", "策略2"),STRATEGY_THREE("strategyThree", "策略3");private String code;private String name;StrategyEnums(String code, String name) {this.code = code;this.name = name;}
}
2.创建一个接口,含执行方法
public interface StrategyService {void run();
}
3.创建3个策略的业务类,实现【StrategyService】接口
public class StrategyServiceImpl1 implements StrategyService{@Overridepublic void run() {System.out.println("运行策略1的业务代码");}
}
4.创建工厂类,用于初始化各策略,提供获取策略的静态方法
public class StrategyFactory {static final Map<StrategyEnums, StrategyService> map = new HashMap<>();static {map.put(StrategyEnums.STRATEGY_ONE, new StrategyServiceImpl1());map.put(StrategyEnums.STRATEGY_TWO, new StrategyServiceImpl2());map.put(StrategyEnums.STRATEGY_THREE, new StrategyServiceImpl3());}public static StrategyService getStrategy(StrategyEnums strategyEnums) {return map.get(strategyEnums);}
}
5.创建控制器,对前端提供接口【可选】
@RestController("design")
public class DesignController {@GetMapping("strategy")public void strategyTest(StrategyEnums strategyEnums) {StrategyService strategy = StrategyFactory.getStrategy(strategyEnums);strategy.run();}
}
6.创建测试类,调用接口【可选】
@SpringBootTest
public class DesignTest {@AutowiredDesignController designController;/*** 策略模式测试*/@Testpublic void strategyTest() {designController.strategyTest(StrategyEnums.STRATEGY_ONE);designController.strategyTest(StrategyEnums.STRATEGY_TWO);designController.strategyTest(StrategyEnums.STRATEGY_THREE);}
}
为了方便演示,我是从测试方法里调用的接口,实际上可以用postman走接口执行,或在测试类里使用工厂直接调用。下面查看执行结果:
假设后续新增一个需要对接的设备,我只需要在枚举增加一个类型,在工厂里初始化,然后创建一个新的策略,完成业务代码即可,优雅!实在是优雅~