前言

在 设计模式（一）策略模式 —— 策略模式结构 和 设计模式（二）策略模式 —— 在程序中通过枚举使用策略模式 两篇博文中分析了策略模式的基础使用，在实际的项目开发中要结合spring容器使用策略模式，这篇博文以电商优惠券处理为例演示在spring中优雅的使用策略模式。

场景分析

比如在电商系统中，对各种类型的优惠券优惠金额的算法处理。如果直接写代码可能就出现了下面这段代码一样

    public Object couponTypeHandler(String couponType){if ("抵扣券".equals(couponType)){// doSomething}else if ("满减券".equals(couponType)){// doSomething}else if ("折扣券".equals(couponType)){// doSomething}else {// doSomething}return new Object();}

如果优惠券类型越来越多，这里的if...else或者switch的判断条件也会越来越多，后续对这个方法的维护也会越来越困难，也不符合开闭原则。

在spring中优雅的使用策略模式

定义优惠券相关基础类

定义Demo中使用的DTO对象和枚举对象

/*** 优惠券计算传输对象** @author lishuzhen* @createTime 2022年05月29日 22:23:00*/
public class CouponDTO {private CouponTypeEnum couponType;// other fieldpublic CouponTypeEnum getCouponType() {return couponType;}public void setCouponType(CouponTypeEnum couponType) {this.couponType = couponType;}
}/*** 优惠券类型枚举类** @author lishuzhen* @createTime 2022年05月29日 22:28:35*/
public enum CouponTypeEnum {DISCOUNT("折扣券"),REBATE("抵扣券"),FULL_REDUCE("满减券");private String name;CouponTypeEnum(String name) {this.name = name;}
}

定义策略接口

策略接口中包含两个方法的定义。

• 定义当前策略类的适用的优惠券类型
• 定义具体的计算策略交给子类实现。

/*** 优惠券策略接口** @author lishuzhen* @createTime 2022年05月29日 22:48:00*/
public interface CouponStrategy {/*** 适用的优惠券类型** @return*/CouponTypeEnum applyCouponType();/*** 优惠券策略算法执行入口** @param couponDTO* @return*/Object couponHandler(CouponDTO couponDTO);
}

定义优惠券策略计算的上下文对象

策略上下文对象主要完成两项操作

• 应用启动时从spring容器中获取所有策略接口的实现类，注册到优惠券策略集合中
• 提供对外暴露的优惠券计算接口，通过优惠券类型从优惠券策略集合中获取适用的策略类，委派具体的策略类完成计算逻辑。

/*** 优惠券策略上下文对象** @author lishuzhen* @createTime 2022年05月29日 22:58:00*/
@Component
public class CouponStrategyContext implements ApplicationContextAware {/*** 优惠券策略类集合*/private static Map<String, CouponStrategy> couponStrategyMap;/*** 策略上下文对象委派具体的策略执行算法** @param couponDTO* @return*/public static Object couponHandler(CouponDTO couponDTO) {return getCouponStrategy(couponDTO.getCouponType()).couponHandler(couponDTO);}/*** 获取适用的策略处理类** @param couponType* @return*/private static CouponStrategy getCouponStrategy(CouponTypeEnum couponType) {return Optional.ofNullable(couponStrategyMap.get(couponType.name())).orElseThrow(() -> new RuntimeException(String.format("not found coupon type strategy , coupon type is %s", couponType.name())));}/*** 从容器中加载所有优惠券策略接口的实现类，注册到优惠券策略集合中** @param applicationContext* @throws BeansException*/@Overridepublic void setApplicationContext(ApplicationContext applicationContext) throws BeansException {Map<String, CouponStrategy> strategyBeans = applicationContext.getBeansOfType(CouponStrategy.class);if (strategyBeans == null || strategyBeans.size() == 0) {// 无可用的优惠券策略return;}couponStrategyMap = new HashMap<>(strategyBeans.size());for (CouponStrategy strategy : strategyBeans.values()) {couponStrategyMap.put(strategy.applyCouponType().name(), strategy);System.out.println("register strategy " + strategy.applyCouponType());}}
}

定义具体的策略实现类

策略实现类要完成三个操作

• 实现applyCouponType方法，表明当前策略类适用的优惠券类型。供策略上下文对象中注册策略集合使用。
• 实现couponHandler方法，完成自己的主要职责，对优惠券进行计算
• 将自己注册到spring容器中

/*** 折扣券策略** @author lishuzhen* @createTime 2022年05月29日 23:01:00*/
@Service
public class DiscountCouponStrategy implements CouponStrategy {/*** 适用的优惠券类型** @return*/@Overridepublic CouponTypeEnum applyCouponType() {return CouponTypeEnum.DISCOUNT;}/*** 优惠券策略算法执行入口** @param couponDTO* @return*/@Overridepublic Object couponHandler(CouponDTO couponDTO) {System.out.println("执行折扣券策略");return null;}
}/*** 满减券计算策略** @author lishuzhen* @createTime 2022年05月29日 23:01:55*/
@Service
public class FullReduceCouponStrategy implements CouponStrategy {/*** 适用的优惠券类型** @return*/@Overridepublic CouponTypeEnum applyCouponType() {return CouponTypeEnum.FULL_REDUCE;}/*** 优惠券策略算法执行入口** @param couponDTO* @return*/@Overridepublic Object couponHandler(CouponDTO couponDTO) {System.out.println("执行满减券策略");return null;}
}/*** 抵扣券计算策略** @author lishuzhen* @createTime 2022年05月29日 23:02:40*/
@Service
public class RebateCouponStrategy implements CouponStrategy {/*** 适用的优惠券类型** @return*/@Overridepublic CouponTypeEnum applyCouponType() {return CouponTypeEnum.REBATE;}/*** 优惠券策略算法执行入口** @param couponDTO* @return*/@Overridepublic Object couponHandler(CouponDTO couponDTO) {System.out.println("执行抵扣券策略");return null;}
}

调用方代码

调用只需要调用策略上下文对象即可。

        // 模拟业务参数中的优惠券类型CouponDTO couponDTO = new CouponDTO();couponDTO.setCouponType(CouponTypeEnum.REBATE);// 委派优惠券策略上下文对象对优惠券进行计算CouponStrategyContext.couponHandler(couponDTO);

启动应用

注册优惠券策略日志

日志中可以看到，在CouponStrategyContext的setApplicationContext方法中打印的注册日志已经打印到控制台，说明注册优惠券策略集合已完成。

23:51:36.505 [restartedMain] INFO  o.a.c.c.StandardEngine - [log,173] - Starting Servlet engine: [Apache Tomcat/9.0.41]
23:51:36.584 [restartedMain] INFO  o.a.c.c.C.[.[.[/] - [log,173] - Initializing Spring embedded WebApplicationContext
register strategy DISCOUNT
register strategy FULL_REDUCE
register strategy REBATE

调用策略上下文对象，模拟业务场景

在模拟调用方的代码中写了CouponTypeEnum.REBATE类型，控制台正常打印了“执行抵扣券策略”，说明已经匹配到了具体的策略类。

23:55:06.333 [http-nio-8080-exec-1] INFO  o.a.c.c.C.[.[.[/] - [log,173] - Initializing Spring DispatcherServlet 'dispatcherServlet'
执行抵扣券策略

总结

优点

• 利用Spring容器IOC和DI的特性，简化了手动创建策略类实例的过程。
• 调用方无需知道具体的策略类，降低模块间的耦合。
• 方便扩展，只需要新增或者修改某一具体策略类即可，完全符合开闭原则。

缺点

• 暂时没找到缺点...