Spring进阶：三步搞定MybatisPlus多数据源，详细配置及原理解析

前言

MybatisPlus(MP)作为mybatis的增强工具，提供了配置多数据源的扩展，通过简单的几步配置，即可使用注解轻松切换数据源。

以下是dynamic-datasource提供的功能列表：

使用方法

1、引入
dynamic-datasource-spring-boot-starter。

<dependency>
  <groupId>com.baomidou</groupId>
  <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
  <version>${version}</version>
</dependency>

2、配置多数据源。

spring:
  datasource:
    dynamic:
      primary: master #默认主库名为master
      strict: false #不使用严格模式
      datasource:
        master:
          url: jdbc:mysql://127.0.0.1:3306/master
          username: root
          password: 66668888
          driver-class-name: com.mysql.jdbc.Driver 
        slave_1:
          url: jdbc:mysql://127.0.0.1:3307/slave_1
          username: root
          password: 66668888
          driver-class-name: com.mysql.jdbc.Driver
        slave_2:
          url: ENC(xxxxx) # 内置加密
          username: ENC(xxxxx)
          password: ENC(xxxxx)
          driver-class-name: com.mysql.jdbc.Driver

主库默认为master，从库命名的格式默认以_分割。

3、在方法或者类上使用@DS切换数据源

​DS注解既可以写在类上，也可以写在方法上，方法上的优先级高于类。

如果没有指定DS的属性值或者没有使用DS注解，就自动走master主库。

通过以上3步，就可以轻松使用MP切换多数据源的能力了。当然，除了支持主从模式，还支持多主多从、多库，混合模式。

原理探究

打开dynamic-datasource-spring-boot-starter.jar包，在spring.factories文件中，可以看到配置了DynamicDataSourceAutoConfiguration启动类。

打开DynamicDataSourceAutoConfiguration类，可以看到通过@AutoConfigureBefore注解，提前初始化了MP自己的DataSource数据源。

而DataSourceAutoConfiguration类的核心作用就是初始化DataSrouce，MP通过抢先配置的方式，使DataSource变成了自己指定的数据源。

除了初始化MP自己的数据源，DynamicDataSourceAutoConfiguration类还做了一些其他的初始化工作，比如DynamicDataSourceProperties配置，初始化DS注解的切面Advisor等。

初始化工作做完以后，接下来就是根据DS注解判断走哪个数据源了。

DynamicDataSourceAnnotationInterceptor类是一个方法拦截器，它的invoke里面有判断DS key的操作。

在determineDatasourceKey方法中，会调用提前注入的DataSourceClassResolver类的findKey方法，判断类或者方法上是否指定了DS注解。

在这个方法中，MP会将方法的判断结果缓存下来，下次执行相同的类方法就可以直接缓存，从而大大提升查找效率。

获取到dsKey以后，会将其设置在DynamicDataSourceContextHolder中的LOOKUP_KEY_HOLDER中，它是一个泛型为Deque<String>的ThreadLocal，为什么设置成栈呢？官方给出的解释是：

接下来就是获取数据库链接的AbstractRoutingDataSource类的getConnection方法，

而determineDataSource方法就会去获取指定的数据源，

这里的逻辑是：如果没有指定DS，就走主库，如果指定了DS，就根据DS的属性值进行条件匹配，看走哪个判断分支。

获取到connection链接之后，接下来就是执行真正的数据库语句了。

以上就是一个常见的数据库操作的流程，大体思路就是先找MP的启动类，看看启动类都做了哪些工作，之后再按照找数据源，找链接的思路，看MP是如何具体实现多数据源切换的操作的。

当然，MP多数据源还有其他的功能点，但整体来说，代码逻辑并不是很复杂，大家可以通过debug对源码进行梳理。