前言
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的ThreadLocal,为什么设置成栈呢?官方给出的解释是:
接下来就是获取数据库链接的AbstractRoutingDataSource类的getConnection方法,
而determineDataSource方法就会去获取指定的数据源,
这里的逻辑是:如果没有指定DS,就走主库,如果指定了DS,就根据DS的属性值进行条件匹配,看走哪个判断分支。
获取到connection链接之后,接下来就是执行真正的数据库语句了。
以上就是一个常见的数据库操作的流程,大体思路就是先找MP的启动类,看看启动类都做了哪些工作,之后再按照找数据源,找链接的思路,看MP是如何具体实现多数据源切换的操作的。
当然,MP多数据源还有其他的功能点,但整体来说,代码逻辑并不是很复杂,大家可以通过debug对源码进行梳理。
写文不易,感谢您的点赞和关注。