java切换多个数据源

原创

mob649e8163af7d 2024-10-20 05:53:16 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e8163af7d的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java切换多个数据源的实现

在现代软件开发中，使用多个数据库源是一种常见需求，尤其是在微服务架构中。Java提供了灵活的方式来实现这一点，本文将教你如何实现 Java 切换多个数据源的功能。

实现流程

首先，我们需要明确处理多个数据源的整体流程，下面的表格展示了实现的步骤。

步骤	描述
1	创建数据源配置类
2	创建数据源选择器
3	配置 Spring Boot 的数据源
4	使用自定义注解切换数据源
5	测试切换功能

步骤详解

步骤1: 创建数据源配置类

首先，我们需要为每个数据源创建一个数据源配置类。以下是一个示例：

@Configuration
public class DataSourceConfig {

    @Bean(name = "dataSource1")
    @Primary // 设定为主要数据源
    @ConfigurationProperties(prefix = "spring.datasource.datasource1")
    public DataSource dataSource1() {
        return DataSourceBuilder.create().build();
    }

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

    // 在这里可以继续添加更多数据源
}

这段代码用于配置两个数据源，分别命名为 dataSource1 和 dataSource2。

步骤2: 创建数据源选择器

接下来，我们需要创建一个数据源选择器，用于在运行时选择数据源：

@Component
public class DataSourceContextHolder {

    private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>();

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

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

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

这段代码通过 ThreadLocal 维护每个线程独立的数据源类型。

步骤3: 配置Spring Boot的多数据源

在 Spring Boot 中，我们可以通过配置文件来管理多个数据源。以下是 application.yml 的示例：

spring:
  datasource:
    datasource1:
      url: jdbc:mysql://localhost:3306/db1
      username: root
      password: password1
      driver-class-name: com.mysql.cj.jdbc.Driver
    datasource2:
      url: jdbc:mysql://localhost:3307/db2
      username: root
      password: password2
      driver-class-name: com.mysql.cj.jdbc.Driver

这段代码配置信息用于连接不同的数据库。

步骤4: 使用自定义注解切换数据源

我们需要创建一个注解来动态选择数据源：

@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface SwitchDataSource {
    String value();
}

然后，编写切面类来处理数据源切换逻辑：

@Aspect
@Component
public class DataSourceSwitchAspect {

    @Around("@annotation(switchDataSource)")
    public Object around(ProceedingJoinPoint point, SwitchDataSource switchDataSource) throws Throwable {
        try {
            DataSourceContextHolder.setDataSourceType(switchDataSource.value());
            return point.proceed();
        } finally {
            DataSourceContextHolder.clearDataSourceType(); // 清理上下文
        }
    }
}

步骤5: 测试切换功能

最后，我们可以编写一个简单的服务来测试数据源切换：

@Service
public class UserService {

    @SwitchDataSource("dataSource1")
    public List<User> getUsersFromDataSource1() {
        // 从 dataSource1 获取用户数据的逻辑
    }

    @SwitchDataSource("dataSource2")
    public List<User> getUsersFromDataSource2() {
        // 从 dataSource2 获取用户数据的逻辑
    }
}

饼状图展示

使用 Mermaid 语法生成饼状图：

pie
    title 数据源使用情况
    "数据源1": 50
    "数据源2": 50

序列图展示

使用 Mermaid 语法生成序列图：

sequenceDiagram
    participant UserService
    participant DataSourceSwitchAspect
    participant DataSourceContextHolder

    UserService->>DataSourceSwitchAspect: 调用方法
    DataSourceSwitchAspect->>DataSourceContextHolder: 设置数据源
    DataSourceContextHolder-->>DataSourceSwitchAspect: 返回
    DataSourceSwitchAspect->>UserService: 继续执行方法
    DataSourceSwitchAspect->>DataSourceContextHolder: 清理数据源