最近重构项目发现存在巨多的if else ,if …else if … else…这确实是我们在各种项目研发中的一个利器,但是一旦他过于多,就会使逻辑混乱,让代码的可读性大大下降,因而打算使用策略模式来代替过多的if else。
策略模式的介绍
策略模式
引入百科的解释:策略模式作为一种软件设计模式,指对象有某个行为,但是在不同的场景中,该行为有不同的实现算法。比如每个人都要“交个人所得税”,但是“在美国交个人所得税”和“在中国交个人所得税”就有不同的算税方法。
策略模式:顾名思义就是当要在不同的情况下做出该情况下所适用的策略,按照面向对象的是思想就是,就使用继承和多态机制,实现同一行为在不同场景下具备不同实现。
策略模式的本质:分离算法,选择实现。
主要解决
从上述的策略模式的本质中的选择实现是不是可以联想到if…else if… else…没错,在有多种算法相似的情况下,使用 if…else 或 switch…case 所带来的复杂性和臃肿性。
优缺点
优点
策略的多样性,能够自由切换
有效的避免了多重判断,提高了代码的可读性,扩展性,封装性
降低了操作难度以及错误率。
缺点
增加了开发难度,策略类数量增多,并且所要策略类都要对外开放,以便客户端访问
你还在用if else吗?
这里给大家举一个典型的例子
我们每天扫码支付,选用支付方式,选择的微信,支付宝,或者是直接银行卡支付。想必大家已经屡见不鲜了。
那我们现在我们需要做一个支付系统,通过扫描二维码对一个商户进行付款,我们可以选用我们已经绑定的支付宝账号,微信,或者所绑定的银行,直接进行支付,由于每家商户不同,可能付款的时候可能也会有一些优惠政策的不同,并且他们为外界所提供的API也会有所不同。例如,支付宝可以使用红包进行抵用,直接使用银行卡付款,每家银行可能会有不同的折扣。具体情况比较复杂,这里就不做考虑,为了为接下来的策略模式做铺垫,我们这里就简单的分为,微信支付,支付宝支付,银行卡支付,抵用券支付。
我相信很多人拿到需求后,最先考虑的写法就是:
package pay; public class Example { public String selectPayWay(Integer payCode){ if(PayWayEnum.WEICHAT_PAY.equals(payCode)){ //do something return "微信支付成功"; }else if(PayWayEnum.AL_PAY.equals(payCode)){ // do something return "支付宝支付成功"; }else if(PayWayEnum.CARD_PAY.equals(payCode)){ // do something return "银行卡成功"; }else if(PayWayEnum.PONIT_COUPON_PAY.equals(payCode)){ // do something return "优惠券支付成功"; }else { return ""; } } }
可以看出上面四种不同的支付方式在同一个方法体内部,不利于扩展和维护,并且也不符合面向对象设计原则。对以上的代码使用策略模式进行代替,类图如下:
从 上面的UML 类图中,我们可以看到,策略模式 主要包含三种角色:
- 上下文角色(Context):用来操作策略的上下文环境,屏蔽高层模块(客户端)对策略,算法的直接访问,封装可能存在的变化;
- 抽象策略角色(Strategy):规定策略或算法的行为;
- 具体策略角色(ConcreteStrategy):具体的策略或算法实现;
java代码实现具体如下
package strategy; public interface PayStrategy { String selectPayWay(Integer paycode); }
规定具体策略行为
package strategy; public class PayContext { private PayStrategy payStrategy; public String selectPayWay(Integer payCode){ payStrategy = StrategyFactory.getInstance().create(payCode); return payStrategy.selectPayWay(payCode); } public PayStrategy getPayStrategy() { return payStrategy; } public void setPayStrategy(PayStrategy payStrategy) { this.payStrategy = payStrategy; } }
支付宝付款实现类
package strategy.paystrategy; import strategy.PayStrategy; public class ALPayStrategy implements PayStrategy { @Override public String selectPayWay(Integer paycode) { //do something return "支付宝支付成功"; } }
银行卡支付实现类
package strategy.paystrategy; import strategy.PayStrategy; public class CardPayStrategy implements PayStrategy { @Override public String selectPayWay(Integer paycode) { //do something return "银行卡支付成功"; } }
点券支付实现类
package strategy.paystrategy; import strategy.PayStrategy; public class PointCouponPayStrategy implements PayStrategy { @Override public String selectPayWay(Integer paycode) { //do something return "点券支付成功"; } }
微信支付实现类
package strategy.paystrategy; import strategy.PayStrategy; public class WeChatPayStrategy implements PayStrategy { @Override public String selectPayWay(Integer paycode) { //do something return "微信支付成功"; } }
创建策略类的工厂
package strategy; import pay.PayWayEnum; import strategy.paystrategy.ALPayStrategy; import strategy.paystrategy.CardPayStrategy; import strategy.paystrategy.PointCouponPayStrategy; import strategy.paystrategy.WeChatPayStrategy; import java.util.HashMap; import java.util.Map; public class StrategyFactory { private static StrategyFactory factory = new StrategyFactory(); private StrategyFactory(){} private static Map strategyMap = new HashMap<>(16); static { strategyMap.put(PayWayEnum.AL_PAY.getCode(),new ALPayStrategy()); strategyMap.put(PayWayEnum.WEICHAT_PAY.getCode(),new WeChatPayStrategy()); strategyMap.put(PayWayEnum.CARD_PAY.getCode(),new CardPayStrategy()); strategyMap.put(PayWayEnum.PONIT_COUPON_PAY.getCode(),new PointCouponPayStrategy()); } public PayStrategy create(Integer payCode){ return (PayStrategy) strategyMap.get(payCode); } public static StrategyFactory getInstance(){ return factory; } }
实现客户端
package strategy; public class Client { public static void main(String[] args) { PayContext payContext = new PayContext(); System.out.println(payContext.selectPayWay(100)); } }
最后在贴一个枚举类
package pay; public enum PayWayEnum { /**微信支付*/ WEICHAT_PAY("微信支付",100), /**支付宝支付*/ AL_PAY("支付宝支付",101), /**银行卡支付*/ CARD_PAY("银行卡支付",102), /**点券支付*/ PONIT_COUPON_PAY("点券支付",103); private String msg; private Integer code; PayWayEnum(String msg, Integer code) { this.msg = msg; this.code = code; } public String getMsg() { return msg; } public Integer getCode() { return code; } }