详细聊聊SpringBoot中动态切换数据源的方法

下面是详细聊聊SpringBoot中动态切换数据源的方法的完整攻略，主要分为以下几个部分：

1. 前置条件

在使用SpringBoot动态切换数据源之前，我们需要准备以下环境和工具：

• SpringBoot框架（推荐使用最新版）
• 数据源配置文件（指定一个或多个数据库配置，其中至少要包含一个默认数据源配置）
• 动态数据源切换工具类（可自己实现或者选择第三方库）

接下来，将逐一进行讲解。

2. 数据源配置

在SpringBoot中，我们可以通过在application.properties或者application.yml中配置数据源信息。例如，在application.yml中，我们可以使用如下的方式指定数据库连接信息：

datasource:
  master:
    url: jdbc:mysql://localhost:3306/db_master?useUnicode=true&characterEncoding=utf8&serverTimezone=Asia/Shanghai
    username: root
    password: root
    driver-class-name: com.mysql.cj.jdbc.Driver
  slave:
    url: jdbc:mysql://localhost:3307/db_slave?useUnicode=true&characterEncoding=utf8&serverTimezone=Asia/Shanghai
    username: root
    password: root
    driver-class-name: com.mysql.cj.jdbc.Driver

配置中，我们指定了两个数据源，一个是master，一个是slave，它们的配置信息分别包括：url、username、password和driver-class-name。其中，driver-class-name是数据库驱动的类名，SpringBoot会根据这个信息来初始化数据源。

需要注意的是，我们必须要给一个数据源配置一个名字，这里我们给master和slave都命名了。默认情况下，SpringBoot会使用名字为"default"的数据源作为默认数据源。如果需要在多个数据源中切换，我们需要使用一个特殊的工具类来帮助我们实现动态数据源切换。

3. 动态数据源切换工具类

在SpringBoot中，有多种实现动态数据源切换的工具类，比如：AbstractRoutingDataSource、DynamicRoutingDataSource等。这里，我们以 DynamicRoutingDataSource 为例，来介绍如何实现动态数据源切换。

DynamicRoutingDataSource 是一个继承了 AbstractRoutingDataSource 的数据源切换类，我们需要在其子类中，重写 determineCurrentLookupKey 方法，来实现数据源的动态切换。例如：

public class DynamicDataSource extends AbstractRoutingDataSource {

    /**
     * 获取数据源
     * @return 获取当前数据源
     */
    @Override
    protected Object determineCurrentLookupKey() {
        return DataSourceContextHolder.getDataSource();
    }

}

在上面的代码中，我们通过获取上下文中当前的数据源，来确认应该使用哪个数据源。其中 DataSourceContextHolder 是一个线程局部变量，用于维护当前线程使用的数据源。下面是具体实现：

public class DataSourceContextHolder {

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

    public static void setDataSource(String dataSource) {
        contextHolder.set(dataSource);
    }

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

    public static void clearDataSource() {
        contextHolder.remove();
    }

}

在上面的代码中，我们使用 ThreadLocal 来存储当前线程使用的数据源 key 值。这样，我们就可以轻松实现动态数据源的切换了。

4. 实现动态数据源切换

在上面的步骤中，我们已经准备好了数据源配置和动态数据源切换的工具类，接下来，我们需要在 SpringBoot 中实现动态数据源的切换。

首先，我们需要在 SpringBoot 的配置文件中，配置数据源和动态数据源的信息，例如在 application.yml 文件中：

# 主数据源
datasource:
  master:
    url: jdbc:mysql://127.0.0.1:3306/mydb?useUnicode=true&characterEncoding=utf8&useSSL=false&serverTimezone=GMT%2B8
    username: root
    password: root
    driver-class-name: com.mysql.cj.jdbc.Driver
# 从数据源
  slave:
    url: jdbc:mysql://127.0.0.1:3307/mydb?useUnicode=true&characterEncoding=utf8&useSSL=false&serverTimezone=GMT%2B8
    username: root
    password: root
    driver-class-name: com.mysql.cj.jdbc.Driver

# 动态数据源
custom:
  dynamic-datasource:
    master: master
    slave: slave

其中，custom.dynamic-datasource 是我们自定义的配置，这里我们将master和slave数据源的名称分别指定为 master 和 slave。接下来，在 SpringBoot 应用的启动类中，我们需要获取 custom.dynamic-datasource 的配置信息，来初始化动态数据源切换的工具类，例如：

@SpringBootApplication
public class Application {

    public static void main(String[] args) {
        ConfigurableApplicationContext context = SpringApplication.run(Application.class, args);

        // 初始化动态数据源切换工具类
        DynamicDataSource dynamicDataSource = context.getBean(DynamicDataSource.class);
        Map<Object, Object> dataSourceMap = new HashMap<>();
        dataSourceMap.put("master", context.getBean("master"));
        dataSourceMap.put("slave", context.getBean("slave"));
        dynamicDataSource.setTargetDataSources(dataSourceMap);
        dynamicDataSource.setDefaultTargetDataSource(context.getBean("master"));
    }

}

在上面的代码中，我们使用 SpringBoot 应用上下文（ApplicationContext）来获取数据源和动态数据源切换工具类的对象，并初始化 DynamicDataSource 对象的数据源信息。其中，dataSourceMap 包含了我们配置的 master 和 slave 数据源信息，defaultTargetDataSource 指定了默认的数据源，这个可以按需设置。

最后，我们需要实现动态数据源的切换。在需要使用不同数据源的地方，我们可以通过以下方式进行动态切换：

@RestController
public class UserController {

    @Autowired
    private UserService userService;

    @GetMapping("/users")
    public List<User> getUsers(@RequestParam String dataSourceName) {
        DataSourceContextHolder.setDataSource(dataSourceName);
        List<User> users = userService.listUsers();
        DataSourceContextHolder.clearDataSource();
        return users;
    }

}

在上面的代码中，我们通过修改 DataSourceContextHolder 的数据源 key 的值，来切换数据源。这里特别说明一下，最好使用 try...finally... 结构来设置和清除数据源，以防止出现数据源切换失败的情况。

到这里，我们已经完成了动态数据源切换的工作。当需要切换数据源时，只需要调用 setDataSource 方法即可，相比于传统的配置文件切换方式，这种方式更加灵活和简便。