SpringBoot实现动态切换数据源（含源码）

`ThreadLocal` 是 Java 中的一个类，用于存储线程局部变量。线程局部变量与普通的变量不同，它不是共享的，每个线程都有其自己的独立的线程局部变量副本。这使得我们可以在多线程环境中为每个线程提供独立的变量副本，从而实现线程间的数据隔离。 在数据源切换的场景中，我们通常将数据源信息存储在 `ThreadLocal` 中，然后在数据访问层（如 DAO）中通过 `ThreadLocal` 来获取当前线程的数据源信息，从而动态地切换数据源。 `AbstractRoutingDataSource` 是 MyBatis-Plus 提供的一个数据源路由类。它可以基于某个条件来动态地切换数据源。当调用 `selectAnyDataSources()` 方法时，它会根据当前线程的上下文信息来返回一个数据源。如果没有设置上下文信息，它会返回默认的数据源。 结合 `ThreadLocal` 和 `AbstractRoutingDataSource`，我们可以轻松地在运行时动态地切换数据源。 在Spring Boot应用中，动态切换数据源是一项关键功能，尤其在多数据库环境或者读写分离的架构中。这里我们将深入探讨如何使用`ThreadLocal`和`AbstractRoutingDataSource`来实现这一目标。 `ThreadLocal`是Java提供的一个特殊类型的变量，它确保了每个线程都拥有该变量的独立副本。这意味着在多线程环境下，每个线程可以独立地修改自己的`ThreadLocal`变量，而不会影响其他线程。在数据源切换场景中，`ThreadLocal`用于存储当前线程所要使用的数据源信息。例如，在业务逻辑开始时，我们可以根据业务需求设置`ThreadLocal`中的数据源，比如设置为"slave"，表示当前线程将使用从库进行数据操作。这样，在DAO层中，通过`ThreadLocal`获取到的数据源信息，就能确保操作发生在正确的目标数据库上。 `AbstractRoutingDataSource`是MyBatis-Plus提供的一种抽象数据源路由类，它可以根据特定的条件动态选择数据源。在实现时，我们需要继承`AbstractRoutingDataSource`并覆盖`determineCurrentLookupKey`方法。这个方法会在执行SQL前被调用，以确定当前应该使用哪个数据源。通常，我们会在这个方法中获取`ThreadLocal`中的数据源信息，并作为返回值，这样`AbstractRoutingDataSource`就会根据这个信息选择合适的数据源。 下面是一个简单的代码实现示例： ```java // DataSourceContextHolder用于存储线程数据源信息 public class DataSourceContextHolder { private static final ThreadLocal<String> DATASOURCE_HOLDER = new ThreadLocal<>(); public static void setDataSource(String dataSourceName) { DATASOURCE_HOLDER.set(dataSourceName); } public static String getDataSource() { return DATASOURCE_HOLDER.get(); } public static void removeDataSource() { DATASOURCE_HOLDER.remove(); } } // DynamicDataSource继承自AbstractRoutingDataSource，根据ThreadLocal中的数据源信息动态切换 public class DynamicDataSource extends AbstractRoutingDataSource { @Override protected Object determineCurrentLookupKey() { return DataSourceContextHolder.getDataSource(); } // 其他初始化和配置代码... } ``` 在实际应用中，我们需要在Spring Boot的配置文件(`application.yml`或`application.properties`)中定义多个数据源，然后在`@Configuration`类中配置`DynamicDataSource`，使其能够根据业务逻辑动态选择数据源。同时，为了保证数据源切换的正确性和线程安全，业务代码在执行完特定逻辑后，务必记得清除`ThreadLocal`中的数据源信息，以免影响其他线程。 总结起来，使用`ThreadLocal`和`AbstractRoutingDataSource`实现动态数据源切换的关键步骤包括： 1. 创建`ThreadLocal`实例来保存线程的数据源信息。 2. 实现`AbstractRoutingDataSource`，并重写`determineCurrentLookupKey`方法，使其根据`ThreadLocal`获取数据源。 3. 在业务代码中设置和清除`ThreadLocal`中的数据源信息。 4. 配置Spring Boot的数据源和`DynamicDataSource`。 通过这种方式，我们可以在运行时根据业务需求灵活地切换数据源，实现读写分离、多数据库管理等高级功能，提高系统的可扩展性和灵活性。