如何在Spring运行时动态修改spring.datasource.url并重置数据库Bean？

一、基于AbstractRoutingDataSource的动态数据源切换 #

你提到的AbstractRoutingDataSource确实是Spring实现动态数据源切换的标准方案，核心是通过线程上下文的路由标识，在运行时切换不同的数据源实例。

1. 配置多数据源 #

先通过配置类加载两个目标数据源（my_schema1和my_schema2）：

@Configuration
public class DataSourceConfig {

    @Bean
    @ConfigurationProperties(prefix = "spring.datasource.schema1")
    public DataSource schema1DataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    @ConfigurationProperties(prefix = "spring.datasource.schema2")
    public DataSource schema2DataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    public DataSource routingDataSource(DataSource schema1DataSource, DataSource schema2DataSource) {
        AbstractRoutingDataSource routingDataSource = new AbstractRoutingDataSource() {
            @Override
            protected Object determineCurrentLookupKey() {
                // 从线程上下文获取当前数据源标识
                return DataSourceContextHolder.getDataSourceKey();
            }
        };

        // 绑定数据源与标识的映射关系
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put("schema1", schema1DataSource);
        targetDataSources.put("schema2", schema2DataSource);
        routingDataSource.setTargetDataSources(targetDataSources);
        routingDataSource.setDefaultTargetDataSource(schema1DataSource);
        return routingDataSource;
    }

    // 将路由数据源设为Spring默认数据源
    @Primary
    @Bean
    public DataSource dataSource(DataSource routingDataSource) {
        return routingDataSource;
    }
}

对应的配置文件补充：

spring.datasource.schema1.url=jdbc:mariadb://localhost/my_schema1
spring.datasource.schema1.username=xxx
spring.datasource.schema1.password=xxx

spring.datasource.schema2.url=jdbc:mariadb://localhost/my_schema2
spring.datasource.schema2.username=xxx
spring.datasource.schema2.password=xxx

2. 实现线程上下文持有者 #

用于存储和切换当前线程的数据源标识：

public class DataSourceContextHolder {
    private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>();

    public static void setDataSourceKey(String key) {
        CONTEXT_HOLDER.set(key);
    }

    public static String getDataSourceKey() {
        return CONTEXT_HOLDER.get();
    }

    public static void clearDataSourceKey() {
        CONTEXT_HOLDER.remove();
    }
}

3. 运行时切换数据源 #

在业务代码中直接设置上下文标识即可完成切换：

// 切换到my_schema2
DataSourceContextHolder.setDataSourceKey("schema2");
// 执行数据库操作（此时会使用my_schema2的连接）
// 操作完成后清除标识，避免线程复用导致的数据源混乱
DataSourceContextHolder.clearDataSourceKey();

二、重置并重新初始化数据库Bean #

如果需要运行时动态修改数据源配置（比如临时变更URL）并重新初始化Bean，可以通过Spring的Bean工厂手动管理Bean生命周期：

1. 动态刷新数据源实现 #

@Component
public class DataSourceRefresher {

    @Autowired
    private ConfigurableApplicationContext applicationContext;

    public void refreshDataSource(String newUrl) {
        ConfigurableBeanFactory beanFactory = applicationContext.getBeanFactory();
        // 销毁旧数据源并关闭连接
        if (beanFactory.containsBean("dataSource")) {
            DataSource oldDataSource = beanFactory.getBean(DataSource.class);
            try {
                oldDataSource.close();
            } catch (SQLException e) {
                // 处理连接关闭异常
            }
            beanFactory.destroyBean("dataSource");
        }

        // 创建新数据源实例
        DataSource newDataSource = DataSourceBuilder.create()
                .url(newUrl)
                .username("xxx")
                .password("xxx")
                .driverClassName("org.mariadb.jdbc.Driver")
                .build();

        // 重新注册数据源Bean到容器
        beanFactory.registerSingleton("dataSource", newDataSource);

        // 刷新依赖数据源的Bean（比如JdbcTemplate）
        if (beanFactory.containsBean("jdbcTemplate")) {
            beanFactory.destroyBean("jdbcTemplate");
            JdbcTemplate jdbcTemplate = new JdbcTemplate(newDataSource);
            beanFactory.registerSingleton("jdbcTemplate", jdbcTemplate);
        }
    }
}

调用时传入新URL即可完成重置：

dataSourceRefresher.refreshDataSource("jdbc:mariadb://localhost/my_schema2");

注意事项 #

• 这种方式会打破Spring单例Bean的生命周期，多线程环境下需加锁控制
• 若使用JPA/Hibernate，还需手动刷新EntityManagerFactory，可通过重新创建LocalContainerEntityManagerFactoryBean实现
• 纯Spring环境下建议用上述手动方式，若为Spring Cloud项目，可结合@RefreshScope简化配置刷新

三、两种方案对比 #

内容的提问来源于stack exchange，提问作者membersound