阅读源码解析dynamic-datasource-spring-boot-starter中是如何动态切换数据源的

作者 : admin 本文共1087个字,预计阅读时间需要3分钟 发布时间: 2024-06-11 共1人阅读

dynamic-datasource-spring-boot-starter是苞米豆提供的一个动态切换数据源的工具,可以帮助企业或者个人实现多数据源的切换,这里通过阅读源码的方式解析是如何动态的切换数据源的,采用的版本是3.5.1

源码解析

通过官方文档可以看到,如果想要动态的切换数据源,就需要在类或者方法上加上@DS注解

阅读源码解析dynamic-datasource-spring-boot-starter中是如何动态切换数据源的插图

加上这个注解以后通过aop就可以拦截且识别到dsName究竟是什么,然后将其存储在ThreadLocal中
阅读源码解析dynamic-datasource-spring-boot-starter中是如何动态切换数据源的插图(1)
这个ThreadLocal中存储的是一个队列,我认为存储队列的目的是对应官方文档中的“就近原则”
阅读源码解析dynamic-datasource-spring-boot-starter中是如何动态切换数据源的插图(2)

再来看到DynamicRoutingDataSource这个实现类,找到determineDataSource这个方法,可以看到实际上他是实现方法,点击左边的绿色小图标

阅读源码解析dynamic-datasource-spring-boot-starter中是如何动态切换数据源的插图(3)

看完以后就知道了,实际上这个就是数据库连接,根据这个数据源获取相应的数据库连接然后进行相应的操作

阅读源码解析dynamic-datasource-spring-boot-starter中是如何动态切换数据源的插图(4)

再回到determineDataSource这个方法上,可以看到他是获取到队列中的第一个元素,然后再去map中获取相应的数据源,那么再去看看这个getDataSource方法

阅读源码解析dynamic-datasource-spring-boot-starter中是如何动态切换数据源的插图(5)

点开以后看到实际上这个是有一个dataSourceMap的map,里面存储着各种数据源,那么疑问来了,这些数据源是什么时候被put进这个map里面的

阅读源码解析dynamic-datasource-spring-boot-starter中是如何动态切换数据源的插图(6)

如果仔细观察可以发现实际上DynamicRoutingDataSource这个类实现了InitializingBean这个接口(具体什么作用可以自己搜索)然后调用了afterPropertiesSet这个方法,在bean被创建完了以后就会调用这个方法,于是数据源是什么时候被put进这个map了

阅读源码解析dynamic-datasource-spring-boot-starter中是如何动态切换数据源的插图(7)

我们知道实际上,动态切换数据源是通过yaml文件的相关配置进行加载数据源的,那么加载数据源的这一步实际上也在afterPropertiesSet这个方法里面

阅读源码解析dynamic-datasource-spring-boot-starter中是如何动态切换数据源的插图(8)

点进去以后发现这是一个接口有两个实现类,选择yml这个实现类来看下

阅读源码解析dynamic-datasource-spring-boot-starter中是如何动态切换数据源的插图(9)

进来以后发现loadDataSources这个方法就是用于加载数据源的,可以看到他在这个方法里面传入了一个Map dataSourcePropertiesMap,点进这个方法里面

阅读源码解析dynamic-datasource-spring-boot-starter中是如何动态切换数据源的插图(10)

这while循环里面的内容就是用于加载数据源了,根据传入进来的存储着数据源配置类的map,遍历map里面的数据源配置类然后加载数据源

阅读源码解析dynamic-datasource-spring-boot-starter中是如何动态切换数据源的插图(11)

这个map里面的数据源配置类又是根据依赖注入注入进来的

阅读源码解析dynamic-datasource-spring-boot-starter中是如何动态切换数据源的插图(12)
最后,在获取完了数据源以后不要忘了aop里面的方法是环绕方法,由于遵循就近原则,最后会将队列里面的内容弹出,获取下一个数据源

阅读源码解析dynamic-datasource-spring-boot-starter中是如何动态切换数据源的插图(13)

完整流程

最后贴上一张完整的流程图,如果有不正确的地方还请各位指正
阅读源码解析dynamic-datasource-spring-boot-starter中是如何动态切换数据源的插图(14)

本站无任何商业行为
个人在线分享 » 阅读源码解析dynamic-datasource-spring-boot-starter中是如何动态切换数据源的
E-->