学习笔记-设计模式之策略模式

本文内容源于视频教程，若有侵权，请联系作者删除。

一、概念

策略模式（Strategy Pattern）是指定义了算法家族、分别封装起来，让它们之间可以互相替换，此模式让算法的变化不会影响到使用算法的用户。

简言之：为达到某种目的有多个方案，策略模式就是将这些方案封装起来，以便使用。

二、实现

需求：实现促销活动（无促销，限时，优惠券），用户可以任意选择一种优惠策略。

创建促销活动抽象类，并持有开启活动方法。

/**
 * 活动策略
 */
public interface PromotionStrategy {

    /**
     * 执行活动
     */
    void execute();
}

创建无促销，限时，优惠券活动并实现活动抽象类

public class CouponPromotion implements PromotionStrategy {
    @Override
    public void execute() {
        System.out.println("优惠券活动");
    }
}

public class DiscountPromotion implements PromotionStrategy {
    @Override
    public void execute() {
        System.out.println("限时折扣活动");
    }
}

public class EmptyPromotion implements PromotionStrategy {
    @Override
    public void execute() {
        System.out.println("无促销活动");
    }
}

编写测试类

public class PromotionStrategyTest {

    public static void main(String[] args) {
        String promotionKey = "COUPON";
        PromotionStrategy promotionStrategy = getPromotionStrategy(promotionKey);
        promotionStrategy.execute();
    }
    
    private static PromotionStrategy getPromotionStrategy(String promotionKey){
        PromotionStrategy promotionStrategy;
        if ("COUPON".equals(promotionKey)){
            promotionStrategy = new CouponPromotion();
        } else if ("DISCOUNT".equals(promotionKey)){
            promotionStrategy = new DiscountPromotion();
        } else {
            promotionStrategy = new EmptyPromotion();
        }
        return promotionStrategy;
    }
}

执行结果：

优惠券活动

由此可见，策略模式是把限时，优惠券活动封装起来，由调用方决定使用哪一种活动。

然而，当活动增加时，每个调用方要新增else if判断，违背了开闭原则，下面通过单例+工厂去掉if else判断。

新增工厂类

public class PromotionStrategyFactory {

    private static Map<String, PromotionStrategy> promotionMap = new HashMap<>();

    static {
        promotionMap.put("COUPON", new CouponPromotion());
        promotionMap.put("DISCOUNT", new DiscountPromotion());
    }

    private PromotionStrategyFactory() {
    }

    public static PromotionStrategy getInstance(String promotionKey){
        PromotionStrategy promotionStrategy = promotionMap.get(promotionKey);
        if (null == promotionStrategy){
            promotionStrategy = new EmptyPromotion();
        }
        return promotionStrategy;
    }

}

测试类

public class PromotionStrategyTest {

    public static void main(String[] args) {
        String promotionKey = "coupon";
        PromotionStrategy activity = PromotionStrategyFactory.getInstance(promotionKey);
        activity.execute();
    }

}