SpringBoot多数据源

SpringBoot多数据源

mybitsplus用户使用双数据源是很方便的可以直接使用mybits的双数据源
大家可以点进去看官方文档，但是有些文档地方是没有的或者收费的。

先看mybitsplus的实现 【官方文档节选】

  com.baomidou
  dynamic-datasource-spring-boot-starter
  ${version}

spring:
  datasource:
    dynamic:
      primary: master #设置默认的数据源或者数据源组,默认值即为master
      strict: false #严格匹配数据源,默认false. true未匹配到指定数据源时抛异常,false使用默认数据源
      datasource:
        master:
          url: jdbc:mysql://xx.xx.xx.xx:3306/dynamic
          username: root
          password: 123456
          driver-class-name: com.mysql.jdbc.Driver # 3.2.0开始支持SPI可省略此配置
        slave_1:
          url: jdbc:mysql://xx.xx.xx.xx:3307/dynamic
          username: root
          password: 123456
          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，一个组slave下有两个子库slave_1,slave_2

# 多主多从                      纯粹多库（记得设置primary）                   混合配置
spring:                               spring:                               spring:
  datasource:                           datasource:                           datasource:
    dynamic:                              dynamic:                              dynamic:
      datasource:                           datasource:                           datasource:
        master_1:                             mysql:                                master:
        master_2:                             oracle:                               slave_1:
        slave_1:                              sqlserver:                            slave_2:
        slave_2:                              postgresql:                           oracle_1:
        slave_3:                              h2:                                   oracle_2:

当你想要切换数据源时可以使用注解@DS()

@Target({ElementType.TYPE, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DS {

    /**
     * groupName or specific database name or spring SPEL name.
     *
     * @return the database you want to switch
     */
    String value();
}

可以使用在方法上面或者类上面。

当然也可以手动切换数据源， mybitsplus提供了DynamicDataSourceContextHolder 来进行手动切换

DynamicDataSourceContextHolder.push("master");
DynamicDataSourceContextHolder.poll();

当然里面放的参数名为你配置文件里写的数据源名称,如上面配置的master,slave_1,slave_2
使用之前调用push将数据源推进去，使用完成之后调用poll。
DynamicDataSourceContextHolder 的实现是一个ThreadLocal栈因此将数据源使用的时候push进去
使用完成之后记得poll清除。

当然如果你的代码逻辑不复杂根本不用这么麻烦，直接使用@DS注解会方便很多。