Spring多数据源管理源码分析

Spring多数据源管理实现原理

应用场景：

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

源码分析：

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

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

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

lenientFallback：默认为true，无需改动

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



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

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

resolvedDefaultDataSource：改变后的数据源

深入理解：

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



直接进入determineTargetDataSource方法



该方法的作用是检索目标数据源

第一句代码

Assert.notNull(this.resolvedDataSources, “DataSource router not initialized”);



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



通过afterPropertiesSet方法看出如果resolvedDataSources为空就遍历targetDataSources并把对应的key，value put到resolvedDataSources

第二句代码

Object lookupKey = determineCurrentLookupKey();

进入determineCurrentLookupKey方法



该方法是一个抽象方法，返回要从resolvedDataSources查找key，该方法还会实现检查线程绑定事务上下文。

第三局代码

DataSource dataSource = this.resolvedDataSources.get(lookupKey);

从resolvedDataSources中取出数据源并返回

代码实现

实现AbstractRoutingDataSource重写determineCurrentLookupKey



定义DataSourceContextHolder



通过ThreadLocal类使每个线程获取独立的数据源，防止并发访问时获取错误的数据源

基于自定义注解+aop实现数据源动态切换

自定义注解



定义aop实现类



Spring配置文件中配置自定义DataSource类DynamicDataSource



所有工作都已完成，只需要在Dao层方法上添加DataSource注解即可



没有加DataSource注解的方法都会使用配置文件中配置的默认数据源

启示

在开发中如果数据库事物是在Service层，这时在Dao层中即使添加DataSoure注解也会使用默认数据源，请各位开发者注意！！！原因：请认真查看源码分析中第二句代码部分的内容