前言
传统的实现方式,通过传统if代码判断。这样就会导致后期的维护性非常差。当后期需要新增计费方式,还需要在这里再加上if(),也不符合设计模式的开闭原则。
一、策略模式简介
策略模式是一种行为型模式,它将对象和行为分开,将行为定义为 一个行为接口 和 具体行为的实现。策略模式最大的特点是行为的变化,行为之间可以相互替换。
策略模式定义一系列算法,把它们一个个封装起来,并且使它们可相互替换。本模式使得算法可独立于使用它的客户而变化。
二、代码实现
场景:支付接口自动找到对应的实现类完成支付
1.定义一个支付接口
代码如下(示例):
public interface PayService {
void pay();
}
2.定义策略上下文
这里通过@Service
定义注册到容器,SpringBoot会自动获取实现PayService接口的实现类列表。
@Service
public class PayStrategyContext {
private final Map<String, PayService> strategyServiceMap = new ConcurrentHashMap<>();
public PayStrategyContext(Map<String, PayService> strategyMap) {
this.strategyServiceMap.clear();
strategyMap.forEach((k, v) -> this.strategyServiceMap.put(k, v));
}
public PayService getResource(String payType) {
return strategyServiceMap.get(payType);
}
}
3.定义支付类型实现类
通过@Component
注解设置组件名,后续会通过组件名来获取对应的实现类
@Component("wxPay")
public class WxPayService implements PayService {
@Override
public void pay() {
System.out.println("微信支付。。。");
}
}
@Component("aliPay")
public class AliPayService implements PayService {
@Override
public void pay() {
System.out.println("支付宝支付。。。");
}
}
4.测试
@Autowired
private PayStrategyContext payStrategyContext;
@Test
public void test() {
payStrategyContext.getResource("wxPay").pay();
payStrategyContext.getResource("aliPay").pay();
}
控制台输出:
总结
使用策略模式可以避免使用多重条件if…else if…else语句, 多重条件不易维护且代码可读性差