使用策略模式代替繁琐的if else

前言

这两天在知乎上看到一篇文章名叫《一个程序员的水平能差到什么程度?》这上面写的代码确实让看的人当做饭后笑话的谈资，让下一个接盘的人又磨刀霍霍向猪羊的冲动

里面大部分的吐槽都是针对于if else 无作为的使用以及滥用的吐槽，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 somethingreturn "微信支付成功";}else if(PayWayEnum.AL_PAY.equals(payCode)){// do somethingreturn "支付宝支付成功";}else if(PayWayEnum.CARD_PAY.equals(payCode)){// do somethingreturn "银行卡成功";}else if(PayWayEnum.PONIT_COUPON_PAY.equals(payCode)){// do somethingreturn "优惠券支付成功";}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 {@Overridepublic String selectPayWay(Integer paycode) {//do somethingreturn "支付宝支付成功";}
}

银行卡支付实现类

package strategy.paystrategy;import strategy.PayStrategy;public class CardPayStrategy implements PayStrategy {@Overridepublic String selectPayWay(Integer paycode) {//do somethingreturn "银行卡支付成功";}
}

点券支付实现类

package strategy.paystrategy;import strategy.PayStrategy;public class PointCouponPayStrategy implements PayStrategy {@Overridepublic String selectPayWay(Integer paycode) {//do somethingreturn "点券支付成功";}
}

微信支付实现类

package strategy.paystrategy;import strategy.PayStrategy;public class WeChatPayStrategy implements PayStrategy {@Overridepublic String selectPayWay(Integer paycode) {//do somethingreturn "微信支付成功";}}

创建策略类的工厂

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;}
}

从上面代码中，我们可以看到，我们完全消除了对状态码进行判断的那些if…else的冗余代码，取而代之是直接由客户端决定使用哪种算法，然后交由上下文获取结果。极大的增强的了扩展性，隐藏了内部实现的细节，当然我们也可以看到，策略模式由于独立策略的实现，也使得系统内部增加了很多策略，对客户端来必须知道这些策略类实现的是那些策略，而且需要知道具体的策略条件。

然而在实际的生产环境中我们可能不会像示例代码这样干，在实际的生产环境中我们一般会结合spring来对我们所需的策略进行操作，这边我可以给大家有个具体的思路，大家可以通过这个思路来想想，自己在实际生产环境会怎样操作

整体思路：

首先和上面代码一样定义一个Strategy接口，其中有一个operation的函数交给具体业务去实现
根据自己的业务，会拥有很多的策略类实现该接口；将这些类注册到spring Bean容器中提供使用。
通过客户端传值，从spring容器中获取Strategy实例
最终来执行operation函数

好了这次的分享就到此结束啦，下次我在真正的开发项目中使用到了策略模式代替if else的话我会结合我的项目在给大家看下怎样在实际的生产环境中应用，感谢大家的阅读。