SpringBoot在运行时动态添加数据源的方法
这篇文章主要介绍了SpringBoot在运行时动态添加数据源的方法,具有一定借鉴价值,感兴趣的朋友可以参考下,希望大家阅读完这篇文章之后大有收获,下面让小编带着大家一起了解一下。
专业从事网站设计制作、成都网站设计,高端网站制作设计,小程序定制开发,网站推广的成都做网站的公司。优秀技术团队竭力真诚服务,采用H5高端网站建设+CSS3前端渲染技术,响应式网站,让网站在手机、平板、PC、微信下都能呈现。建站过程建立专项小组,与您实时在线互动,随时提供解决方案,畅聊想法和感受。
一、多数据源应用场景:
1.配置文件配置多数据源,如默认数据源:master,数据源1:salve1...,运行时动态切换已配置的数据源(master、salve1互相切换),无法在运行时动态添加配置文件中未配置的数据源。
2.配置一个默认数据源,运行时动态添加新数据源使用(本博客适用于此场景)
二、解决方案:
Spring提供了AbstractRoutingDataSource用于动态路由数据源,第一种场景继承AbstractRoutingDataSource类并覆写其protected abstract Object determineCurrentLookupKey()即可;
而第二种场景我们直接覆写protected DataSource determineTargetDataSource方法即可。原理可看下AbstractRoutingDataSource对应源码,比较简单,不做赘述。
直接上干货:
import com.fizz.utils.spring.SpringUtils; import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource; import javax.sql.DataSource; public class DynamicDataSource extends AbstractRoutingDataSource { private static final ThreadLocaldataSource = ThreadLocal.withInitial(() -> (DataSource) SpringUtils.getBean("defaultDataSource")); public static void setDataSource(DataSource dataSource) { DynamicDataSource.dataSource.set(dataSource); } public static DataSource getDataSource() { return DynamicDataSource.dataSource.get(); } @Override protected Object determineCurrentLookupKey() { return null; } @Override protected DataSource determineTargetDataSource() { return getDataSource(); } public static void clear() { DynamicDataSource.dataSource.remove(); } }
import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder; import org.springframework.boot.context.properties.ConfigurationProperties; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Primary; import javax.sql.DataSource; import java.util.HashMap; @Configuration public class DataSourceConfig { @Bean @ConfigurationProperties("spring.datasource.druid") public DataSource defaultDataSource() { return DruidDataSourceBuilder.create().build(); } @Bean @Primary public DynamicDataSource dataSource() { DynamicDataSource dynamicDataSource = new DynamicDataSource(); dynamicDataSource.setTargetDataSources(new HashMap<>()); return dynamicDataSource; } }
使用时直接调用DynamicDataSource.setDataSource(DataSource dataSource)方法即可,使用完后调用DynamicDataSource.clear()防止内存泄漏并重置默认数据源。
附上详细使用方法:
DruidDataSource druidDataSource = new DruidDataSource(); 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 :
import lombok.Getter; import org.springframework.beans.BeansException; import org.springframework.context.ApplicationContext; import org.springframework.context.ApplicationContextAware; import org.springframework.stereotype.Component; @Component public final class SpringUtils implements ApplicationContextAware { @Getter private static ApplicationContext applicationContext; @Override public void setApplicationContext(ApplicationContext applicationContext) throws BeansException { if (SpringUtils.applicationContext == null) { SpringUtils.applicationContext = applicationContext; } } public staticT getBean(Class clazz) { return SpringUtils.applicationContext.getBean(clazz); } public static Object getBean(String name) { return SpringUtils.applicationContext.getBean(name); } public static String getProperty(String key) { return SpringUtils.applicationContext.getEnvironment().getProperty(key); } }
感谢你能够认真阅读完这篇文章,希望小编分享的“SpringBoot在运行时动态添加数据源的方法”这篇文章对大家有帮助,同时也希望大家多多支持创新互联,关注创新互联行业资讯频道,更多相关知识等着你来学习!
当前标题:SpringBoot在运行时动态添加数据源的方法
分享URL:http://pwwzsj.com/article/psgeds.html