首页 > 其他分享> 文章详细

原理解密→SpringAOP实现动态数据源（读写分离），实质上原理是什么？

2020-04-14 19:56:33 阅读：260 来源： 互联网

标签：Spring 数据源 ThreadLocal Connection DynamicDataSource AOP SpringAOP 原理

原理解密

我们逐个讲解其中涉及的点，然后串起来理解读写分离的底层原理

Spring AOP

AOP：Aspect Oriented Program

关于 Spring AOP，相信大家耳熟能详，它是对 OOP 的一种补充，OOP 是纵向的，AOP 则是横向的

bc6c40bde4d344338d7d186216f11968

如上图所示，OOP 属于一种纵向拓展，AOP 则是一种横向拓展。AOP 依托于 OOP，将公共功能代码抽象出来作为一个切面，减少重复代码量，降低耦合

AOP 的底层实现是动态代理，具体的表现形式粗略如下

39101fd77bf84e84b46a401c2899c9c1

对 Spring AOP 有个大致了解了，我们就可以接着往下看了

Spring 数据源

无论是 Spring JDBC，还是 Hibernate，亦或是 MyBatis，其实都是对 JDBC 的封装；对于JDBC，我们不要太熟，大体流程如下

70225eaa87724a18b2ac6ee115cac309

然而，在实际应用中，我们往往不会直接使用 JDBC，而是使用 ORM，ORM 会封装上述的流程，也就说我们不再需要关注了；MyBatis 使用步骤大致如下

abe4bd1c1eb34a3fb72be8d64976fef9

我们以 SpringBoot + pagehelper + Druid（ssm）为例，来看看具体是怎么获取 Connection 对象的

bbd73c70d407476296d8db3a06a06ff9

可以看到，如果事务管理器中存在 Connection 对象，则直接返回，否则从数据源中获取返回（同时也赋值给了事务管理器）；当取到 Connection 对象后，后续的流程大家就非常清楚了

然而我们不需要关注 Connection 对象，只需要关注数据源，为什么呢？因为我们的配置文件中配置的是数据源而不是 Connection，是不是很有道理？

ThreadLocal

如果我们需要在各层之间进行参数的传递，实现方式有哪些？

最常见的方式可能就是方法参数，但还有一种容易忽略的方式：ThreadLocal，可以在当前线程内传递参数，减少方法的参数个数

关于 ThreadLocal，有兴趣的可以查看：结合ThreadLocal来看spring事务源码，感受下清泉般的洗涤！

当我们熟悉上面的三点后，后面的就好理解了，接着往下看

动态数据源

一个数据源只能对应一个数据库，如果我们有多个数据库（一主多从），那么就需要配置多个数据源，类似如下

 <!-- master数据源 -->
    <bean id="masterDataSource" class="com.alibaba.druid.pool.DruidDataSource">
        <!-- 基本属性 url、user、password -->  
        <property name="driverClassName" value="${jdbc.driverClassName}" />  
        <property name="url" value="${jdbc.url}" />  
        <property name="username" value="${jdbc.username}" />  
        <property name="password" value="${jdbc.password}" />  
        <property name="initialSize" value="${jdbc.initialSize}" />  
        <property name="minIdle" value="${jdbc.minIdle}" />   
        <property name="maxActive" value="${jdbc.maxActive}" />  
        <property name="maxWait" value="${jdbc.maxWait}" />        <!-- 超过时间限制是否回收 -->
        <property name="removeAbandoned" value="${jdbc.removeAbandoned}" />        <!-- 超过时间限制多长； -->
        <property name="removeAbandonedTimeout" value="${jdbc.removeAbandonedTimeout}" />        <!-- 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒 -->
        <property name="timeBetweenEvictionRunsMillis" value="${jdbc.timeBetweenEvictionRunsMillis}" />        <!-- 配置一个连接在池中最小生存的时间，单位是毫秒 -->
        <property name="minEvictableIdleTimeMillis" value="${jdbc.minEvictableIdleTimeMillis}" />        <!-- 用来检测连接是否有效的sql，要求是一个查询语句-->
        <property name="validationQuery" value="${jdbc.validationQuery}" />        <!-- 申请连接的时候检测 -->
        <property name="testWhileIdle" value="${jdbc.testWhileIdle}" />        <!-- 申请连接时执行validationQuery检测连接是否有效，配置为true会降低性能 -->
        <property name="testOnBorrow" value="${jdbc.testOnBorrow}" />        <!-- 归还连接时执行validationQuery检测连接是否有效，配置为true会降低性能  -->
        <property name="testOnReturn" value="${jdbc.testOnReturn}" />    </bean>

    <!-- slave数据源 -->
    <bean id="slaveDataSource" class="com.alibaba.druid.pool.DruidDataSource">
        <property name="driverClassName" value="${slave.jdbc.driverClassName}" />  
        <property name="url" value="${slave.jdbc.url}" />  
        <property name="username" value="${slave.jdbc.username}" />  
        <property name="password" value="${slave.jdbc.password}" />  
        <property name="initialSize" value="${slave.jdbc.initialSize}" />  
        <property name="minIdle" value="${slave.jdbc.minIdle}" />   
        <property name="maxActive" value="${slave.jdbc.maxActive}" />  
        <property name="maxWait" value="${slave.jdbc.maxWait}" />        <property name="removeAbandoned" value="${slave.jdbc.removeAbandoned}" />        <property name="removeAbandonedTimeout" value="${slave.jdbc.removeAbandonedTimeout}" />        <property name="timeBetweenEvictionRunsMillis" value="${slave.jdbc.timeBetweenEvictionRunsMillis}" />        <property name="minEvictableIdleTimeMillis" value="${slave.jdbc.minEvictableIdleTimeMillis}" />        <property name="validationQuery" value="${slave.jdbc.validationQuery}" />        <property name="testWhileIdle" value="${slave.jdbc.testWhileIdle}" />        <property name="testOnBorrow" value="${slave.jdbc.testOnBorrow}" />        <property name="testOnReturn" value="${slave.jdbc.testOnReturn}" />    </bean>

可是事务管理器中只有一个数据源的引用

aa081d5256634512bbb4437e07f4a235

那怎么对应我们配置文件中的多个数据源呢？其实，我们可以自定义一个类 DynamicDataSource 来实现 DataSource，DynamicDataSource 中存储我们配置的多数据源，然后将 DynamicDataSource 的实例配置给事务管理器；当从事务管理器获取 Connection 对象的时候，会从 DynamicDataSource 实例获取，然后再由 DynamicDataSource 根据 routeKey 路由到某个具体的数据源，从中获取 Connection；大体流程如下

81e823c4de024c3bbe84b2b81a6b7cd3

Spring 也考虑到了这一点，提供了一个抽象类：AbstractRoutingDataSource，DynamicDataSource 继承它可以为我们省非常多的代码

56b07256a3114715a7dbcce24ec63c28

配置文件中增加如下配置

1a672ae808f34bca9c12cd632a8674ed

但是问题又来了，这个 routeKey 怎么处理，也就说 DynamicDataSource 怎么知道用哪个数据源？ AbstractRoutingDataSource 提供了一个方法： determineCurrentLookupKey 我们只需要实现它，DynamicDataSource 就知道是使用哪个 lookupKey （routeKey 在 Spring 中的命名）了；determineCurrentLookupKey 具体该如何实现了，我们可以结合 ThreadLocal 来实现；整个流程大致如下

fe8b5fa84e6345d8b1c82efd8f5421b6

一旦我们在切面中指定了 lookupKey，那么后续就会使用 lookupKey 对应的数据源来操作数据库了

自此，相信大家已经明白了动态数据源的底层原理

总结

Spring AOP → 将我们指定的 lookupKey 放入 ThreadLocal

ThreadLocal → 线程内共享 lookupKey

DynamicDataSource → 对多数据源进行封装，根据 ThreadLocal 中的 lookupKey 动态选择具体的数据源

如果我们对其中的某个环节不懂，可以试着删掉它，然后看这个流程能否正常串起来，这样就能明白各个环节的作用了

悬念

Spring AOP 实现多数据源，是否与 Spring 事务冲突，若冲突了该如何解决？

ICode9