前两天有小伙伴谈起Boot实现多数据源的解决方案,我之前有见过一个方案是实例化多个数据源bean,每个数据源对应一套独立的mybatis映射(dao接口和xml),首先这个方案是可以解决问题的,但是随之带来的问题是你得维护多个mybatis映射空间,不优雅,且一旦成长下去…… 画面太美,不敢想象。这边介绍另一个实现方案,通过注解动态切换数据源。


核心依赖

org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource是Spring-jdbc包对DataSource的抽象封装,该抽象类维护了一个默认的主数据源defaultTargetDataSource和一个可选的数据源集合Map<Object, Object> targetDataSources,并提供了一个决策方法通过lockupkey来获取合适的数据源来处理业务,使得我们只需要注入主数据源和备选数据源列表,在合适的时间和地点动态设置适当的lockupkey,就可以方便的get到你希望的姿势


spring boot clickhouse 多数据源 spring boot多个数据源_抽象类

我们继承上面抽象类并重写其determineCurrentLookupKey()方法,这边通过数据源持有器静态方法返回当前的数据源名称,于是数据源的动态切换问题就变成该数据源持有器的当前数据源名称的切换了


spring boot clickhouse 多数据源 spring boot多个数据源_自定义_02


自定义数据源持有器

接上文,给出数据源持有器,实际上是维护一个ThreadLocal实例,通过该实例持有当前线程的已选数据源的DataSourceName(关于ThreadLocal原理不做赘述,有兴趣的可以@我)

spring boot clickhouse 多数据源 spring boot多个数据源_数据源_03


配置属性实体

数据源持有器准备妥当后,不要着急,我们再来处理数据源的问题。自定义数据源封装类AicloudDataSourceProperties,并在其内部定义配置文件的映射实体类DetailProperties,这里不赘述原理。

spring boot clickhouse 多数据源 spring boot多个数据源_抽象类_04

静态内部类


spring boot clickhouse 多数据源 spring boot多个数据源_sprint boot 多数据源切换_05

同时封装一个check()方法,用来监测过滤数据源配置

spring boot clickhouse 多数据源 spring boot多个数据源_抽象类_06


定义数据源动态配置类

我们沿用Spring Boot的自动化配置原理,定义数据源动态配置类,用我们的数据源配置信息生成一个个数据源实例(这里使用druid数据源)

spring boot clickhouse 多数据源 spring boot多个数据源_sprint boot 多数据源切换_07

spring boot clickhouse 多数据源 spring boot多个数据源_sprint boot 多数据源切换_08


自定义@DataSource注解

文章开始我们就说使用注解实现数据源切换,所以呢我们自定义注解@DataSource。敲黑板,注意这里的生效范围,一般业务使用是方法级别。

spring boot clickhouse 多数据源 spring boot多个数据源_sprint boot 多数据源切换_09


依赖AOP实现数据源持有器对数据源名称的update,黑板黑板,这里是重点

spring boot clickhouse 多数据源 spring boot多个数据源_自定义_10

spring boot clickhouse 多数据源 spring boot多个数据源_自定义_11




工程使用方法

  1. 上配置

  2. 上用法


    姿势A:默认使用主数据源

spring boot clickhouse 多数据源 spring boot多个数据源_sprint boot 多数据源切换_12


      姿势B:根据注解切换到myDatasource1数据源,方法结束自动切换到主数         据源(DataSourceAop后置实现)

      

spring boot clickhouse 多数据源 spring boot多个数据源_数据源_13