SpringBoot项目运行时动态添加切换数据源

配置默认数据源和动态数据源
在这里插入图片描述
配置DynamicDataSource,重写determineCurrentLookupKey方法(重要),通过他来找到dataSourceMap中对应数据源
在这里插入图片描述
DataSourceUtil充当DynamicDataSourceContextHolder,通过setDataSourceKey方法切换数据源,通过addDataSource添加数据源

@Component
public class DataSourceUtil {

    @Autowired
    DynamicDataSource dynamicDataSource;

    private final ThreadLocal<String> contextHolder = ThreadLocal.withInitial(() -> "defaultDataSource");

    /**
     * 切换数据源
     * @param key
     */
    public void setDataSourceKey(String key) {
        contextHolder.set(key);
        dynamicDataSource.afterPropertiesSet();
    }

    /**
     * 获取当前数据源
     * @return
     */
    public String getDataSourceKey() {
        return contextHolder.get();
    }

    /**
     * 切换到默认数据源
     */
    public void clearDataSourceKey() {
        contextHolder.remove();
    }

    /**
     * 根据租户ID设置数据源
     * @param tenantId
     * @return
     */
    public void addDataSource(String tenantId, DataSource dataSource) {
        // 没有数据源时添加数据源,有数据源直接使用
        if (!dynamicDataSource.isExistDataSource(tenantId)) {
            // 新增数据源
            dynamicDataSource.addDataSource(tenantId, dataSource);
        }
        // 切换数据源
        setDataSourceKey(tenantId);
        dynamicDataSource.afterPropertiesSet();
    }
}

通过setDataSourceKey切换到默认数据源

通过setDataSourceKey切换到对应的数据源
在这里插入图片描述
在这里插入图片描述

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值