优化if-else的几种方式

策略模式

策略模式允许在运行时选择算法。策略模式是将算法定义成独立的类，并在运行时动态选择要使用的具体的算法，以此来避免多个if-else或switch语句的使用。
下面以支付功能为例子进行说明。
假设我们有一个支付系统，支持微信、QQ等多种支付方式。用户在支付时会选择自己需要的支付方式，后台功能接口中接收到用户的支付方式选择时，会进行不同的处理。在这里会产生if-else或者switch。如何使用策略模式来消除这些if-else的使用，下面示例说明。

1、创建支付策略接口

/**
 * 支付策略
 */
public interface PaymentStrategy {

    void pay(double amount);

}

2、书写不同的支付方式逻辑代码

微信支付

import org.springframework.stereotype.Component;

@Component
public class WeiXinPayment implements PaymentStrategy {

    @Override
    public void pay(double amount) {
        System.out.println("微信支付" + amount);
    }
}

QQ支付

import org.springframework.stereotype.Component;

@Component
public class QQPayment implements PaymentStrategy {

    @Override
    public void pay(double amount) {
        System.out.println("QQ支付" + amount);
    }
}

3、service层的实现类使用

import com.hysoft.study.service.PaymentStrategy;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * 支付实现类
 */
@Service
public class PaymentServiceImpl {

    private final Map<String, PaymentStrategy> strategies;

    @Autowired
    public PaymentServiceImpl(List<PaymentStrategy> paymentStrategies){
        this.strategies = paymentStrategies.stream()
                .collect(Collectors.toMap(s -> s.getClass().getSimpleName().toLowerCase(), Function.identity()));
    }

    public void processPayment(String strategyName,double amount){
        PaymentStrategy strategy = strategies.getOrDefault(strategyName,null);
        if (strategy != null){
            strategy.pay(amount);
        }else {
            throw new IllegalArgumentException("Strategy not found" + strategyName);
        }
    }

}

4、controller层的调用

import com.hysoft.study.service.impl.PaymentServiceImpl;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequestMapping("payment")
public class PaymentController {

    @Autowired
    private PaymentServiceImpl paymentService;

    @PostMapping("test")
    public void test(String paymentname,double amount){
        this.paymentService.processPayment(paymentname,amount);
    }

}

说明

需要注意的是，controller层调用service层实现方法的适合，传递了参数paymentname(支付方式)，这个支付方式需要提前和前端调用人员协商好，这里的名称是各种支付方式的bean名称，在service处理中已经有所体现

因此在这里名称传值可以是qqpayment或weixinpayment。因此传值需要提前和前端进行协商。

枚举与策略模式结合

枚举类型不仅可以用来表示一组常量，还可以定义与这些常量相关联的行为。结合策略模式，可以进一步简化代码。

1、创建枚举

public enum OrderStatus {

    NEW {
        @Override
        public void process () {
            System.out.println("处理新建订单");
        }
    },

    PAID {
        @Override
        public void process () {
            System.out.println("订单已支付");
        }
    },

    UNPAD {
        @Override
        public void process() {
            System.out.println("订单未支付");
        }
    };

    public abstract void process();
}

2、service层书写处理方法

import com.hysoft.study.model.OrderStatus;
import org.springframework.stereotype.Service;

@Service
public class OrderServiceImpl {

    public void handleOrder(OrderStatus status) {
        status.process();
    }

}

3、controller层调用

import com.hysoft.study.model.OrderStatus;
import com.hysoft.study.service.impl.OrderServiceImpl;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequestMapping("order")
public class OrderController {

    @Autowired
    private OrderServiceImpl orderService;

    @PostMapping("test")
    public void test(String status){
        OrderStatus aNew = OrderStatus.valueOf(status);
        this.orderService.handleOrder(aNew);
    }

}

4、说明

controller层调用时传输的参数status即使在枚举中的各常量，例如NEW或者PAID、UNPAD等

Lambda表达式与函数接口

以下示例在service层实现类中直接书写了不同vip等级的结算金额的逻辑

import org.springframework.stereotype.Service;
import java.util.HashMap;
import java.util.Map;
import java.util.function.Function;

@Service
public class StreamServiceImpl {

    private final Map<String, Function<Double, Double>> discountFunctions = new HashMap<>();

    public StreamServiceImpl() {
        discountFunctions.put("VIP1", e -> e * 0.95);
        discountFunctions.put("VIP2", e -> e * 0.95 - 20);
    }

    public double applyDiscount(String vipname, double price) {
        Double apply = discountFunctions.getOrDefault(vipname, Function.identity()).apply(price);
        return apply;
    }
}

在controller层调用时，需要传入vip等级和总计算金额，计算结果时打折后金额

import com.hysoft.study.service.impl.StreamServiceImpl;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequestMapping("stream")
public class StreamController {

    @Autowired
    private StreamServiceImpl streamService;

    @PostMapping("test")
    public double test(String vipname,Double price){
        return this.streamService.applyDiscount(vipname,price);
    }

}

说明

传参vipname的值应参考service层类中的vip参数