设计模式之策略模式

定义

概念：
定义定义一系列的算法,把它们一个个封装起来, 并且使它们可相互替换，此模式让算法的变化，不会影响到使用算法的客户。

通俗的一点来讲，当同一个操作，可能有不同的表现或者动作的时候，我们可以将这个操作抽象出来，然后将具体的表现或者动作分别的封装起来，并且都实现抽象的操作，这样我们就可以在执行这个操作的时候，根据不同的表现类型，来执行不同表现或者动作。

策略模式结构可分为，策略接口，具体策略实现和策略容器。

策略接口用于抽象定义具体的操作。
策略实现用于定义不同策略下的具体实现。
策略容器用于存放策略和执行策略。

策略模式结构图

在这里插入图片描述

应用场景

我们在电商平台购物，到支付支付的时候，通常会有很多的支付方式可以选择，比如说有：支付宝、微信、银联云闪付等等。

这里我们就可以思考一下，电商平台在执行支付这个操作的时候，是如何根据我们选择的支付方式来判断和执行之后的操作的呢？

最容易想到的，可能就是在代码中根据用户选择的支付方式进行一系列的if else 操作了把。

if (Objects.equals(paymentWay, "支付宝")) {
    支付宝支付处理逻辑...
} else if (Objects.equals(paymentWay, "微信")) {
    微信支付处理逻辑...
} else if (Objects.equals(paymentWay, "银联云闪付")) {
    银联云闪付支付处理逻辑...
} else if (Objects.equals(paymentWay, "xxx")) {
    xxx支付处理逻辑...
}

看看上面的代码是不是觉得头晕胸闷。每个if else中都有大量的逻辑处理，新增一个支付方式，就需要新增一个 if else，这样写了之后，不仅代码的可读性低，而且后期的维护成本也是大大的增高了。而且每次的修改出错的几率也是比较高。

所以这里就适合使用策略模式来进行一波重构。

定义策略接口

我们可以把支付这个操作抽象为一个策略接口。

public interface Payment {

    void payment();
}

定义策略具体实现

支付宝支付：

public class AliPayPayment implements Payment {

    @Override
    public void payment() {
        System.out.println("支付宝支付");
    }
}

微信支付：

public class WechatPayPayment implements Payment {
    @Override
    public void payment() {
        System.out.println("微信支付");
    }
}

银联云闪付：

public class UnionPayPayment implements Payment {
    @Override
    public void payment() {
        System.out.println("银联云闪付");
    }
}

定义策略容器类

public class PaymentContext {

    private Payment payment;

    public PaymentContext(Payment payment) {
        this.payment = payment;
    }

    public void payment(){
        payment.payment();
    }
}

定义好了之后，现在再看支付的时候，我们只需要根据不同的支付方式，初始化策略容器并传入对应支付方式实现，然后调用策略容器的 payment() 方法即可。

public static void main(String[] args) {
    //支付宝支付
    PaymentContext aliPay = new PaymentContext(new AliPayPayment());
    aliPay.payment();
    //微信支付
    PaymentContext wechatPay = new PaymentContext(new WechatPayPayment());
    wechatPay.payment();
    //银联云闪付
    PaymentContext unionPay = new PaymentContext(new UnionPayPayment());
    unionPay.payment();
}

策略模式和简单工厂结合

public class PaymentContext {

    private Payment payment;

    public PaymentContext(String payType) {
    	switch(payType){
    		case "alipay":
    			this.payment = new AliPayPayment();break;
    		case "wechatPay":
    			this.payment = new WechatPayPayment();break;
   			case "unionPay":
    			this.payment = new UnionPayPayment();break;
    		default:
    			break;
		}
    }

    public void payment(){
        payment.payment();
    }
}

这样做了之后，虽说我们是将不同的支付方式的实现都封装到了具体的实现类中，都独立了出来，但是我们在确认初始化策略容器的时候到底应该传如那种支付方式的实现的时候，我们还是要去先判断一下支付方式，然后才能明确到底应该传入那种支付方式的实现，那么应该如何简化一下呢？

改造策略模式

策略接口

新增一个支付方式枚举：

public enum PaymentWay {
    ALI_PAY, //支付宝支付
    WECHAT_PAY,	//微信支付
    UNION_PAY 	//银联云闪付支付
}

修改策略接口，新增一个获取支付方式的方法：

public interface Payment {
    void payment();
    PaymentWay getPaymentWay();
}

具体策略实现

具体的支付方式的实现也加上 getPaymentWay()方法的实现，并且返回对应的支付方式枚举。

支付宝支付：

public class AliPayPayment implements Payment {

    @Override
    public void payment() {
        System.out.println("支付宝支付");
    }

	@Override   
	public PaymentWay getPaymentWay() {
    	return PaymentWay.ALI_PAY;
 	}
 }

微信支付：

public class WechatPayPayment implements Payment {

    @Override
    public void payment() {
        System.out.println("微信支付");
    }
    
   @Override
   public PaymentWay getPaymentWay() {
       return PaymentWay.WECHAT_PAY;
   }
}

银联云闪付：

public class UnionPayPayment implements Payment {

    @Override
    public void payment() {
        System.out.println("银联云闪付");
    }

  @Override
   public PaymentWay getPaymentWay() {
       return PaymentWay.UNION_PAY;
   }
}

改造策略容器

public class PaymentContext {
	private Payment payment;

	/**
     * 将所有实现了 {@link Payment} 接口的实现都放进来
     * 具体怎么获取接口的所有实现，这里就不展开细讲了
     * 如果使用了 Spring 的话，直接使用 {@link Autowired} 注解即可将所有的实现注入进来
     */
	private List<Payment> payment;
	private Map<PaymentWay, Payment> paymentProvider;

	public PaymentContext() {
		payments.forEach(payment -> {
				paymentProvider.put(payment.getPaymentWay(), payment);
		});
	}

	public void payment(PaymentWay paymentWay){
		paymentProvider.get(paymentWay).payment();
	}
}

改造完了之后，我们只需要初始化一次策略容器，然后调用 payment 方法的时候，直接传入支付方式，就能自动的匹配到该使用哪个支付方式的实现了。

借助Spring容器实现获取接口的所有实现类

方法1，使用注解：

import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;

@Component
public class ServiceLocator implements ApplicationContextAware{
    /**
     * 用于保存接口实现类名及对应的类
     */
    private Map<String, IService> map;

    /**
     * 获取应用上下文并获取相应的接口实现类
     * @param applicationContext
     * @throws BeansException
     */
    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        //根据接口类型返回相应的所有bean
        Map<String, IService> map = applicationContext.getBeansOfType(IService.class);
    }

    public Map<String, IService> getMap() {
        return map;
    }
}

方法2，实现接口所有类集合（实现接口）:

/**
 * @author: tianyong
 * @Time: 2019/6/26 11:17
 * @description:服务工厂(主要用于动态注入接口实现类)
 */
@Component
public class ServiceFactory implements ApplicationContextAware {
 
 
    //定义成员变量
    private static Map<String,ChartService> res;
 
 
    /**
      * @author: tianyong
      * @time: 2019/6/27 16:04
      * @description:设置上下文参数
      */
    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        res = applicationContext.getBeansOfType(ChartService.class);
    }
 
 
    /**
      * @author: tianyong
      * @time: 2019/6/27 16:05
      * @description:根据标记返回当前接口实现类
      */
    public static <T extends ChartService> T getServices(String flag){
        return (T)res.get(flag);
    }
}

参考：
（1）Java设计模式-策略模式
（2）大话设计模式
（3）Java获取接口所有实现类的方式

posted @ 2022-10-30 23:22 寒小韩阅读(51) 评论(0) 编辑收藏举报

刷新页面返回顶部

寒小韩

dream what you want to dream, do what you want to do!