dynamic-datasource-spring-boot-starter是苞米豆提供的一个动态切换数据源的工具,可以帮助企业或者个人实现多数据源的切换,这里通过阅读源码的方式解析是如何动态的切换数据源的,采用的版本是3.5.1
源码解析
通过官方文档可以看到,如果想要动态的切换数据源,就需要在类或者方法上加上@DS注解
加上这个注解以后通过aop就可以拦截且识别到dsName究竟是什么,然后将其存储在ThreadLocal中
这个ThreadLocal中存储的是一个队列,我认为存储队列的目的是对应官方文档中的“就近原则”
再来看到DynamicRoutingDataSource这个实现类,找到determineDataSource这个方法,可以看到实际上他是实现方法,点击左边的绿色小图标
看完以后就知道了,实际上这个就是数据库连接,根据这个数据源获取相应的数据库连接然后进行相应的操作
再回到determineDataSource这个方法上,可以看到他是获取到队列中的第一个元素,然后再去map中获取相应的数据源,那么再去看看这个getDataSource方法
点开以后看到实际上这个是有一个dataSourceMap的map,里面存储着各种数据源,那么疑问来了,这些数据源是什么时候被put进这个map里面的
如果仔细观察可以发现实际上DynamicRoutingDataSource这个类实现了InitializingBean这个接口(具体什么作用可以自己搜索)然后调用了afterPropertiesSet这个方法,在bean被创建完了以后就会调用这个方法,于是数据源是什么时候被put进这个map了
我们知道实际上,动态切换数据源是通过yaml文件的相关配置进行加载数据源的,那么加载数据源的这一步实际上也在afterPropertiesSet这个方法里面
点进去以后发现这是一个接口有两个实现类,选择yml这个实现类来看下
进来以后发现loadDataSources这个方法就是用于加载数据源的,可以看到他在这个方法里面传入了一个Map<String, DataSourceProperty> dataSourcePropertiesMap,点进这个方法里面
这while循环里面的内容就是用于加载数据源了,根据传入进来的存储着数据源配置类的map,遍历map里面的数据源配置类然后加载数据源
这个map里面的数据源配置类又是根据依赖注入注入进来的
最后,在获取完了数据源以后不要忘了aop里面的方法是环绕方法,由于遵循就近原则,最后会将队列里面的内容弹出,获取下一个数据源
完整流程
最后贴上一张完整的流程图,如果有不正确的地方还请各位指正