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

前言

在 spring boot 中实现多数据源连接和切换，可以通过以下几种方案来实现，具体取决于项目的需求、数据库的使用模式和管理的复杂性。以下是一个常见的多数据源切换的实现方案，使用 abstractroutingdatasource 来动态选择数据源。

一、多数据源配置与切换方案

在多数据源场景中，通常有如下步骤：

配置多个数据源的 datasource bean。
使用 abstractroutingdatasource 来动态切换数据源。
使用 threadlocal 存储当前的数据库类型或数据源标识符。
配置数据源切换的逻辑，例如基于当前的用户、请求路径、服务标识等来选择不同的数据源。

二、实现步骤

1. 创建多个 datasource 配置类

首先，为每个数据源创建单独的配置类，通常你会在 application.yml 或 application.properties 中配置每个数据源的连接信息。

spring:
  datasource:
    # 默认数据源配置
    primary:
      url: jdbc:mysql://localhost:3306/primary_db
      username: root
      password: password
      driver-class-name: com.mysql.cj.jdbc.driver
      hikari:
        maximum-pool-size: 10
    # 第二数据源配置
    secondary:
      url: jdbc:mysql://localhost:3306/secondary_db
      username: root
      password: password
      driver-class-name: com.mysql.cj.jdbc.driver
      hikari:
        maximum-pool-size: 10

2. 创建 datasource 配置类

@configuration
@enabletransactionmanagement
public class datasourceconfig {
    @bean(name = "primarydatasource")
    @primary
    @configurationproperties(prefix = "spring.datasource.primary")
    public datasource primarydatasource() {
        return datasourcebuilder.create().build();
    }
    @bean(name = "secondarydatasource")
    @configurationproperties(prefix = "spring.datasource.secondary")
    public datasource secondarydatasource() {
        return datasourcebuilder.create().build();
    }
}

3. 创建动态数据源路由类

abstractroutingdatasource 允许我们在运行时根据某些条件动态选择数据源。

@configuration
public class dynamicdatasourceconfig {
    @autowired
    @qualifier("primarydatasource")
    private datasource primarydatasource;
    @autowired
    @qualifier("secondarydatasource")
    private datasource secondarydatasource;
    @bean
    public datasource datasource() {
        // 创建一个路由数据源
        dynamicdatasource datasource = new dynamicdatasource();
        datasource.setdefaulttargetdatasource(primarydatasource); // 默认数据源
        map<object, object> targetdatasources = new hashmap<>();
        targetdatasources.put("primary", primarydatasource);
        targetdatasources.put("secondary", secondarydatasource);
        datasource.settargetdatasources(targetdatasources);
        return datasource;
    }
}

4. 实现 dynamicdatasource 类

dynamicdatasource 是继承自 abstractroutingdatasource，它通过 determinecurrentlookupkey() 方法来动态确定当前的数据源。

public class dynamicdatasource extends abstractroutingdatasource {
    // 从 threadlocal 获取当前的数据库标识
    @override
    protected object determinecurrentlookupkey() {
        return datasourcecontextholder.getdatasourcetype();
    }
}

5. 创建 datasourcecontextholder 来存储当前的数据源标识

使用 threadlocal 来保持当前线程的数据库标识，以便在不同的数据源之间切换。

public class datasourcecontextholder {
    // 使用 threadlocal 存储当前线程的数据源标识
    private static final threadlocal<string> contextholder = new threadlocal<>();
    public static void setdatasourcetype(string datasourcetype) {
        contextholder.set(datasourcetype);
    }
    public static string getdatasourcetype() {
        return contextholder.get();
    }
    public static void cleardatasourcetype() {
        contextholder.remove();
    }
}

6. aop 方式切换数据源

为了在运行时动态切换数据源，通常会使用 aop 切面来拦截方法执行并指定数据源。

@aspect
@component
public class datasourceaspect {
    // 通过注解来指定使用哪个数据源
    @before("@annotation(datasource)")
    public void switchdatasource(datasourcetype datasource) {
        // 切换数据源
        datasourcecontextholder.setdatasourcetype(datasource.value());
    }
    @after("@annotation(datasource)")
    public void cleardatasource(datasourcetype datasource) {
        // 清理数据源标识，避免影响其他线程
        datasourcecontextholder.cleardatasourcetype();
    }
}

7. 自定义注解来指定数据源

创建一个自定义注解 datasourcetype，用于指定当前方法执行时需要使用的数据源。

@target(elementtype.method)
@retention(retentionpolicy.runtime)
public @interface datasourcetype {
    string value() default "primary";  // 数据源标识，默认使用primary数据源
}

8. 在 service 层使用注解指定数据源

在 service 层，可以使用刚才创建的 @datasourcetype 注解来指定不同的方法使用不同的数据源。

@service
public class userservice {
    @datasourcetype("primary")
    public list<user> getprimaryusers() {
        // 查询主数据库
        return userrepository.findall();
    }
    @datasourcetype("secondary")
    public list<user> getsecondaryusers() {
        // 查询次数据库
        return secondaryuserrepository.findall();
    }
}