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{

    @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走接口执行，或在测试类里使用工厂直接调用。下面查看执行结果：

假设后续新增一个需要对接的设备，我只需要在枚举增加一个类型，在工厂里初始化，然后创建一个新的策略，完成业务代码即可，优雅！实在是优雅~