Springboot+mybatis-plus+dynamic-datasource+Hikari 手动切换数据源

已于 2024-12-10 14:38:18 修改
阅读量2.5k 收藏 27
点赞数 9
分类专栏： 服务器 java0基础 文章标签： spring boot mybatis java
于 2024-05-10 09:46:11 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_36120342/article/details/138650080
版权

文章目录

Springboot+mybatis-plus+dynamic-datasource+Hikari 手动切换数据源

0.前言

苞米豆团队 dynamic-datasource 支持多种数据源切换方案，核心都是基于DynamicDataSourceContextHolder。本文我们利用filter和拦截器，以及方法中硬编码和注解 这四种方式动态手动切换数据源。

在这里插入图片描述

<dependency>
        <groupId>com.baomidou</groupId>
        <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
        <version>${dynamic.datasource.version}</version>
   </dependency>

1.多数据源核心类浅析

在写切换方法之前，我们先来了解一下dynamic-datasource 的两个核心类。

1.1.DynamicDataSourceContextHolder切换数据源核心类

DynamicDataSourceContextHolder 用于动态切换数据源的上下文工具类。在使用多数据源的情况下，它可以帮助在运行时选择要使用的数据源。

  1. 存储当前线程的数据源标识：DynamicDataSourceContextHolder使用线程本地变量（ThreadLocal）来存储当前线程所使用的数据源标识。这允许在同一应用程序中的不同线程使用不同的数据源。
 ThreadLocal<Deque<String>> LOOKUP_KEY_HOLDER = new NamedThreadLocal<Deque<String>>("dynamic-datasource")

  1. 通过调用**DynamicDataSourceContextHolder.setDataSourceKey(String dataSourceKey)**方法，可以将当前线程的数据源标识设置为指定的值。数据源标识通常是一个字符串，用于识别要使用的具体数据源。

  2. 通过调用DynamicDataSourceContextHolder.getDataSourceKey()方法，可以获取当前线程正在使用的数据源标识。这对于在代码中动态选择数据源非常有用。

  3. 在使用完特定数据源后，通过调用DynamicDataSourceContextHolder.clearDataSourceKey()方法，可以清除当前线程的数据源标识。这将避免数据源标识被错误地保留在其他线程中。
    在这里插入图片描述

1.2. DynamicRoutingDataSource

基于Spring 的JDBC 提供的AbstractDataSource类来创建自定义的数据源实现.
·DynamicRoutingDataSource·通过扩展AbstractRoutingDataSource，可以自定义路由规则。您可以实现**determineCurrentLookupKey()**方法，根据特定的规则选择要使用的数据源标识（如数据库名称、租户ID等）。根据路由规则，每个数据访问操作将使用相应的数据源。DynamicRoutingDataSource可根据运行时的条件或业务需求动态切换数据源。通过更新或改变路由规则，您可以实现动态切换数据源，以适应不同的场景或需求。例如，根据请求的租户ID选择不同的数据库，或者根据时间段选择不同的读写数据源。我们打开源码可以看到此处已经包含了所有数据库。
在这里插入图片描述

2.基于核心类的理解我们实现自定义的数据源切换方案

2.1. 在过滤器[filter]里切换

在使用 dynamic-datasource 库时，您可以通过过滤器（Filter）来实现在请求处理过程中切换数据源。下面是一种基本的实现方式：

创建 DynamicDataSourceFilter 在过滤器的 doFilter 方法中，获取当前请求的上下文信息，例如请求参数、请求头等。
根据上下文信息，判断应该使用哪个数据源，然后调用 DynamicDataSourceContextHolder.push(dataSourceKey ) 方法设置数据源标识。

  @Component
  @WebFilter(filterName = "dsFilter", urlPatterns = {"/userInfo/*"})
  public class DynamicDataSourceFilter implements Filter {
      @Override
      public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
          // 获取上下文信息，判断应该使用哪个数据源
          String dataSourceKey = determineDataSourceKey(request);
          
          // 设置数据源标识
          DynamicDataSourceContextHolder.push(dataSourceKey );
          
          try {
              // 继续处理请求
              chain.doFilter(request, response);
          } finally {
              // 清除数据源标识
           DynamicDataSourceContextHolder.poll();
          }
      }
  
      // 根据请求信息确定数据源标识
      private String determineDataSourceKey(ServletRequest request) {
          // 根据请求参数、请求头等进行逻辑判断，返回相应的数据源标识
          // ...
      }
  
      // 其他方法实现，如初始化和销毁方法
      // ...
  }
从Header 中获取数据库标识
import com.baomidou.dynamic.datasource.DynamicRoutingDataSource;
import com.baomidou.dynamic.datasource.annotation.DS;
import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceProperties;
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.servlet.*;
import javax.servlet.annotation.WebFilter;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;

@Component
@WebFilter(filterName = "dsFilter", urlPatterns = {"/userInfo/*"})
public class DynamicDataSourceFilter implements Filter {
    private static final String DATA_SOURCE_HEADER = "X-Data-Source"; // 假设 Header 名称为 "X-Data-Source"

    @Autowired
    private DynamicRoutingDataSource dynamicRoutingDataSource;

    @Override
    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain)
            throws IOException, ServletException {
        HttpServletRequest request = (HttpServletRequest) servletRequest;
        String dataSourceKey = determineDataSourceKey(request);

        try {
            // 切换数据源
            DynamicDataSourceContextHolder.push(dataSourceKey);
            
            filterChain.doFilter(servletRequest, servletResponse);
        } finally {
            // 恢复默认数据源
            DynamicDataSourceContextHolder.poll();
        }
    }

    private String determineDataSourceKey(HttpServletRequest request) {
        String dataSourceKey = request.getHeader(DATA_SOURCE_HEADER);
        // 根据实际需求进行数据源 key 的处理，例如校验、转换等

        return dataSourceKey;
    }
}

2.2. 拦截器里切换数据源

可以使用DynamicDataSourceContextHolder 在拦截器中使用

DataSourceInterceptor的拦截器
import org.springframework.web.servlet.HandlerInterceptor;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

public class DataSourceInterceptor implements HandlerInterceptor {

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
        String dataBaseCode = request.getHeader("dataBaseCode");
        DynamicDataSourceContextHolder.push(dataBaseCode);
        return true;
    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) {
        DynamicDataSourceContextHolder.poll();
    }
}

在Spring的配置中注册这个拦截器：

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;

@Configuration
public class WebMvcConfig implements WebMvcConfigurer {

    @Autowired
    private DataSourceInterceptor dataSourceInterceptor;

    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(dataSourceInterceptor).addPathPatterns("/**");
    }
}

2.3. 方法内部硬编码硬编码切换

 if (dbHandelVO.getParam() == null) {
            return WrapperUtil.ok();
        }
        String dataBaseCode = dbHandelVO.getDataBaseCode();
        if (!StringUtils.isEmpty(dataBaseCode)) {
            DynamicDataSourceContextHolder.push(dataBaseCode);
        }
        try {
            dbHandelMapper.executeInsert(dbHandelVO.getParam());
        } catch (Exception e) {
            throw new E6Exception(e.getMessage(), e);
        } finally {
            //移除当前数据源
            if (!StringUtils.isEmpty(dataBaseCode)) {
                DynamicDataSourceContextHolder.poll();
            }
        }

2.4. 注解进行切换

每个service 和mapper加入注解@DS

service调用本类的方法不会有切面进行切换数据源

    @Bean
    public DsProcessor dsProcessor(){
        TenantDsProcessor tenantDsProcessor = new TenantDsProcessor();
        tenantDsProcessor.setNextProcessor(new DsSpelExpressionProcessor());
        return tenantDsProcessor;
    }


@RequiredArgsConstructor
@Slf4j
public class TenantDsProcessor extends DsProcessor {

    /**
     * 用于获取租户数据源配置的 Service
     */
    @Resource
    @Lazy
    private TenantDatasourceCache tenantDatasourceCache;

    /**
     * 动态数据源
     */
    @Resource
    @Lazy  
    private DynamicRoutingDataSource dynamicRoutingDataSource;

    /**
     * 用于创建租户数据源的 Creator
     */
    @Resource
    @Lazy
    private DefaultDataSourceCreator dataSourceCreator;

    @Override
    public boolean matches(String key){
        return Objects.equals(key, DS.KEY);
    }

    @Override
    public String doDetermineDatasource(MethodInvocation methodInvocation, String s){
        // 获得数据源配置
        String tenantId = TenantContextHolder.getRequiredTenantId();
        if (tenantId.equals(GlobalConsts.MAIN_TENANT_ID)) {
            return "master";
        }
        DataSourceProperty dataSourceProperty = tenantDatasourceCache.getDataSourceProperty(tenantId);
        // 创建 or 创建数据源，并返回数据源名字
        return createDatasourceIfAbsent(dataSourceProperty);
    }

}

博主自己的改造方式

3. 参考资料

  1. dynamic-datasource GitHub 仓库 ↗：dynamic-datasource 的官方 GitHub 仓库，包含源代码、文档和示例等资源。

借鉴地址https://blog.csdn.net/wangshuai6707/article/details/132669192

博客地址

代码下载

下面的springcloud_dynamic_datasource

在MyBatis-Plus中实现多数据源切换
三心二逸的博客
04-24 1213
在复杂的业务场景中，我们可能需要从不同的数据库获取数据。MyBatis-Plus提供了一种便捷的方式来实现这一需求。本文将介绍如何在MyBatis-Plus中配置和使用多数据源。
springBoot如何动态切换数据源
爪哇人的博客
01-18 3522
mybatisplus提供的有插件用@DS注解就能够实现，但是这种在mysql服务宕机的情况下不能够进行自动切换，于是就想着用aop+自定义注解的方式来实现。此时你在代码层面用注解指定了一个增删改方法到从数据源，但是碰巧此时从数据源失效了，那么就会自动的切换到其它服务器。DynamicDataSourceHolder类主要是设置当前线程的数据源名称，移除数据源名称，以及获取当前数据源的名称，便于动态切换。我在这个方法上指定的是从数据库，如果此时从数据库发生宕机，那么就会自动切换到主数据库进行操作。
dynamic动态数据源编码切换数据源
pleaseprintf的博客
03-21 1358
前些天发现了一个巨牛的人工智能学习网站，通俗易懂，风趣幽默，忍不住分享一下给大家。点击跳转到网站，这篇文章男女通用，看懂了就去分享给你的码吧。dynamic特性支持 数据源分组 ，适用于多种场景 纯粹多库 读写分离 一主多从 混合模式。支持数据库敏感配置信息 加密(可自定义) ENC()。支持每个数据库独立初始化表结构schema和数据库database。支持无数据源启动，支持懒加载数据源（需要的时候再创建连接）。支持 自定义注解 ，需继承DS(3.2.0+)。
使用dynamic-datasource-spring-boot-starter，通过接口实现动态切换数据源
weixin_61933730的博客
04-14 357
使用dynamic-datasource-spring-boot-starter，通过接口实现动态切换数据源
【mybatis-plus】多数据源切换[dynamic-datasource] 手动切换数据源
热门推荐
日拱一卒无有尽，功不唐捐终入海
09-04 1万+
苞米豆团队支持多种数据源切换方案，核心都是基于。本文我们利用filter和拦截器，以及方法中硬编码 这三种方式动态手动切换数据源。在使用 dynamic-datasource 库时，您可以通过过滤器（Filter）来实现在请求处理过程中切换数据源。创建在过滤器的doFilter方法中，获取当前请求的上下文信息，例如请求参数、请求头等。根据上下文信息，判断应该使用哪个数据源，然后调用方法设置数据源标识。@Component@Override// 获取上下文信息，判断应该使用哪个数据源。
使用Mybatis-Plus实现动态多数据源切换
abelethan的博客
05-24 3863
本文将介绍如何使用Mybatis-Plus配合Spring Boot来实现动态多数据源的切换。我们将讨论依赖引入、配置修改、自定义数据源提供者、配置类编写、数据源工具类实现以及AOP切面编程的应用。通过上述步骤，我们已经成功地实现了Mybatis-Plus配合Spring Boot的动态多数据源切换。在现代的企业级应用中，常常需要根据不同的业务场景动态地切换数据源。Mybatis-Plus提供了一个基于Spring Boot的快速集成多数据源的启动器，在请求拦截器里面添加线程请求的机构。
mybatis plus 配置多数据源(数据源进行切换)
qq_42152032的博客
05-24 2264
AbstractRoutingDataSource 根据用户定义的规则选择当前的数据源，这样我们可以在执行查询之前，设置使用的数据源。//log.info("当前数据源 [{}]", DynamicDataSourceContextHolder.getDataSourceKey());) // MybatisSqlSessionFactoryBean中有各种MybatisPlus的配置属性(globalConfig、//sqlSession就是用来操作sql语句的。2、主启动类添加注解。
springBoot-Mybatis-Plus 多数据源切换实现
拽着尾巴的鱼儿的博客
05-19 3846
封装数据库连接池和连接的获取逻辑：AbstractRoutingDataSource 通过封装多个数据源连接池的实现细节，屏蔽底层数据源的细节，使得业务代码不需要关心连接的获取和释放，从而简化了业务代码的编写。(3). 实现数据源的动态切换：AbstractRoutingDataSource 可以通过动态切换数据源，实现数据源的动态切换，从而在不影响系统正常运行的情况下，能够对数据源进行升级、迁移等操作。2.5 请求：header 头放入本次的db。
SpringBoot整合mybatis-plus实现多数据源的动态切换且支持分页查询.pdf
10-07
在SpringBoot项目中，整合Mybatis-Plus并实现多数据源的动态切换，同时支持分页查询是一项常见的需求。以下将详细阐述这个过程中的关键步骤和技术要点。 首先，我们需要引入必要的Maven依赖。这里提到了四个关键...
SpringBoot+Mybatis-Plus整合Sharding-JDBC5.1.1实现分库分表
BUG指挥课堂
06-06 1315
SpringBoot+Mybatis-Plus整合Sharding-JDBC4.0.0实现单库分表
SpringBoot+Mybatis-Plus整合Sharding-JDBC实现分库分表
qq_70009637的博客
04-17 888
先建一个数据库,创建三张表 把表名改成对应的。首先创建一个springboot单体工程。执行之后会把数据随机存入到一张表中。
Spring-boot+mybatis多数据源整合（动态切换数据源）
08-12
Spring-boot+mybatis多数据源整合（动态切换数据源）,亲测可用
在 Spring Boot 中配置多个数据源并使用 MyBatis-Plus 进行数据源切换
最新发布
一碗黄焖鸡三碗米饭的博客
04-15 1657
配置多个数据源并在 Spring Boot 中进行切换是一个常见的需求，尤其是在涉及多数据库架构或者分库分表的场景下。通过合理使用@Primary@Qualifier和配置，配合第三方动态数据源库，我们可以轻松实现数据源切换。最重要的是，要根据实际需求选择合适的配置方式，确保数据源的切换逻辑清晰、稳定。希望本文能够帮助你解决 MyBatis-Plus 无法切换多个数据源的问题，提升你的项目配置和开发效率。使用 MyBatis-Plus 分页插件时返回所有数据的解决方案-CSDN博客。
Mybatis plus 多数据源切换
chengmin123456789的博客
08-19 2961
dynamic-datasource-spring-boot-starter 是一个基于springboot的快速集成多数据源的启动器。 其支持Jdk 1.7+, SpringBoot 1.4.x 1.5.x 2.x.x。 步骤1、引入依赖（myabits plus依赖 、jdbc 或者mysql connector 依赖等另行按需引入） <dependency> <groupId>com.baomidou</groupId> .
mybatis多数据源配置_SpringBoot 整合 Mybatis 多数据源并动态切换
weixin_39604516的博客
11-26 303
作者：hy_xiaobin来源：https://juejin.im/post/5d8705e65188253f4b629f47首先需要建立两个库进行测试，我这里使用的是master_test和slave_test两个库，两张库都有一张同样的表(偷懒)，表名 t_user字段名类型备注idint主键自增IDnamevarchar名称表中分别添加两条不同数据，方便测试 主数据库记录name...
mp的切换多数据源
xlj778的博客
06-29 605
xxxx
mybatis-plus 多数据源切换
pemelapple的博客
02-22 369
mybatis-plus 多数据源动态切换
Mybatis多数据源动态切换，原来这么简单！
sssspider的博客
09-30 1481
首先需要建立两个库进行测试，我这里使用的是master_test和slave_test两个库，两张库都有一张同样的表（偷懒），表名 t_user 字段名 类型 备 注 id i nt 主键自增ID name varchar 名称 表中分别添加两条不同数据，方便测试 主数据库记录name为xiaobin，从库为xiaoliu。 目录结构 开始使用Springboot 整...
springboot集成mybatisplus三步实现多数据源切换
mifly1208的博客
10-05 628
第三步，在service类中使用@DS注解来实现数据源切换，注解可以写在类上，也可以写在方法上，方法上的优先级高于类，如果没有指定DS的属性值或者没有使用DS注解，就自动走master主库。多数据源是java项目中常遇到的需求，传统的方式我们需要编写大量代码来实现，过程比较复杂，mybatisplus的出现使得多数据源的实现十分快速简单。引入mybatisplus相关依赖，多数据源相关的是dynamic-datasource-spring-boot-starter依赖。第二步，yml文件中配置多数据源。
springboot使用mybatis-plus的d详细代码ynamic多数据源集成tcc实现事务的控制
09-20
Spring Boot结合MyBatis-Plus和动态数据源（Dynamic DataSource）可以方便地管理分布式系统中的事务。TCC（Try-Confirm-Cancel）是一种补偿型分布式事务策略。以下是集成步骤： 1. 添加依赖： 首先，在项目中添加Spring Cloud Data Flow、Spring Cloud Config和相关数据源库的依赖。 ```xml <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-data-jpa</artifactId> </dependency> <dependency> <groupId>com.baomidou</groupId> <artifactId>mybatis-plus-boot-starter</artifactId> <version>3.x</version> </dependency> <dependency> <groupId>org.springframework.cloud</groupId> <artifactId>spring-cloud-config-client</artifactId> </dependency> <dependency> <groupId>org.springframework.cloud</groupId> <artifactId>spring-cloud-starter-datasource</artifactId> </dependency> ``` 2. 配置动态数据源： 在`application.yml`中配置两个数据源（如dataSourceA和dataSourceB），并设置TCC相关的配置。 ```yaml spring: cloud: config: client: uri: http://config-server-url:8888 datasource: primary: dataSourceA secondary: name: dataSourceB type: com.zaxxer.hikari.HikariConfig driver-class-name: ... (对应数据库驱动) url: ... (数据库URL) username: ... (用户名) password: ... (密码) tcc: enabled: true retry-interval: 5000 # 重试间隔毫秒 timeout: 30000 # TCC超时时间 ``` 3. 定义实体和Mapper： 创建对应的实体类和Mapper接口，使用@TableGlobal注解标记全局分表。 4. 创建TCC服务： 创建一个Service类，注入多个Repository（每个Repository对应一个数据源）。实现try、confirm和cancel方法，并通过`@Transactional(rollbackFor = ...)`来指定事务隔离级别。 ```java @Service public class MyServiceImpl implements MyService { @Autowired private RepositoryA repositoryA; @Autowired private RepositoryB repositoryB; @Override @Transactional(tcc = true) // 使用TCC事务 public void doSomething() { try { // 尝试操作 dataSourceA repositoryA.doSomething(); // 如果确认成功，切换到dataSourceB if (confirmSuccess()) { repositoryB.confirm(); } } catch (Exception e) { // 如果抛出异常，回滚并取消 dataSourceA 的操作 cancel(); throw e; // 或者记录错误后重新抛出 } } // 其他辅助方法如confirmSuccess(), cancel() 等 } ``` 5. 配置TCC事务处理器： 在`application.yml`或`application.properties`中配置TCC事务处理器，例如使用HystrixCommand或FeignClient。 ```yaml tcc: transaction-manager: mybatis-tcc-transaction-manager command-factory: feign-tcc-command-factory ``` 然后，创建`MybatisTccTransactionManager`和`FeignTccCommandFactory`，它们将负责管理和协调各个数据源之间的事务处理。 以上就是基本的流程，具体的实现细节可能会因项目需求而略有调整。在实际应用中，记得测试所有边界情况，确保TCC模式下的事务处理能够正确运行。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

liuhm~

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

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

余额充值