浅析动态切换数据源的原理（接上篇）

​​上一篇​​我们实现了多数据源动态切换的功能，这次我们来看一下是如何实现的。

没看过上一篇的点击​​这里​​

我们在​​DynamicDataSourceConfig​​中配置了所有数据源信息，并存在一个map中

因为我们的​​DynamicDataSource​​继承了​​AbstractRoutingDataSource​​抽象类，​​AbstractRoutingDataSource​​中有几个重要的属性在这里我们用到了

@Nullable
private Map<Object, Object> targetDataSources;
@Nullable
private Object defaultTargetDataSource;

• ​​targetDataSources​​：持有我们多数据源的所有数据源，key为不重复的对象用于唯一标识一个数据源，value为数据源DataSource实例
• ​​defaultTargetDataSource​​：默认的数据源实例

可以看到我们在配置类中​​new DynamicDataSource​​在构造方法中传了两个参数，就对应了​​defaultTargetDataSource​​和​​targetDataSources​​，这样DynamicDataSource就持有了我们的默认数据源和所有数据源。

然后我们可以看到​​DynamicDataSource​​实现了​​AbstractRoutingDataSource​​抽象类的抽象方法​​determineCurrentLookupKey()​​：

这个方法的作用就是返回一个key，该key对应当前你希望使用的DataSource。在这个里，我们优化了一下，使用一个类​​DataSourceHolder​​持有我们的这个key，所以​​DynamicDataSource​​方法就变得很简洁了，就是在​​determineCurrentLookupKey​​方法中调用​​DataSourceHolder​​的静态方法​​getDataSourceName​​获取到当前用户想要使用的​​DataSource​​的​​key​​。

在​​AbstractRoutingDataSource​​的​​determineTargetDataSource​​方法中调用了​​determineCurrentLookupKey​​方法：

我们可以看到拿到key后，是从​​resolvedDataSources​​（一个map对象）中get对应的​​DataSource​​，get不到返回​​resolvedDefaultDataSource​​，看样子是默认数据源

@Nullable
private Map<Object, DataSource> resolvedDataSources;

但是​​resolvedDataSources​​和​​resolvedDefaultDataSource​​都不是刚刚我们所介绍的​​targetDataSources​​和​​defaultTargetDataSource​​。

那是因为做了赋值操作，

org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource#afterPropertiesSet

​​AbstractRoutingDataSource​​实现了​​InitializingBean​​接口的​​afterPropertiesSet​​，会在实例化bean后调用此方法（​​AbstractAutowireCapableBeanFactory#invokeInitMethods​​方法调用的），所以​​resolvedDataSources​​和​​resolvedDefaultDataSource​​就有值了。

继续回到​​determineTargetDataSource​​方法，该返回​​DataSource​​实例，既然是返回实例，肯定是有地方去获取​​DataSource​​，因为我们用的持久层框架是​​mybatis​​，所以获取​​DataSource​​的这个操作势必是​​mybatis​​发起的。

清楚了这一点就好办了，那就要看​​mybatis​​了，查阅资料我们可以发现，mybatis包中有这样一个类负责管理JDBC连接的生命周期，它就是​​org.mybatis.spring.transaction.SpringManagedTransaction​​（​​mybatis-spring.jar​​，​​spring​​集成​​mybatis​​的jar包）它实现了​​mybatis​​中的​​Transaction​​接口

我们可以看到接口中有一个​​getConnection()​​方法，即是要从数据库连接池中获取一个连接：

@Override
public Connection getConnection() throws{
  if (this.connection == null) {
    openConnection();
  }
  return this.connection;
}

在连接为空时，调用​​openConnection()​​方法获取连接，我们看看​​openConnection​​方法做了什么操作：

该方法第一个调用​​DataSourceUtils.getConnection(this.dataSource)​​获取连接，其调用​​DataSourceUtils​​中的​​doGetConnection(dataSource)​​获取连接

然后在这个方法内部又调用​​fetchConnection(dataSource)​​方法获取连接

​​fetchConnection(datasource)​​内直接就调用了​​DataSource​​对象的​​getConnection​​方法

因为我们的​​DynamicDataSource​​没有重写​​getConnection​​方法，那么就会调父类的

@Override
public Connection getConnection() throws{
   return determineTargetDataSource().getConnection();
}

可以看到先调了​​determineTargetDataSource​​方法获取​​DataSource​​再​​getConnection()​​。​​determineTargetDataSource​​方法刚刚我们已经看过了。

总结

大致思想就是，mybatis执行SQL时会从DataSource拿一个JDBC连接，所以spring巧妙的利用了这个特点，它提前拿到多个数据源的实例信息，在mybatis获取连接时根据用户的指令，动态的选择返回哪个连接，这些操作对mybatis是透明的。

spring把一些通用方法高度抽象到​​AbstractRoutingDataSource​​抽象类中，使用两个变量管理数据源，​​targetDataSources​​和​​defaultTargetDataSource​​，然后再预留出​​determineCurrentLookupKey​​方法供我们实现，即只需要返回一个key就可以，返回的key用于从​​targetDataSources​​中选取出我们指定的数据源。

所以，除了​​determineCurrentLookupKey​​方法外，其他的操作对于使用者来说也是透明的，使用者只需要关心如何在​​determineCurrentLookupKey​​方法中实现自己的选择数据源的规则即可。

只不过，我们是通过AOP切面拦截的方式增强方法，在持久层方法执行前即在mybatis执行SQL前我们事先把​​DataSource​​换成我们指定的即可。

在这里我们换数据源的方式是通过一个​​DataSourceHolder​​类中的​​ThreadLocal​​实现的，原因是为了保证多线程并发环境下不同线程切换数据源时不会乱，​​Threadlocal​​线程独有的一个对象，在其内部保存我们的key，在​​determineCurrentLookupKey​​中获取并返回即可。

代码已上传到码云，springboot版本和SSM都有

感兴趣的可以去下载

​​​​https://gitee.com/itwalking/springboot-dynamic-datasource​​​

​​​​https://gitee.com/itwalking/ssm-dynamic-datasource​​

本文转载自公众号BiggerBoy