首页 文章资讯内容详情

SpringBoot运行时动态添加数据源

2026-06-01 4 花语

本文内容纲要:

-此方案适用于解决springboot项目运行时动态添加数据源,非静态切换多数据源!!! -一、多数据源应用场景: -1.配置文件配置多数据源,如默认数据源:master,数据源1:salve1...,运行时动态切换已配置的数据源(master、salve1互相切换),无法在运行时动态添加配置文件中未配置的数据源。 -2.配置一个默认数据源,运行时动态添加新数据源使用(本博客适用于此场景) -二、解决方案: -使用时直接调用DynamicDataSource.setDataSource(DataSourcedataSource)方法即可,使用完后调用DynamicDataSource.clear()防止内存泄漏并重置默认数据源。 -附上详细使用方法: -附上工具类SpringUtils:

此方案适用于解决springboot项目运行时动态添加数据源,非静态切换多数据源!!!

一、多数据源应用场景:

1.配置文件配置多数据源,如默认数据源:master,数据源1:salve1...,运行时动态切换已配置的数据源(master、salve1互相切换),无法在运行时动态添加配置文件中未配置的数据源。

2.配置一个默认数据源,运行时动态添加新数据源使用(本博客适用于此场景)

二、解决方案:

Spring提供了AbstractRoutingDataSource用于动态路由数据源,第一种场景继承AbstractRoutingDataSource类并覆写其protectedabstractObjectdetermineCurrentLookupKey()即可;

而第二种场景我们直接覆写protectedDataSourcedetermineTargetDataSource方法即可。原理可看下AbstractRoutingDataSource对应源码,比较简单,不做赘述。

直接上干货: importcom.yaoshun.util.spring.SpringUtils; importorg.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource; importjava.util.Map; importjava.util.concurrent.ConcurrentHashMap; /** *<p>使用步骤:</p> *<blockquote><pre> *DynamicDataSource.dataSourcesMap.put(dataSourceKey,druidDataSource); *DynamicDataSource.setDataSource(dataSourceKey); *调用业务代码</i> *DynamicDataSource.clear(); *</pre></blockquote> * */ publicclassDynamicDataSourceextendsAbstractRoutingDataSource{ privatestaticfinalThreadLocal<String>dataSourceKey=ThreadLocal.withInitial(()->"defaultDataSource"); publicstaticMap<Object,Object>dataSourcesMap=newConcurrentHashMap<>(10); static{ dataSourcesMap.put("defaultDataSource",SpringUtils.getBean("defaultDataSource")); } @Override protectedObjectdetermineCurrentLookupKey(){ returnDynamicDataSource.dataSourceKey.get(); } publicstaticvoidsetDataSource(StringdataSource){ DynamicDataSource.dataSourceKey.set(dataSource); DynamicDataSourcedynamicDataSource=(DynamicDataSource)SpringUtils.getBean("dataSource"); dynamicDataSource.afterPropertiesSet(); } publicstaticStringgetDataSource(){ returnDynamicDataSource.dataSourceKey.get(); } publicstaticvoidclear(){ DynamicDataSource.dataSourceKey.remove(); } } importcom.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder; importorg.springframework.boot.context.properties.ConfigurationProperties; importorg.springframework.context.annotation.Bean; importorg.springframework.context.annotation.Configuration; importorg.springframework.context.annotation.DependsOn; importorg.springframework.context.annotation.Primary; importjavax.sql.DataSource; @Configuration publicclassDataSourceConfig{ @Bean @ConfigurationProperties("spring.datasource.druid") publicDataSourcedefaultDataSource(){ returnDruidDataSourceBuilder.create().build(); } @Bean @Primary @DependsOn({"springUtils","defaultDataSource"}) publicDynamicDataSourcedataSource(){ DynamicDataSourcedynamicDataSource=newDynamicDataSource(); dynamicDataSource.setTargetDataSources(DynamicDataSource.dataSourcesMap); returndynamicDataSource; } }

使用时直接调用DynamicDataSource.setDataSource(DataSourcedataSource)方法即可,使用完后调用DynamicDataSource.clear()防止内存泄漏并重置默认数据源。

附上详细使用方法:

DruidDataSourcedruidDataSource=newDruidDataSource(); druidDataSource.setUrl("jdbc:mysql://localhost:3306/sys?characterEncoding=utf-8&useSSL=false&serverTimezone=UTC&useAffectedRows=true"); druidDataSource.setUsername("root"); druidDataSource.setPassword("root"); DynamicDataSource.dataSourcesMap.put("dbkey",druidDataSource); DynamicDataSource.setDataSource("dbkey"); 此时数据源已切换到druidDataSource,调用自己的业务方法即可。 使用完后调用DynamicDataSource.clear();重置为默认数据源。

附上工具类SpringUtils:

importlombok.Getter; importorg.springframework.beans.BeansException; importorg.springframework.context.ApplicationContext; importorg.springframework.context.ApplicationContextAware; importorg.springframework.stereotype.Component; @Component publicfinalclassSpringUtilsimplementsApplicationContextAware{ @Getter privatestaticApplicationContextapplicationContext; @Override publicvoidsetApplicationContext(ApplicationContextapplicationContext)throwsBeansException{ if(SpringUtils.applicationContext==null){ SpringUtils.applicationContext=applicationContext; } } publicstatic<T>TgetBean(Class<T>clazz){ returnSpringUtils.applicationContext.getBean(clazz); } publicstaticObjectgetBean(Stringname){ returnSpringUtils.applicationContext.getBean(name); } publicstaticStringgetProperty(Stringkey){ returnSpringUtils.applicationContext.getEnvironment().getProperty(key); } }

本文内容总结:此方案适用于解决springboot项目运行时动态添加数据源,非静态切换多数据源!!!,一、多数据源应用场景:,1.配置文件配置多数据源,如默认数据源:master,数据源1:salve1...,运行时动态切换已配置的数据源(master、salve1互相切换),无法在运行时动态添加配置文件中未配置的数据源。,2.配置一个默认数据源,运行时动态添加新数据源使用(本博客适用于此场景),二、解决方案:,使用时直接调用DynamicDataSource.setDataSource(DataSourcedataSource)方法即可,使用完后调用DynamicDataSource.clear()防止内存泄漏并重置默认数据源。,附上详细使用方法:,附上工具类SpringUtils:,

原文链接:https://www.cnblogs.com/tswhq/p/11668078.html