java使用dynamic-datasource实现动态数据源切换

在Java应用程序中，实现动态数据源切换通常是为了支持多租户、读写分离或数据库分片等场景。dynamic-datasource-spring-boot-starter 是一个流行的Spring Boot Starter，它可以帮助你轻松地在Spring Boot应用程序中实现动态数据源切换。

以下是如何使用 dynamic-datasource-spring-boot-starter 实现动态数据源切换的步骤：

》1、添加依赖：
首先，你需要在你的Spring Boot项目的 pom.xml 文件中添加 dynamic-datasource-spring-boot-starter 的依赖。

<dependency>  
    <groupId>com.baomidou</groupId>  
    <artifactId>dynamic-datasource-spring-boot-starter</artifactId>  
    <version>你的版本号</version>  
</dependency>

        请确保你使用的是与你的Spring Boot版本兼容的 dynamic-datasource-spring-boot-starter 版本。

》2、配置数据源：
在 application.yml 或 application.properties 文件中配置你的数据源。你可以配置多个数据源，并使用 spring.datasource.dynamic 前缀来指定它们。

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_1:  
          url: jdbc:mysql://localhost:3306/slave_db_1  
          username: root  
          password: root  
          driver-class-name: com.mysql.cj.jdbc.Driver  
        slave_2:  
          url: jdbc:mysql://localhost:3306/slave_db_2  
          username: root  
          password: root  
          driver-class-name: com.mysql.cj.jdbc.Driver

》3、配置数据源策略（可选）：
你可以根据需要配置数据源切换的策略，比如基于线程、事务或AOP的切换策略。默认情况下，dynamic-datasource-spring-boot-starter 提供了基于线程的数据源切换策略。

》4、使用数据源：
在你的代码中，你可以通过 @DS 注解来指定要使用的数据源。这个注解可以放在方法上或类上。

@Service  
public class UserService {  

    @DS("slave_1")  
    public User getUserByIdFromSlave1(Long id) {  
        // 从slave_1数据源查询用户  
    }  

    @DS("slave_2")  
    public User getUserByIdFromSlave2(Long id) {  
        // 从slave_2数据源查询用户  
    }  

    @DS("master")  
    public void createUser(User user) {  
        // 在master数据源上创建用户  
    }  
}

• 如果你没有指定 @DS 注解，那么将使用默认的数据源（在上面的配置中是 master）。

• 通过 AOP 动态切换数据源：

@RestController
@RequestMapping("/api")
public class SampleController {
 
    @Autowired
    private DataSourceRoutingDataSource routingDataSource;
 
    @GetMapping("/data")
    public String getData(@RequestHeader("DataSource") String dataSourceHeader) {
        // 通过请求头来动态切换数据源
        Map<Object, Object> dataSourceMap = routingDataSource.getDataSources();
        if (dataSourceMap.containsKey(dataSourceHeader)) {
            routingDataSource.setCurrentDataSource(dataSourceHeader);
        }
        // 执行数据库操作...
        return "Data from " + routingDataSource.getCurrentDataSource();
    }
}

》5、测试数据源切换：
编写单元测试或集成测试来验证数据源切换是否按预期工作。你可以通过日志或数据库查询日志来检查实际使用的数据源。

》6、处理事务：
当使用动态数据源时，需要特别注意事务的管理。确保在同一个事务中使用的所有操作都指向同一个数据源，否则可能会导致事务不一致的问题。

》7、自定义数据源配置（可选）：
如果你需要更复杂的配置，比如自定义数据源初始化、连接池配置等，你可以通过实现 DataSource 和 DynamicRoutingDataSource 的相关接口来自定义数据源的行为。

请注意，dynamic-datasource-spring-boot-starter 是一个第三方库，其API和配置可能会随着版本的更新而发生变化。因此，建议查阅该库的官方文档以获取最新的使用指南和最佳实践。