SpringBoot如何在运行时动态添加数据源
此方案适用于解决springboot项目运行时动态添加数据源,非静态切换多数据源!!!
一、多数据源应用场景:
1.配置文件配置多数据源,如默认数据源:master,数据源1:salve1...,运行时动态切换已配置的数据源(master、salve1互相切换),无法在运行时动态添加配置文件中未配置的数据源。
2.配置一个默认数据源,运行时动态添加新数据源使用(本博客适用于此场景)
二、解决方案:
Spring提供了AbstractRoutingDataSource用于动态路由数据源,第一种场景继承AbstractRoutingDataSource类并覆写其protectedabstractObjectdetermineCurrentLookupKey()即可;
而第二种场景我们直接覆写protectedDataSourcedetermineTargetDataSource方法即可。原理可看下AbstractRoutingDataSource对应源码,比较简单,不做赘述。
直接上干货:
importcom.fizz.utils.spring.SpringUtils; importorg.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource; importjavax.sql.DataSource; publicclassDynamicDataSourceextendsAbstractRoutingDataSource{ privatestaticfinalThreadLocaldataSource=ThreadLocal.withInitial(()->(DataSource)SpringUtils.getBean("defaultDataSource")); publicstaticvoidsetDataSource(DataSourcedataSource){ DynamicDataSource.dataSource.set(dataSource); } publicstaticDataSourcegetDataSource(){ returnDynamicDataSource.dataSource.get(); } @Override protectedObjectdetermineCurrentLookupKey(){ returnnull; } @Override protectedDataSourcedetermineTargetDataSource(){ returngetDataSource(); } publicstaticvoidclear(){ DynamicDataSource.dataSource.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.Primary; importjavax.sql.DataSource; importjava.util.HashMap; @Configuration publicclassDataSourceConfig{ @Bean @ConfigurationProperties("spring.datasource.druid") publicDataSourcedefaultDataSource(){ returnDruidDataSourceBuilder.create().build(); } @Bean @Primary publicDynamicDataSourcedataSource(){ DynamicDataSourcedynamicDataSource=newDynamicDataSource(); dynamicDataSource.setTargetDataSources(newHashMap<>()); 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.setDataSource(druidDataSource); 此时数据源已切换到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; } } publicstaticTgetBean(Class clazz){ returnSpringUtils.applicationContext.getBean(clazz); } publicstaticObjectgetBean(Stringname){ returnSpringUtils.applicationContext.getBean(name); } publicstaticStringgetProperty(Stringkey){ returnSpringUtils.applicationContext.getEnvironment().getProperty(key); } }
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持毛票票。