策略模式

问题说明

当前项目中，订单的状态比较多，不同状态下取消订单的逻辑是不同的；而且用户不同，可操作的内容也不同

普通用户：可取消待支付、派单中、待服务的订单
运营人员：可取消派单中、待服务、服务中、完成的订单

因此最终实现取消订单的代码逻辑如下所示：

public void cancel(OrderCancelDTO orderCancelDTO) {
    
    Orders orders = getById(orderCancelDTO.getId());//查询订单信息
    Integer ordersStatus = orders.getOrdersStatus();//订单状态
    CurrentUserInfo currentUserInfo = UserContext.currentUser();//获取当前用户
    Integer userType = currentUserInfo.getUserType();//用户类型
    
    if(UserType.C_USER==userType){//普通用户取消订单
        if(OrderStatusEnum.NO_PAY.getStatus()==ordersStatus){ //订单状态为待支付
            //...
        }else if(OrderStatusEnum.DISPATCHING.getStatus()==ordersStatus){ //订单状态为派单中
            //...
        }else if(OrderStatusEnum.NO_SERVE.getStatus()==ordersStatus){ //订单状态为待服务
            //...
        }else{
            throw new CommonException("当前订单状态不支持取消");
        }
    }else if(UserType.OPERATION==userType){//运营人员取消订单
        if(OrderStatusEnum.DISPATCHING.getStatus()==ordersStatus) { //订单状态为派单中
            //...
        }else if(OrderStatusEnum.NO_SERVE.getStatus()==ordersStatus){//订单状态为待服务
            //...
        }else if(OrderStatusEnum.SERVING.getStatus()==ordersStatus){//订单状态为服务中
            //...
        }else if(OrderStatusEnum.FINISHED.getStatus()==ordersStatus){//订单状态为已完成
            //...
        }else{
            throw new CommonException("当前订单状态不支持取消");
        }
    }
}

这种代码虽然可以实现功能，但是可读性和扩展性都很差，我们可以使用策略模式进行改造

策略模式

使用场景：针对一个业务，有多个不同的执行分支，需要根据不同的条件，切换到不同的分支上去

好处：解决使用条件语句（如if…else）导致的复杂性和扩展性问题

策略模式需要开发下面的接口和类:

策略接口：定义策略方法
策略实现类：需要实现策略接口，并重写接口中声明的策略方法
上下文环境类：需要维护所有策略对象，然后根据不同的条件，调用不同的策略对象

下面以一个**支付方式**选择的场景为例子来讲解一下，需求如下：

当前项目支持微信和阿里两种支付方式
用户可以自行选择一种支付方式
根据用户选择，执行对应的支付代码

1）定义策略接口

创建PayStrategy接口

package com.jzo2o.orders.manager.service.strategy;

/**
 * 支付策略接口
 */
public interface PayStrategy {

    //策略方法
    void pay();
}

2）定义微信支付策略类

package com.jzo2o.orders.manager.service.strategy.impl;

import com.jzo2o.orders.manager.service.strategy.PayStrategy;
import org.springframework.stereotype.Component;

/**
 * 微信支付策略实现类
 */
@Component("wxPay")
public class WxPayStrategy implements PayStrategy {
    @Override
    public void pay() {
        System.out.println("使用微信进行支付");
    }
}

3）定义阿里支付策略类

package com.jzo2o.orders.manager.service.strategy.impl;

import com.jzo2o.orders.manager.service.strategy.PayStrategy;
import org.springframework.stereotype.Component;

/**
 * 阿里支付策略实现类
 */
@Component("aliPay")
public class AliPayStrategy implements PayStrategy {
    @Override
    public void pay() {
        System.out.println("使用阿里进行支付");
    }
}

4）定义上下文环境类

package com.jzo2o.orders.manager.service.strategy;

import cn.hutool.extra.spring.SpringUtil;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.util.Map;

/**
 * 上下文环境类(策略管理类)
 */
@Component
public class PayStrategyManager {
    Map<String, PayStrategy> payStrategyMap;

    //维护所有策略对象
    @PostConstruct //此注解标注的方法会在当前对象PayStrategyManager创建之后自动执行
    public void init() {
        //key: 当前对象在容器中的id
        //value: 当前对象
        payStrategyMap = SpringUtil.getBeansOfType(PayStrategy.class);
    }

    //根据用户需求, 执行指定的策略对象
    public void pay(String key) {
        PayStrategy payStrategy = payStrategyMap.get(key);
        if (payStrategy == null) {
            throw new RuntimeException("暂不支持当前支付方式");
        }
        payStrategy.pay();
    }
}

5）测试类

package com.jzo2o.orders.manager.service.strategy;

import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;

/**
 * 测试支付方式的选择
 */
@SpringBootTest
public class PayTest {

    @Autowired
    private PayStrategyManager payStrategyManager;

    @Test
    public void test(){
        //测试微信支付
        payStrategyManager.pay("wxPay");
        //测试阿里支付
        payStrategyManager.pay("aliPay");
    }
}