此方案适用于解决springboot项目运行时动态添加数据源，非静态切换多数据源！！！
一、多数据源应用场景：
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 ThreadLocal<DataSource> dataSource = 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;@Configurationpublic 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;@Componentpublic 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 static <T> T getBean(Class<T> 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如何在运行时动态添加数据源的相关介绍，要了解更多springboot,运行时,动态添加数据源内容请登录学步园。