spring boot 中如何优雅的使用自动注入实现策略模式

最新推荐文章于 2023-12-21 16:49:32 发布

"贰拾叁"

最新推荐文章于 2023-12-21 16:49:32 发布

阅读量456

点赞数 1

分类专栏：后端杂谈文章标签： spring boot 策略模式后端

本文链接：https://blog.csdn.net/Ka__ze/article/details/132533232

版权

后端杂谈专栏收录该内容

9 篇文章 0 订阅

订阅专栏

文章讲述了如何利用Spring框架的自动注入功能和策略模式，实现在运行时动态决定调用哪个PayService接口的实现类。通过Set存储服务实例，Map映射渠道号到服务，`@Autowired`和`@PostConstruct`确保了初始化过程的正确执行。

摘要由CSDN通过智能技术生成

前言

关系说明：PayService为支付接口，核心业务方法为epay()，该接口有好几个不同的实现类。
需求：在运行时需要动态决定走哪个实现类的逻辑。

解决方案

先上成品代码：

@Service
public class RouteService {
 
    @Autowired
    Set<PayService> payServiceSet;
    
    Map<String, PayService> payServiceMap;
 
    public PayResult epay(PayRequest payRequest) {
        PayService payService = payServiceMap.get(payRequest.getChannelNo());
        return  payService.epay(payRequest);
    }
 
    @PostConstruct
    public void init() {
        for (PayService payService : payServiceSet) {
            payServiceMap = new HashMap<>();
            payServiceMap.put(payService.channel(), payService);
        }
    }
}

代码分析

使用Set<PayService>类型会使Spring将所有实现了PayService接口的类都注入进Set中。

@Autowired
Set<PayService> payServiceSet;

Map<String, PayService>用于保存实现类与调用判断逻辑之间的映射关系。

Map<String, PayService> payServiceMap;

使用@PostConstruct注解修饰的init方法会在Spring容器的启动时自动的执行，在其中完成映射Map的构建。
其中的payService的channel()方法来自于接口定义，PayService接口要求每个实现类都实现该方法，该方法会返回不同实现类的唯一标识符。

@PostConstruct
public void init() {
    for (PayService payService : payServiceSet) {
        payServiceMap = new HashMap<>();
        payServiceMap.put(payService.channel(), payService);
    }
}

在调用时，请求参数中必须携带一个唯一标识符，用以表明本次调用需要走哪个实现类，也就是上述映射Map中的Key的值。

public PayResult epay(PayRequest payRequest) {
   PayService payService = payServiceMap.get(payRequest.getChannelNo());
   return  payService.epay(payRequest);
}