当前位置：首页 > news >正文

Springboot 使用设计模式- 策略模式

news 来源：原创 2024/5/3 17:54:32

前言

直白点，什么场景我们需要使用到设计模式- 策略模式。

在平常的springboot项目里面做CRUD，我们的习惯性基本是一个mapper，一个service，一个serviceImpl。

但是有时候会出现一种情况，就是实现类impl里面虽然业务主线是归为一致的，但是需要进行类别判断去执行不同的业务逻辑，

导致impl里面出现一个类别的if 后面跟上一大段业务代码，然后同时存在这一几个 if 判断，这种情况恨不得单独把代码抽出去写成方法，看起来稍微舒服一些（其实这种情况使用策略模式是非常适合的）。

举例场景：

我现在需要实现一个关于支付预下单的功能，然后这个预下单功能大致用到的参数都一样，
但是在下单类型上分为国内单，海外单，折扣单。

三种不同的下单流程都属于预下单功能的一部分，但是结合业务后，逻辑算法都是基本独立的。

按照往常的写法，可能会出现如下的代码写法：

或者有的为了不想麻烦，直接就建了三个service，然后三个serviceImpl。

那么就针对上面这个场景，我们使用设计模式-策略模式，就能非常润滑地去实现。

正文

订单信息实体类，OrderInfo.java：

import lombok.Data;

/**
 * @Author : JCccc
 * @CreateTime : 2020/5/11
 * @Description :
 **/
@Data
public class OrderInfo {

    private String orderId; //订单id
    private String platFormType; //（平台）订单类型
    private Double amount;  //金额
    private String createTime; //创建时间
    
    //....省略若干业务字段
}

预下单的策略接口，OrderStrategyService.java ：

/**
 * @Author : JCccc
 * @CreateTime : 2020/5/11
 * @Description :
 **/
public interface OrderStrategyService  {

   //预下单
   String  preCreateOrder(OrderInfo orderInfo);

}

接下来是三个类型订单的实现类，都实现策略接口：

国内，OrderDomestic.java：

/**
 * @Author : JCccc
 * @CreateTime : 2020/5/11
 * @Description :国内
 **/
@Component("Domestic")
public class OrderDomestic implements OrderStrategyService {
    @Override
    public String preCreateOrder(OrderInfo orderInfo) {

        System.out.println("*处理国内预下单的相关业务*");
        return orderInfo.getPlatFormType()+"-国内预下单";
    }
}

海外，OrderOverseas.java ：

/**
 * @Author : JCccc
 * @CreateTime : 2020/5/11
 * @Description :海外
 **/
@Component("Overseas")
public class OrderOverseas implements OrderStrategyService {
    @Override
    public String preCreateOrder(OrderInfo orderInfo) {
        System.out.println("**处理海外预下单的相关业务**");
        return orderInfo.getPlatFormType()+"-海外预下单";
    }
}

特殊回扣，

/**
 * @Author : JCccc
 * @CreateTime : 2020/5/11
 * @Description :特殊订单（回扣）
 **/
@Component("Rebate")
public class OrderRebate implements OrderStrategyService {
    @Override
    public String preCreateOrder(OrderInfo orderInfo) {
        
        System.out.println("***处理国内特殊回扣预下单的相关业务***");
        return orderInfo.getPlatFormType()+"-特殊回扣预下单";


    }
}

可以看到上面三个实现类都实现了策略接口OrderStrategyService，而且使用了注解@Component +组件名。

那么接下来就是策略设计模式的核心，策略接口实现类的调度使用类，OrderStrategyContext.java（留意看里面的注释）：

/**
 * @Author : JCccc
 * @CreateTime : 2020/5/11
 * @Description : 利用Spring的发现机制,将实现了OrderStrategyService的类都put到orderStrategyMap里面。
 *                 后面只需要根据platformId对应好 各个实现类的注解 如： @Component("Domestic") 就可以取出不同的业务实现类
 **/

@Service
public class OrderStrategyContext {
    
    private final Map<String, OrderStrategyService> orderStrategyMap = new ConcurrentHashMap<>();

    public OrderStrategyContext(Map<String, OrderStrategyService> strategyMap) {
        this.orderStrategyMap.clear();
        strategyMap.forEach((k, v)-> this.orderStrategyMap.put(k, v));
    }

    public OrderStrategyService getResource(OrderInfo orderInfo){
        return orderStrategyMap.get(orderInfo.getPlatFormType());
    }
}

ps：可以看到getResource这个方法，获取资源，里面其实就是根据订单类型去取出对应的业务实现类。

ok，到这策略模式的结合使用已经大致完毕，最后是通过策略调度类去使用我们不同类型的实现类。

写一个简单的接口，看看效果：

OrderTestController.java ：

/**
 * @Author : JCccc
 * @CreateTime : 2020/5/11
 * @Description :
 **/
@RestController
public class OrderTestController {

    @Autowired
    private OrderStrategyContext orderStrategyContext;

    @PostMapping("/testStrategy")
    public String testStrategy(@RequestBody OrderInfo orderInfo){
        OrderStrategyService orderServiceImpl = orderStrategyContext.getResource(orderInfo);
        String resultTest = orderServiceImpl.preCreateOrder(orderInfo);
        return resultTest;

    }



}

特殊回扣下单：