如何在业务代码中优雅的使用策略模式

痛定思痛。 2023-10-15 10:44 14阅读 0赞

# 略模式介绍 #

假设你正在开发一个电商平台，其中涉及到商品的折扣策略。优惠策略有很多种可能，如领取优惠券抵扣、返现促销、拼团优惠等。最初的实现可能会在购物车类中嵌入各种折扣逻辑，导致代码的可维护性和扩展性下降。

下面代码在业务开发中经常遇到，代码满足了业务需求，客户可根据自己的需求选择不同的优惠策略。但是，经过一段时间的业务积累，促销活动会越来越多。于是，程序员就开始经常加班，每次上活动之前都要通宵改代码，而且要做重复测试，判断逻辑可能也会变得越来越复杂。此时，我们要思考代码是否需要重构。

Java复制代码public static void main(String[] args) {
        PromotionActivity promotionActivity = null;
    
        String promotionKey = "COUPON";
    
        if(StringUtils.equals(promotionKey,"COUPON")){
            promotionActivity = new PromotionActivity(new CouponStrategy());
        }else if(StringUtils.equals(promotionKey,"CASHBACK")){
            promotionActivity = new PromotionActivity(new CashbackStrategy());
        }//......
        promotionActivity.execute();
    }

下面我们看下**策略模式**如何优雅的解决这个问题。

策略模式是一种行为型设计模式，它允许在运行时选择算法的一种方式，使得算法可以独立于客户端代码进行变化。在业务代码中使用策略模式可以帮助你实现可维护、可扩展和可变化的代码结构。下面是在业务代码中使用策略模式的一般步骤：

1.  **定义策略接口：** 首先，定义一个策略接口，其中声明了策略的方法或行为。这些方法将在不同的具体策略类中实现。
2.  **创建具体策略类：** 创建实现策略接口的具体策略类，每个类实现了策略接口中的方法。每个具体策略类代表了一个算法或行为的具体实现。
3.  **在客户端代码中使用策略：** 在客户端代码中，通过持有策略接口类型的引用，可以在运行时选择不同的策略实现。这样客户端代码可以根据需要切换或替换不同的策略。

![image.png][]

![get_code_OTc0NTM3YWIyYTM1OTc2MDJlODdmNTlhMzdmMjQxM2EsMTY5MzAxNjk1MDY0Ng][]

# 业务代码中如何使用 #

现在后端项目基本都是基于 Spring Boot 开发的，我们基于 Spring Boot 作为基础框架，教你如何使用 Spring 依赖注入的特性，优雅的实现策略模式。

既然是策略模式，那么定义策略肯定是首当其冲，策略我们使用枚举实现最佳

java复制代码public enum StrategyType {
    
        STRATEGY_A(1, "策略A"),
    
        STRATEGY_B(2, "策略B");
    
        private int code;
    
        private String desc;
    
        StrategyType(int code, String desc) {
            this.code = code;
            this.desc = desc;
        }
    
        public int getCode() {
            return code;
        }
    
        public String getDesc() {
            return desc;
        }
    }

再定义一个接口，接口有两个方法，getType用来获取策略的业务类型，execute用来执行业务

java复制代码public interface Strategy {
    
        void execute();
    
        StrategyType getType();
    
    }

这里我们实现个策略StrategyA

java复制代码@Component("strategyA")
    public class StrategyA implements Strategy {
    
        @Override
        public void execute() {
            System.out.println("Executing strategy A");
        }
    
    
        @Override
        public StrategyType getType() {
            return StrategyType.STRATEGY_A;
        }
    }

再实现个策略StrategyB

Java复制代码@Component("strategyB")
    public class StrategyB implements Strategy {
        @Override
        public void execute() {
            System.out.println("Executing strategy B");
        }
    
        @Override
        public StrategyType getType() {
            return StrategyType.STRATEGY_B;
        }
    
    }

我们通过定义一个工厂类，然后使用 Spring 的依赖注入特性，可以注入一个接口的多个实现，这里采用 List<Strategy> 的形式注入，Spring 也支持通过 Map<String,Strategy> 的形式注入，如果使用 Map 注入，那么 key 就是类名，小伙伴们自己也可以测试一下~

为方便调用我们需要将List<Strategy>转换成Map<StrategyType, Strategy>结构，业务执行时可以直接传递业务参数（这里是我们定义的一个业务枚举StrategyType）获取Bean。这里我们直接使用Spring的@PostConstruct注解在方法上，表示此方法是在Spring实例化该Bean之后马上执行此方法。

Java复制代码@Service
    public class StrategyFactory {
        private Map<StrategyType, Strategy> strategyMap = new ConcurrentHashMap<>();
    
        @Resource
        private List<Strategy> strategyList;
    
        public void execute(StrategyType type) {
            strategyMap.get(type).execute();
        }
    
        @PostConstruct
        public void init() {
    
            for (Strategy strategy : strategyList) {
                strategyMap.put(strategy.getType(), strategy);
            }
        }
    }

最后我们在业务类StrategyService直接使用就行了。

Java复制代码@Service
    public class StrategyService {
    
        @Resource
        private StrategyFactory strategyFactory;
    
        public void executeStrategy(StrategyType type) {
            strategyFactory.execute(type);
        }
    }

最终结构如下所示：

![Strategy.png][]

![get_code_MjhiNWM4MDJmM2I5ZGQ1YWU4OWRkNWM0NDJmZjQyNDgsMTY5MzAxNjk1MDY0Ng][]

# 总结 #

*  使用 Spring 的依赖注入特性，可以注入一个接口的多个实现，很容易就实现了策略模式的选择，这样后续添加一种策略的时候，完全不需要改动主要逻辑，只需添加具体实现即可。
 *  虽然我们是讲策略模式，其实里面还包含了工厂模式。

[image.png]: https://img-blog.csdnimg.cn/img_convert/1cf3e83fd2f8fad204a9560088ef9c79.webp?x-oss-process=image/format,png
[get_code_OTc0NTM3YWIyYTM1OTc2MDJlODdmNTlhMzdmMjQxM2EsMTY5MzAxNjk1MDY0Ng]: https://mp.toutiao.com/mp/agw/article_material/open_image/get?code=OTc0NTM3YWIyYTM1OTc2MDJlODdmNTlhMzdmMjQxM2EsMTY5MzAxNjk1MDY0Ng==
[Strategy.png]: https://img-blog.csdnimg.cn/img_convert/fac80c2fbc747fcecd634035a03a3b68.webp?x-oss-process=image/format,png
[get_code_MjhiNWM4MDJmM2I5ZGQ1YWU4OWRkNWM0NDJmZjQyNDgsMTY5MzAxNjk1MDY0Ng]: https://mp.toutiao.com/mp/agw/article_material/open_image/get?code=MjhiNWM4MDJmM2I5ZGQ1YWU4OWRkNWM0NDJmZjQyNDgsMTY5MzAxNjk1MDY0Ng==