策略模式：巧妙替代你的if-else

点击加入：

后端技术内卷群，一起学习！

什么是策略模式？

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

策略模式结构图如下：

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

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

策略模式的优点：

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

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

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

策略模式的缺点：

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

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

策略模式实战

举个实际的例子，某个下单选择支付方式场景，用户下单之后，跳转支付页面，此时有多种支付方式和支付产品可选。比如：使用信用卡支付和本地支付，而本地支付又分为支付宝支付，微信支付，云闪付支付等，甚至还有一些第三方支付，如果使用普通的if else写法，那这一块的代码将会非常复杂，并且后期增加了新的支付方式后，也不好维护。所以这里采用Java的23种设计模式之一的策略模式来解决这个问题。

传统的 if/ else/ switch 等判断写法比较简单，这里就不贴代码了，直接看策略模式

1.定义策略接口

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

策略接口

/**
 * @author fei
 * @version 1.0
 * @title 支付接口
 * @date 2021/5/17 17:59
 * @Description:
 */
public interface IPayment {
    /**
     * 支付
     * @param order
     * @return PayResult
     */
    PayResult pay(Order order);
}

订单信息类

/**
 * 订单信息
 * @author fei
 */
@Data
public class Order {

    /**
     * 金额
     */
    private int amount;

    /**
     * 支付类型
     */
    private String paymentType;

}

返回结果类

/**
 * 支付返回结果类
 * @author fei
 */
@Data
@AllArgsConstructor
public class PayResult {

    /**
     * 支付结果
     */
    private String result;

}

2.定义各种策略

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

微信支付实现

/**
 * 微信支付实现
 * @author fei
 */
@Service("WechatPay")
public class WechatPay implements IPayment {

    @Override
    public PayResult pay(Order order) {
        return new PayResult("微信支付成功");
    }

}

支付宝实现

/**
 * 支付宝支付
 * @author fei
 */
@Service("Alipay")
public class Alipay implements IPayment {

    @Override
    public PayResult pay(Order order) {
        return new PayResult("支付宝支付成功");
    }

}

云闪付实现

/**
 * 银联云闪付
 * @author fei
 */
@Service("UnionPay")
public class UnionPay implements IPayment {

    @Override
    public PayResult pay(Order order) {
        return new PayResult("云闪付支付成功");
    }

}

这里我把所有支付方式类都用 @Service 注解生成 Bean 放入 Spring Bean 容器中了，在使用策略的时候就不用 new 支付对象了，可以直接使用 Bean，这样更贴近业务。

3.使用策略

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

参考示例代码如下：

/**
 * 支付服务
 * @author fei
 */
@RestController
public class PayController {

    @Autowired
    private ApplicationContext applicationContext;

    /**
     * 支付接口
     * @param amount
     * @param paymentType
     * @return
     */
    @ApiOperation("测试策略模式支付方式选择")
    @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;
    }

}