springboot配置多数据源

 springboot配置多数据源

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

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

1-yml文件

spring:datasource:type: com.alibaba.druid.pool.DruidDataSourcedriverClassName: com.mysql.cj.jdbc.Driverdruid:# 主库数据源master:url: jdbc:mysql:/306/andy?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8username: rootpassword:# 从库数据源slave:# 从数据源开关/默认关闭enabled: trueurl: jdbc:mysql/demo_project?useUnicode=true&characterEncodhavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8username: rootpassword: slave-aaa:# 从数据源开关/默认关闭enabled: true
#                driverClassName: org.postgresql.Driverurl: jdbc:mysql:/_databaseuseUnicode=true&characterEncoding=utf8&avior=convertToNull&useSSL=true&serverTimezone=GMT%2B8username: rootpassword: # 初始连接数initialSize: 5# 最小连接池数量minIdle: 10# 最大连接池数量maxActive: 20# 配置获取连接等待超时的时间maxWait: 60000# 配置连接超时时间connectTimeout: 30000# 配置网络超时时间socketTimeout: 60000# 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒timeBetweenEvictionRunsMillis: 60000# 配置一个连接在池中最小生存的时间，单位是毫秒minEvictableIdleTimeMillis: 300000# 配置一个连接在池中最大生存的时间，单位是毫秒maxEvictableIdleTimeMillis: 900000# 配置检测连接是否有效validationQuery: SELECT 1 FROM DUALtestWhileIdle: truetestOnBorrow: falsetestOnReturn: falsewebStatFilter: enabled: truestatViewServlet:enabled: true# 设置白名单，不填则允许所有访问allow:url-pattern: /druid/*# 控制台管理用户名和密码login-username: Andy_Fan login-password: 123ss456filter:stat:enabled: true# 慢SQL记录log-slow-sql: trueslow-sql-millis: 1000merge-sql: truewall: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")@Primarypublic 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();}@Overrideprotected 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
{@Resourceprivate 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] - -----清除数据源

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