Spring 多数据源管理源码分析Spring 多数据源管理源码分析

Spring多数据源管理实现原理

应用场景：

大部分单一架构项目连接一台数据库服务器，但随着业务的增加数据库数据量不断飙升，数据库达到性能瓶颈，大部分技术人员都会对数据库主从配置；既然读写分离那就需要连接两个不同的数据库，这时候Spring多数据源管理类AbstractRoutingDataSource就要派上用场了（排除使用数据库集群管理工具统一管理的应用场景）

源码分析：

Paste_Image.png

通过源码可以看出该类是一个抽象类，定义了6个属性。

targetDataSources：是一个map类型该属性正是用来维护项目中多个数据源

defaultTargetDataSource：通过属性名很直观的可以理解它的作用（默认数据源）

lenientFallback：默认为true，无需改动

dataSourceLookup：查找数据源接口的名称

Paste_Image.png

该类是一个interface并且只有一个方法getDataSource，通过方法的参数名称应该清楚传入一个字符类型的数据源名称获取DataSource

resolvedDataSources：如果该字段没有赋值，就是targetDataSources

resolvedDefaultDataSource：改变后的数据源

深入理解：

使用数据源的目的就是要获取Connection，接下来就从AbstractRoutingDataSource的getConnection方法一探究竟。

Paste_Image.png

直接进入determineTargetDataSource方法

Paste_Image.png

该方法的作用是检索目标数据源
第一句代码
Assert.notNull(this.resolvedDataSources, "DataSource router not initialized");

Paste_Image.png

判断resolvedDataSources是否为空，为空就抛出异常。

Paste_Image.png

通过afterPropertiesSet方法看出如果resolvedDataSources为空就遍历targetDataSources并把对应的key，value put到resolvedDataSources
第二句代码
Object lookupKey = determineCurrentLookupKey();
进入determineCurrentLookupKey方法

Paste_Image.png

该方法是一个抽象方法，返回要从resolvedDataSources查找key，该方法还会实现检查线程绑定事务上下文。
第三局代码
DataSource dataSource = this.resolvedDataSources.get(lookupKey);
从resolvedDataSources中取出数据源并返回