Spring Boot中实现多数据源连接和切换的方案

🕗 发布于 2024-11-08 10:44 spring boot 后端 java

Spring Boot中实现多数据源连接和切换的方案

在Spring Boot项目中，随着业务需求的增长，我们往往需要连接多个数据库，即实现多数据源连接和切换。这种需求可能源于数据库的读写分离、微服务架构下的服务拆分、数据分库分表等场景。本文将详细探讨Spring Boot中实现多数据源连接和切换的多种方案，并给出相应的实现步骤和代码示例。

一、概述

在Spring Boot中，实现多数据源连接和切换的方案有多种，主要包括：

使用AbstractRoutingDataSource实现动态数据源切换
使用MP提供的Dynamic-datasource多数据源框架
通过自定义注解在方法或类上指定数据源
使用数据库代理中间件

每种方案都有其独特的优势和适用场景，下面将分别进行详细阐述。

二、使用AbstractRoutingDataSource实现动态数据源切换

AbstractRoutingDataSource是Spring框架提供的一个抽象类，用于根据用户定义的规则选择当前的数据源。通过继承这个类，并实现其抽象方法determineCurrentLookupKey，我们可以实现动态数据源切换。

步骤：

定义数据源枚举：用于标识不同的数据源。

public enum DataSourceType {
    MASTER("master"),
    SLAVE("slave");

    private String value;

    DataSourceType(String value) {
        this.value = value;
    }

    public String getValue() {
        return value;
    }
}

创建DynamicDataSource类：继承AbstractRoutingDataSource，并重写determineCurrentLookupKey方法。

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

    public DynamicDataSource(DataSource defaultTargetDataSource, Map<Object, Object> targetDataSources) {
        super.setDefaultTargetDataSource(defaultTargetDataSource);
        super.setTargetDataSources(targetDataSources);
        super.afterPropertiesSet();
    }

    @Override
    protected Object determineCurrentLookupKey() {
        return CONTEXT_HOLDER.get();
    }

    public static void setDataSourceType(String dataSource) {
        CONTEXT_HOLDER.set(dataSource);
    }

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

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

配置数据源：在Spring Boot的配置文件中配置多个数据源，并在配置类中创建这些数据源，将它们注入到DynamicDataSource中。

@Configuration
public class DataSourceConfig {

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

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

    @Bean
    public DataSource dataSource() {
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put(DataSourceType.MASTER.getValue(), masterDataSource());
        targetDataSources.put(DataSourceType.SLAVE.getValue(), slaveDataSource());

        DynamicDataSource dataSource = new DynamicDataSource(masterDataSource(), targetDataSources);
        return dataSource;
    }
}

使用AOP切换数据源：通过自定义注解和AOP，在方法执行前后切换数据源。

@Aspect
@Component
public class DataSourceAspect {

    @Before("@annotation(dataSource)")
    public void changeDataSource(JoinPoint point, DataSource dataSource) {
        DynamicDataSource.setDataSourceType(dataSource.value());
    }

    @After("@annotation(dataSource)")
    public void clearDataSource(JoinPoint point, DataSource dataSource) {
        DynamicDataSource.clearDataSourceType();
    }
}

自定义注解：

@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.METHOD, ElementType.TYPE})
public @interface DataSource {
    DataSourceType value() default DataSourceType.MASTER;
}

在业务方法上使用注解：

@Service
public class MyService {

    @DataSource(DataSourceType.SLAVE)
    public void someMethod() {
        // 业务逻辑
    }
}

三、使用MP提供的Dynamic-datasource多数据源框架

MP（MyBatis-Plus）提供了Dynamic-datasource多数据源框架，简化了多数据源的配置和管理。

步骤：

引入依赖：在pom.xml文件中添加Dynamic-datasource的依赖。

<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
    <version>最新版本</version>
</dependency>

配置数据源：在application.yml或application.properties文件中配置多个数据源。

spring:
  datasource:
    dynamic:
      primary: master # 设置默认的数据源或者数据源组,默认值即为master
      datasource:
        master:
          url: jdbc:mysql://localhost:3306/master_db
          username: root
          password: root
          driver-class-name: com.mysql.cj.jdbc.Driver
        slave:
          url: jdbc:mysql://localhost:3306/slave_db
          username: root
          password: root
          driver-class-name: com.mysql.cj.jdbc.Driver

使用注解或配置类指定数据源：在业务方法或配置类上使用@DS注解指定数据源。

@Service
public class MyService {

    @DS("slave")
    public void someMethod() {
        // 业务逻辑
    }
}

四、通过自定义注解在方法或类上指定数据源

这种方案与使用AbstractRoutingDataSource实现动态数据源切换类似，但更加灵活和直观。通过自定义注解，我们可以直接在方法或类上指定使用哪个数据源。

步骤：

定义数据源枚举和自定义注解（同上）。
创建数据源上下文类（同上）。
配置数据源（同上）。
使用AOP切换数据源（同上）。
在业务方法上使用注解（同上）。

五、使用数据库代理中间件

数据库代理中间件，如Mycat、Sharding-JDBC等，提供了更加灵活和强大的多数据源管理和分片功能。这些中间件通常支持读写分离、分库分表、数据聚合等功能，可以大大简化多数据源的配置和管理。

步骤：

选择并引入中间件：根据业务需求选择合适的数据库代理中间件，并在项目中引入相应的依赖。
配置中间件：在配置文件中配置中间件的相关参数，如数据源、分片规则等。
使用中间件：通过中间件提供的接口或配置，实现多数据源连接和切换。

六、总结

在Spring Boot中实现多数据源连接和切换的方案有多种，每种方案都有其独特的优势和适用场景。使用AbstractRoutingDataSource实现动态数据源切换具有较高的灵活性和可控性；使用MP提供的Dynamic-datasource多数据源框架可以简化配置和管理；通过自定义注解在方法或类上指定数据源则更加直观和易用；使用数据库代理中间件则提供了更加灵活和强大的多数据源管理和分片功能。在实际应用中，我们可以根据业务需求和技术栈选择合适的方案，以实现多数据源连接和切换。

原文地址：https://blog.csdn.net/shiming8879/article/details/143563083

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！

上一篇：KTHREAD结构--＞ApcState
下一篇：软件要求英文目录，但是我喜欢中文目录

【学习日记】notebook添加JAVA支持
作者是个大学生这个专栏主要收集课时常用的软件以及女朋友上课用的软件的教程。需提前配置好java环境本篇仅对添加支持进行说明。新开了gitcode 用于上传安装包。解压进入解压后目录复制文件地
阅读更多2024-11-15
Docker与Podman全面比较
Docker和Podman作为两大容器引擎，各自拥有独特的特点和优势。本文将从溯源、特点、技术优势、应用实例和技术前景等方面对Docker和Podman进行全面比较。
阅读更多2024-11-15
算法学习blog：day2 继续记日记
4. 明日计划：至少五道题，并且要学会并实现今天的三道题，看这五道题的思路解法，下一天进行实现优化。除此之外pdf粗略看到了20页，明天继续看，后面才是重点。1.做了三道PAT 76，77，78，差一
阅读更多2024-11-15
基于Python的网上银行综合管理系统
【2025最新】基于python+django+vue+MySQL的网上银行综合管理系统，前后端分离。
阅读更多2024-11-15
自定义注解+拦截器+jwtFilter实现权限控制
GetterSUPER_ADMIN(1, "超级管理员"),SYSTEM_ADMIN(2, "系统管理员"),DOMESTIC_CONSUMER(3, &quo
阅读更多2024-11-15
前端面试题整理-vue指令开发
在 bind 钩子中，我为绑定的元素添加了一个点击事件监听器，当元素被点击时，执行复制操作。我当时在开发点击复制文本的功能，我有很多个元素都想有这个功能，但是我又不想每个元素都绑定一个 onClick
阅读更多2024-11-15
在使用ipc通信时，在渲染进程的Vue + TypeScript 开发过程，给window对象添加属性并赋值时，发生报错解决方法
在使用ipc通信时，在渲染进程的Vue + TypeScript 开发过程，给window对象添加属性并赋值时，发生报错解决方法
阅读更多2024-11-15
GESP4级考试语法知识（贪心算法（四））
GESP4级考试语法知识（贪心算法（四））
阅读更多2024-11-15
20241114在飞凌的OK3588-C的核心板上跑Linux R4时通过iperf3测试以太网卡的实际网速
创建一个eth0配置文件，配置文件的路径为：/etc/network/interfaces.d/eth0,设置动态ip的配置文件。虽然飞凌的OK3588-C的核心板使用的是千兆网卡RTL8211
阅读更多2024-11-15
【EmbeddedGUI】脏矩阵设计说明
脏矩阵设计说明
阅读更多2024-11-15

Spring Boot中实现多数据源连接和切换的方案

Spring Boot中实现多数据源连接和切换的方案

一、概述

二、使用AbstractRoutingDataSource实现动态数据源切换

三、使用MP提供的Dynamic-datasource多数据源框架

四、通过自定义注解在方法或类上指定数据源

五、使用数据库代理中间件

六、总结

相关文章