JAVA策略模式demo【设计模式系列】

发布于:2025-07-11 ⋅ 阅读:(18) ⋅ 点赞:(0)

策略模式用在统一的入口,但需要根据某个类型判断后续执行逻辑,例如我最近遇到的场景:我需要对接一个设备,前端请求我这边,我再去和设备交互,但设备种类很多,各自有自己的接入规则!传统代码会变成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走接口执行,或在测试类里使用工厂直接调用。下面查看执行结果:
在这里插入图片描述
假设后续新增一个需要对接的设备,我只需要在枚举增加一个类型,在工厂里初始化,然后创建一个新的策略,完成业务代码即可,优雅!实在是优雅~
在这里插入图片描述


网站公告

今日签到

点亮在社区的每一天
去签到