SpringBoot动态切换数据源

逛窑子的李靖2023-08-04 13:24

SpringBoot动态切换数据源

1.需求

低代码服务需要给多套系统进行功能配置，要求表结构必须生成在对应系统的数据库中，所以表结构的生成需要动态的获取目标系统的数据库信息，切换当前数据源到目标数据库，然后再生成。

2.创建数据源配置类

java 复制代码 
package com.tyq.datasource.config.datasource;

import com.alibaba.druid.pool.DruidDataSource;
import com.tyq.datasource.config.Properties;
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;

import javax.annotation.Resource;
import javax.sql.DataSource;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;

/**
 * 向Spring容器中注入DruidConfiguration
 *
 * @author 谭永强
 */
@Configuration
@EnableAutoConfiguration(exclude = {DataSourceAutoConfiguration.class})
public class DruidConfiguration {

    @Resource
    private Properties properties;

    /**
     * 数据源(默认)
     */
    public DataSource dataSourceToDefault() {
        DruidDataSource datasource = new DruidDataSource();
        try {
            datasource.setQueryTimeout(0);
            datasource.setUrl("jdbc:oracle:thin:@192.168.0.30:1521/test2");
            datasource.setUsername("SCPS");
            datasource.setPassword("yldtscps");
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return datasource;
    }

    @Bean
    @Primary
    public DataSource dataSource() {
        Map<Object, Object> dataSourceMap = new HashMap<>(2);
        dataSourceMap.put("default", dataSourceToDefault());
        DynamicDatasource dynamicDatasource = new DynamicDatasource();
      	// 注入目标数据源，如果有多个数据源，直接加入map即可
        dynamicDatasource.setTargetDataSources(dataSourceMap);
        // 注入默认数据源
        dynamicDatasource.setDefaultTargetDataSource(dataSourceToDefault());
        return dynamicDatasource;
    }
}

3.切换数据源

SpringBoot动态切换数据源主要依靠AbstractRoutingDataSource类，这个抽象类中有一个属性为targetDataSources。该属性为Map结构，所有需要切换的数据源都存放在其中，根据指定的KEY进行切换。

在AbstractRoutingDataSource源码中，获取数据库连接是通过this.determineTargetDataSource().getConnection()去获取的，而this.determineTargetDataSource()方法中的DataSource则是通过determineCurrentLookupKey()方法返回的key值去前面的dynamicDatasource.setTargetDataSources(dataSourceMap);设置进去的map中查找的。

所以我们需要重写determineCurrentLookupKey()方法，如下：

java 复制代码 
package com.tyq.datasource.config.datasource;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

/**
 * 动态数据源
 *
 * @author 谭永强
 * @date 2023-08-03
 */
public class DynamicDatasource extends AbstractRoutingDataSource {

    @Override
    protected Object determineCurrentLookupKey() {
        return DynamicDatasourceHolder.getDataSource();
    }
}

4.切换数据源管理类

数据源属于公共资源，考虑到多线程的情况下，我们将数据源存储在【ThreadLocal】中，保证线程隔离。

java 复制代码 
package com.tyq.datasource.config.datasource;

import java.util.Queue;
import java.util.concurrent.ArrayBlockingQueue;

/**
 * 数据源切换管理
 *
 * @author 谭永强
 * @date 2023-08-03
 */
public class DynamicDatasourceHolder {
    /**
     * 保存数据源的映射
     */
    private final static ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>();

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

    public static void setDataSource(String dataSourceKey) {
        CONTEXT_HOLDER.set(dataSourceKey);
    }

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

5.使用案例

在Controller中通过DynamicDatasourceHolder指定当前需要使用哪个数据源，具体使用如下：

java 复制代码 
@RequestMapping("findById")
public User findById(String userId) {
  //指定数据源
  DynamicDatasourceHolder.setDataSource("default");
  if (ObjectUtils.isEmpty(userId)) {
      throw new ParamValidateException("userId不能为空");
  }
  User user = userService.findById(userId);
  //移除当前数据源
  DynamicDatasourceHolder.removeDataSource();
  return user;
}

上述案例中已经实现了如果动态切换数据源的过程，但是在实际开发中还是太繁琐，比如每个接口都必须添加相关切换数据源的代码，对代码的侵入性太高，下面我们通过AOP的形式去优化切换数据源的过程。

5.AOP切面拦截

通过AOP切面对方法的前置和后置做切换数据源的操作，这样就降低了与业务代码耦合、

xml 复制代码 
<!--AOP-->
<dependency>
  <groupId>org.aspectj</groupId>
  <artifactId>aspectjweaver</artifactId>
  <version>1.9.6</version>
</dependency>

注意：如果项目中无法导入@Aspect，则需要添加上述依赖。

java 复制代码 
package com.tyq.datasource.aop;

import com.alibaba.druid.pool.DruidDataSource;
import com.tyq.datasource.config.Properties;
import com.tyq.datasource.config.datasource.DynamicDatasource;
import com.tyq.datasource.config.datasource.DynamicDatasourceHolder;
import com.tyq.datasource.config.exception.ParamValidateException;
import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.stereotype.Component;
import org.springframework.util.ObjectUtils;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.sql.DataSource;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;

/**
 * 动态数据源切换AOP
 *
 * @author 谭永强
 * @date 2023-08-03
 */
@Aspect
@Component
public class DynamicDatasourceAop {

    @Autowired
    protected ApplicationContext applicationContext;
    @Resource
    private Properties properties;

    /**
     * 定义切点
     */
    @Pointcut("execution (* com.tyq.datasource.controller.*.*(..))")
    public void pointcut() {
    }

    /**
     * 前置处理
     */
    @Before(value = "pointcut()")
    public void beforeAdvice() {
        // 接收到请求，记录请求内容
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        if (ObjectUtils.isEmpty(attributes)) {
            return;
        }
        HttpServletRequest request = attributes.getRequest();
        String dataSourceKey = request.getParameter("dataSourceKey");
        if (ObjectUtils.isEmpty(dataSourceKey)) {
            throw new ParamValidateException("dataSourceKey不能为空!");
        }
        //获取当前动态数据源
        DynamicDatasource dynamicDatasource = applicationContext.getBean(DynamicDatasource.class);
        //所有已连接的数据源集合
        Map<Object, DataSource> resolvedDataSources = dynamicDatasource.getResolvedDataSources();
        if (ObjectUtils.isEmpty(resolvedDataSources.get(dataSourceKey))) {
            //此处为模拟到数据库中查询数据源信息的过程，查询到数据源信息后，创建数据源并添加到动态数据源管理中
            //动态加入新的数据源
            DataSource dataSource = getDataSource();
            Map<Object, Object> dataSourceMap = new HashMap<>(resolvedDataSources.size() + 1);
            dataSourceMap.putAll(resolvedDataSources);
            dataSourceMap.put("three", dataSource);
            dynamicDatasource.setTargetDataSources(dataSourceMap);
        }
        //动态切换数据源
        DynamicDatasourceHolder.setDataSource(dataSourceKey);
    }

    /**
     * 后置处理
     */
    @AfterReturning(value = "pointcut()")
    public void afterAdvice() {
        // 接收到请求，记录请求内容
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        if (ObjectUtils.isEmpty(attributes)) {
            return;
        }
        HttpServletRequest request = attributes.getRequest();
        String dataSourceKey = request.getParameter("dataSourceKey");
        if (ObjectUtils.isEmpty(dataSourceKey)) {
            throw new ParamValidateException("dataSourceKey不能为空!");
        }
        System.out.println("后置处理:" + dataSourceKey);
        //删除当前数据源
        DynamicDatasourceHolder.removeDataSource();
    }

    /**
     * 模拟成功查询到数据源并创建DataSource
     *
     * @return 数据源
     */
    public DataSource getDataSource() {
        DruidDataSource datasource = new DruidDataSource();
        try {
            datasource.setQueryTimeout(0);
            datasource.setUrl("jdbc:oracle:thin:@192.168.0.59:1526/qstest");
            datasource.setUsername("SCPS");
            datasource.setPassword("yldtscps");
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return datasource;
    }
}
相关推荐
带多刺的玫瑰15 分钟前
Leecode刷题C语言之统计不是特殊数字的数字数量
java·c语言·算法
计算机毕设指导61 小时前
基于 SpringBoot 的作业管理系统【附源码】
java·vue.js·spring boot·后端·mysql·spring·intellij-idea
Gu Gu Study1 小时前
枚举与lambda表达式，枚举实现单例模式为什么是安全的，lambda表达式与函数式接口的小九九~
java·开发语言
Chris _data1 小时前
二叉树oj题解析
java·数据结构
牙牙7051 小时前
Centos7安装Jenkins脚本一键部署
java·servlet·jenkins
paopaokaka_luck1 小时前
[371]基于springboot的高校实习管理系统
java·spring boot·后端
以后不吃煲仔饭1 小时前
Java基础夯实——2.7 线程上下文切换
java·开发语言
进阶的架构师1 小时前
2024年Java面试题及答案整理（1000+面试题附答案解析）
java·开发语言
The_Ticker2 小时前
CFD平台如何接入实时行情源
java·大数据·数据库·人工智能·算法·区块链·软件工程
大数据编程之光2 小时前
Flink Standalone集群模式安装部署全攻略
java·大数据·开发语言·面试·flink
热门推荐
01(欧拉)openEuler系统添加网卡文件配置流程、(欧拉)openEuler系统手动配置ipv6地址流程、(欧拉)openEuler系统网络管理说明02PyTorch机器学习实现液态神经网络03【HarmonyOS】HUAWEI DevEco Studio 下载地址汇总04玄机平台应急响应—webshell查杀05Coze扣子平台完整体验和实践（附国内和国际版对比）06Ubuntu 20.04使用Livox mid 360 测试 FAST_LIO07【目标跟踪】相机运动补偿08Unity中PICO实现 隔空取物 和 接触抓取物体09RAG 实践- Ollama+RagFlow 部署本地知识库10怎样让音频速度变慢？请跟随以下方法进行操作