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