策略模式用在统一的入口,但需要根据某个类型判断后续执行逻辑,例如我最近遇到的场景:我需要对接一个设备,前端请求我这边,我再去和设备交互,但设备种类很多,各自有自己的接入规则!传统代码会变成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{
@Override
public 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 {
@Autowired
DesignController designController;
/**
* 策略模式测试
*/
@Test
public void strategyTest() {
designController.strategyTest(StrategyEnums.STRATEGY_ONE);
designController.strategyTest(StrategyEnums.STRATEGY_TWO);
designController.strategyTest(StrategyEnums.STRATEGY_THREE);
}
}
为了方便演示,我是从测试方法里调用的接口,实际上可以用postman走接口执行,或在测试类里使用工厂直接调用。下面查看执行结果:
假设后续新增一个需要对接的设备,我只需要在枚举增加一个类型,在工厂里初始化,然后创建一个新的策略,完成业务代码即可,优雅!实在是优雅~