别再写满屏的 if、else 了，试试策略模式，真香

Java技术栈 04-12 7524

前言：

此刻小伙伴们对“java可以只写if不写else吗”大约比较着重，大家都想要分析一些“java可以只写if不写else吗”的相关资讯。那么小编在网上搜集了一些有关“java可以只写if不写else吗””的相关文章，希望小伙伴们能喜欢，我们一起来了解一下吧！

你还在写满屏的 if/ else/ switch 之类的判断逻辑吗？

栈长在开发人员的代码中看过太多这样的低级代码了，真的太 low，极不好维护，本文栈长就教你如何用策略模式干掉 if/ else/ switch，让你的代码更优雅。

什么是策略模式？

比如说对象的某个行为，在不同场景中有不同的实现方式，这样就可以将这些实现方式定义成一组策略，每个实现类对应一个策略，在不同的场景就使用不同的实现类，并且可以自由切换策略。

策略模式结构图如下：

策略模式需要一个策略接口，不同的策略实现不同的实现类，在具体业务环境中仅持有该策略接口，根据不同的场景使用不同的实现类即可。

面向接口编程，而不是面向实现。

策略模式的优点：

1、干掉繁琐的 if、switch 判断逻辑；

2、代码优雅、可复用、可读性好；

3、符合开闭原则，扩展性好、便于维护；

策略模式的缺点：

1、策略如果很多的话，会造成策略类膨胀；

2、使用者必须清楚所有的策略类及其用途；

策略模式实战

举个实际的例子，XX 公司是做支付的，根据不同的客户类型会有不同的支付方式和支付产品，比如：信用卡、本地支付，而本地支付在中国又有微信支付、支付宝、云闪付、等更多其他第三方支付公司，这时候策略模式就派上用场了。

传统的 if/ else/ switch 等判断写法大家都会写，这里就不贴代码了，直接看策略模式怎么搞！

1、定义策略接口

定义一个策略接口，所有支付方式的接口。

策略接口：

/** * 支付接口 * @author: 栈长 * @from: Java技术栈 */public interface IPayment {    /**     * 支付     * @param order     * @return     */    PayResult pay(Order order);}

订单信息类：

/** * 订单信息 * @author: 栈长 * @from: Java技术栈 */@Datapublic class Order {    /**     * 金额     */    private int amount;    /**     * 支付类型     */    private String paymentType;}

返回结果类：

/** * @author: 栈长 * @from: Java技术栈 */@Data@AllArgsConstructorpublic class PayResult {    /**     * 支付结果     */    private String result;}

2、定义各种策略

定义各种支付策略，微信支付、支付宝、云闪付等支付实现类都实现这个接口。

微信支付实现：

/** * 微信支付 * @author: 栈长 * @from: Java技术栈 */@Service("WechatPay")public class WechatPay implements IPayment {    @Override    public PayResult pay(Order order) {        return new PayResult("微信支付成功");    }}

支付宝实现：

/** * 支付宝 * @author: 栈长 * @from: Java技术栈 */@Service("Alipay")public class Alipay implements IPayment {    @Override    public PayResult pay(Order order) {        return new PayResult("支付宝支付成功");    }}

云闪付实现：

/** * 银联云闪付 * @author: 栈长 * @from: Java技术栈 */@Service("UnionPay")public class UnionPay implements IPayment {    @Override    public PayResult pay(Order order) {        return new PayResult("云闪付支付成功");    }}

这里我把所有支付方式类都用 @Service 注解生成 Bean 放入 Spring Bean 容器中了，在使用策略的时候就不用 new 支付对象了，可以直接使用 Bean，这样更贴近业务。Spring 基础教程就不介绍了，大家可以关注Java技术栈，历史教程我都整理好了。

3、使用策略

有的文章使用了枚举、HashMap 的方式来根据策略名称映射策略实现类，这样是没有问题，但在使用了 Spring 框架的项目还是有点多此一举，完全可以发挥 Spring 框架的优势，使用 Bean 名称就能找到对应的策略实现类了。

参考示例代码如下：

/** * 支付服务 * @author: 栈长 * @from: Java技术栈 */@RestControllerpublic class PayService {    @Autowired    private ApplicationContext applicationContext;    /**     * 支付接口     * @param amount     * @param paymentType     * @return     */    @RequestMapping("/pay")    public PayResult pay(@RequestParam("amount") int amount,                    @RequestParam("paymentType") String paymentType) {        Order order = new Order();        order.setAmount(amount);        order.setPaymentType(paymentType);        // 根据支付类型获取对应的策略 bean        IPayment payment = applicationContext.getBean(order.getPaymentType(), IPayment.class);        // 开始支付        PayResult payResult = payment.pay(order);        return payResult;    }}

看示例代码，我并没有像策略模式结构图中那样新建一个 Context 类持有策略接口，那是标准的策略模式，其实道理是一样的，关键是怎么施放策略。

测试一下：

...

测试 OK，传入不同的支付方式会调用不同的策略。

本节教程所有实战源码已上传到这个仓库：

策略模式在 JDK 中的应用

现在我们知道如何使用策略模式了，现在我们再看看 JDK 哪些地方运用了策略模式呢。

1、线程池中的拒绝策略

线程池的构造中有一个拒绝策略参数，默认是默认拒绝策略：

其实这就是一个策略接口：

下面有几种拒绝策略的实现：

在创建线程池的时候，就可以传入不同的拒绝策略，这就是 JDK 中策略模式的经典实现了。