springboot配置多数据源

 springboot配置多数据源

学习新技术，争做新青年，欢迎围观，河南老乡在上海请，加，微，andyfau2022，

----获取数据源：null，数据源为null时默认使用主数据源的。

1-yml文件

spring:
    datasource:

        type: com.alibaba.druid.pool.DruidDataSource
        driverClassName: com.mysql.cj.jdbc.Driver
        druid:
            # 主库数据源
            master:
                url: jdbc:mysql:/306/andy?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8
                username: root
                password:
            # 从库数据源
            slave:
                # 从数据源开关/默认关闭
                enabled: true
                url: jdbc:mysql/demo_project?useUnicode=true&characterEncodhavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8
                username: root
                password: 
            slave-aaa:
                # 从数据源开关/默认关闭
                enabled: true
#                driverClassName: org.postgresql.Driver
                url: jdbc:mysql:/_databaseuseUnicode=true&characterEncoding=utf8&avior=convertToNull&useSSL=true&serverTimezone=GMT%2B8
                username: root
                password: 
            # 初始连接数
            initialSize: 5
            # 最小连接池数量
            minIdle: 10
            # 最大连接池数量
            maxActive: 20
            # 配置获取连接等待超时的时间
            maxWait: 60000
            # 配置连接超时时间
            connectTimeout: 30000
            # 配置网络超时时间
            socketTimeout: 60000
            # 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒
            timeBetweenEvictionRunsMillis: 60000
            # 配置一个连接在池中最小生存的时间，单位是毫秒
            minEvictableIdleTimeMillis: 300000
            # 配置一个连接在池中最大生存的时间，单位是毫秒
            maxEvictableIdleTimeMillis: 900000
            # 配置检测连接是否有效
            validationQuery: SELECT 1 FROM DUAL
            testWhileIdle: true
            testOnBorrow: false
            testOnReturn: false
            webStatFilter: 
                enabled: true
            statViewServlet:
                enabled: true
                # 设置白名单，不填则允许所有访问
                allow:
                url-pattern: /druid/*
                # 控制台管理用户名和密码
                login-username: Andy_Fan 
                login-password: 123ss456
            filter:
                stat:
                    enabled: true
                    # 慢SQL记录
                    log-slow-sql: true
                    slow-sql-millis: 1000
                    merge-sql: true
                wall:
                    config:
                        multi-statement-allow: tr

2-DruidConfig 配置

/**
 * druid 配置多数据源
 *
 * @author Andy_Fan
 *
 * 在需要切换数据源的方法上加上@DataSource注解----加在serviceImpl方法上，或者mapper
 *
 *    @DataSource(value = DataSourceType.SLAVE)
 *    @Select("select count(1) from public.\"LX_CJ_DK\"")
 */
@Slf4j
@Configuration
public class DruidConfig
{
   /**
    * 主库数据源
    * @param druidProperties
    * @return
    */
   @Bean
   @ConfigurationProperties("spring.datasource.druid.master")
   public DataSource masterDataSource( DruidProperties druidProperties )
   {
      log.info( "------主库数据源加载完成--masterDataSource-" );
      DruidDataSource dataSource = DruidDataSourceBuilder.create().build();
      return druidProperties.dataSource( dataSource );
   }

   /**
    * 从库数据源 1
    * @param druidProperties
    * @return
    */
   @Bean
   @ConfigurationProperties("spring.datasource.druid.slave")
   @ConditionalOnProperty(prefix = "spring.datasource.druid.slave", name = "enabled", havingValue = "true")
   public DataSource slaveDataSource( DruidProperties druidProperties )
   {
      log.info( "------从库数据源加载完成--slaveDataSource-" );
      DruidDataSource dataSource = DruidDataSourceBuilder.create().build();
      return druidProperties.dataSource( dataSource );
   }

   /**
    * 从库数据源 2 aaa
    * @param
    * @return
    */
   @Bean
   @ConfigurationProperties("spring.datasource.druid.slave-aaa")
   @ConditionalOnProperty(prefix = "spring.datasource.druid.slave-aaa", name = "enabled", havingValue = "true")
   public DataSource slaveAAADataSource( DruidProperties druidProperties )
   {
      log.info( "------从库数据源加载完成--slaveAAADataSource-" );
      DruidDataSource dataSource = DruidDataSourceBuilder.create().build();
      return druidProperties.dataSource( dataSource );
   }

   @Bean(name = "dynamicDataSource")
   @Primary
   public DynamicDataSource dataSource( DataSource masterDataSource, DataSource slaveDataSource, DataSource slaveAAADataSource )
   {
      log.info( "------主从数据源加载完成---" );
      Map< Object, Object > targetDataSources = new HashMap<>();
      targetDataSources.put( DataSourceType.MASTER.name(), masterDataSource );
      targetDataSources.put( DataSourceType.SLAVE.name(), slaveDataSource );
      targetDataSources.put( DataSourceType.SLAVE_AAA.name(), slaveAAADataSource );


      //        在需要切换数据源的方法上加上@DataSource注解,@DataSource(value = DataSourceType.SLAVE)
      return new DynamicDataSource( masterDataSource, targetDataSources );
   }

   /**
    * 设置数据源
    *
    * @param targetDataSources 备选数据源集合
    * @param sourceName 数据源名称
    * @param beanName bean名称
    */
   public void setDataSource( Map< Object, Object > targetDataSources, String sourceName, String beanName )
   {
      try
      {
         log.info( "------加载数据源: targetDataSources{}", targetDataSources + "------: sourceName {}", sourceName + "------: beanName{}", beanName );
         DataSource dataSource = SpringUtils.getBean( beanName );
         targetDataSources.put( sourceName, dataSource );
      }
      catch ( Exception e )
      {
         log.error( "------加载数据源失败, {}", e.getMessage() );
      }
   }
}

3- DynamicDataSource

``` 

/**
 * 动态数据源
 *
 * @author Andy_Fan
 * @date 2024/09/11
 */
@Slf4j
public class DynamicDataSource extends AbstractRoutingDataSource
{
   public DynamicDataSource( DataSource defaultTargetDataSource, Map< Object, Object > targetDataSources )
   {
      log.info( "-----------DynamicDataSource init" );
      super.setDefaultTargetDataSource( defaultTargetDataSource );
      super.setTargetDataSources( targetDataSources );
      super.afterPropertiesSet();
   }

   @Override
   protected Object determineCurrentLookupKey()
   {
      log.info( "------DynamicDataSource determineCurrentLookupKey" );
      return DynamicDataSourceContextHolder.getDataSourceType();
   }
}

4-DynamicDataSourceContext

/**
 * 数据源切换处理
 * @author Andy_Fan
 * @date 2024/09/11
 */
public class DynamicDataSourceContextHolder
{
    public static final Logger log = LoggerFactory.getLogger(DynamicDataSourceContextHolder.class);

    /**
     * 使用ThreadLocal维护变量，ThreadLocal为每个使用该变量的线程提供独立的变量副本，所以每一个线程都可以独立地改变自己的副本，而不会影响其它线程所对应的副本。
     */
    private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>();

    /**
     * 设置数据源的变量
     *
     * 增加DataSourceType类里面的枚举类型
     public enum DataSourceType{MASTER,SLAVE,SLAVE_XZDB}
     *
     */
    public static void setDataSourceType(String dsType)
    {
        log.info("----切换数据源:", dsType);
        CONTEXT_HOLDER.set(dsType);
    }


    /**
     * 获得数据源的变量
     */
    public static String getDataSourceType()
    {
        log.info("-----获取数据源：{}", CONTEXT_HOLDER.get());
        return CONTEXT_HOLDER.get();
    }

    /**
     * 清空数据源变量
     */
    public static void clearDataSourceType()
    {
        log.info("-----清除数据源");
        CONTEXT_HOLDER.remove();
    }

}

5-测试

/**
 * <p>
 * 用户表 服务实现类
 * </p>
 *
 * @author Andy_Fan
 * @since 2024-09-12 14:52:06
 */
@Service
public class DemoUserServiceImpl extends ServiceImpl< DemoUserMapper, DemoUser > implements IDemoUserService
{

   @Resource
   private DemoUserMapper demoUserMapper;

   @Override
   @DataSource(value = DataSourceType.SLAVE)
   public List< DemoUser > selectList()
   {
      return demoUserMapper.selectAll( new DemoUser() );
   }
   
}

----切换到SLAVE数据源

-----获取数据源：SLAVE

----获取数据源：null

.r.f.d.DynamicDataSourceContextHolder - [setDataSourceType,28] - ----切换到SLAVE数据源
15:52:06.280 [http-nio-8080-exec-1] INFO  c.r.f.d.DynamicDataSourceContextHolder - [getDataSourceType,38] - -----获取数据源：SLAVE
15:52:06.283 [http-nio-8080-exec-1] DEBUG c.r.d.m.D.selectAll - [debug,135] - ==>  Preparing: select id, username, password, nickName, third_party_id, avatar, avatar_thumbnail, gender, user_type, last_login_time, registration_ip, deleted, version, create_date, create_by, update_date, update_by from demo_user
15:52:06.289 [http-nio-8080-exec-1] DEBUG c.r.d.m.D.selectAll - [debug,135] - ==> Parameters: 
15:52:06.429 [http-nio-8080-exec-1] DEBUG c.r.d.m.D.selectAll - [debug,135] - <==      Total: 2
15:52:06.430 [http-nio-8080-exec-1] INFO  c.r.f.d.DynamicDataSourceContextHolder - [clearDataSourceType,47] - -----清除数据源

一起加油，一起向前冲。。。