告别满屏的if/else，用了策略模式后让你的代码更优雅

你还在写满屏的 if/ else/ switch 之类的判断逻辑吗？
博主在开发人员的代码中看过太多这样的低级代码了，真的太 low，极不好维护，本文博主就教你如何用策略模式干掉 if/ else/ switch，让你的代码更优雅。

什么是策略模式？
比如说对象的某个行为，在不同场景中有不同的实现方式，这样就可以将这些实现方式定义成一组策略，每个实现类对应一个策略，在不同的场景就使用不同的实现类，并且可以自由切换策略。
策略模式需要一个策略接口，不同的策略实现不同的实现类，在具体业务环境中仅持有该策略接口，根据不同的场景使用不同的实现类即可。

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

import com.example.paydemo.vo.Order;

/**
 * 支付接口
 */
public interface IPayService {
    String pay(Order order);
}

订单vo类

import lombok.Data;
import java.math.BigDecimal;

@Data
public class Order {
    // 支付金额
    private BigDecimal amount;
    // 支付方式
    private String paymentType;
}

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

import com.example.paydemo.service.IPayService;
import com.example.paydemo.vo.Order;
import org.springframework.stereotype.Service;

/**
 * 微信支付实现类
 */
@Service("WechatPay")
public class WechatPayServiceImpl implements IPayService {
    @Override
    public String pay(Order order) {
        return "微信支付成功";
    }
}

• 支付宝实现：

import com.example.paydemo.service.IPayService;
import com.example.paydemo.vo.Order;
import org.springframework.stereotype.Service;

/**
 * 支付宝支付实现类
 */
@Service("AliPay")
public class AliPayServiceImpl implements IPayService {
    @Override
    public String pay(Order order) {
        return "支付宝支付成功";
    }
}

• 云闪付实现：

import com.example.paydemo.service.IPayService;
import com.example.paydemo.vo.Order;
import org.springframework.stereotype.Service;

/**
 * 云闪付支付实现类
 */
@Service("UnionPay")
public class UnionPayServiceImpl implements IPayService {
    @Override
    public String pay(Order order) {
        return "云闪付支付成功";
    }
}

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

import com.example.paydemo.service.IPayService;
import com.example.paydemo.vo.Order;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.math.BigDecimal;

@RestController
public class PayController {
    @Autowired
    private ApplicationContext applicationContext;

    /**
     * 支付
     *
     * @param amount      is 支付金额
     * @param paymentType is 支付方式
     * @return 支付结果
     */
    @GetMapping("/pay")
    public String pay(@RequestParam("amount") BigDecimal amount, @RequestParam("paymentType") String paymentType) {
        Order order = new Order();
        order.setAmount(amount);
        order.setPaymentType(paymentType);
        // 根据支付类型获取对应的策略bean
        IPayService payService = applicationContext.getBean(order.getPaymentType(), IPayService.class);
        // 调用支付接口
        String result = payService.pay(order);
        // 返回支付信息
        return result;
    }
}

测试一下：根据前端传入不同的支付方式参数，spring容器加载不同的实现类



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