Spring 多数据源配置：在同一个方法中调用不同数据源的服务

Spring 多数据源配置：在同一个方法中调用不同数据源的服务

本文将介绍如何在同一个方法中调用使用不同数据源的服务，并使用 Spring 的事务管理来处理不同数据源的交互。

场景： 假设我们有一个方法，需要操作两个不同的数据库。例如，服务 A 需要操作数据库 A，服务 B 需要操作数据库 B。

解决方案： 使用 Spring 的多数据源配置，并使用事务管理来协调不同数据源之间的操作。

步骤：

1. 配置数据源:

   在配置文件 (如 application.yml 或 application.properties) 中配置数据源和事务管理器。假设服务 A 使用数据源 A，服务 B 使用数据源 B，配置如下：

   spring:
     datasource:
       datasourceA:
         url: jdbc:mysql://localhost:3306/databaseA
         username: usernameA
         password: passwordA
       datasourceB:
         url: jdbc:mysql://localhost:3306/databaseB
         username: usernameB
         password: passwordB
   jpa:
     database-platform: org.hibernate.dialect.MySQL5Dialect
   

2. 创建数据源配置类:

   创建两个数据源的配置类，分别用于创建 datasourceA 和 datasourceB：

   import org.springframework.beans.factory.annotation.Autowired;
   import org.springframework.beans.factory.annotation.Qualifier;
   import org.springframework.boot.autoconfigure.jdbc.DataSourceBuilder;
   import org.springframework.boot.context.properties.ConfigurationProperties;
   import org.springframework.context.annotation.Bean;
   import org.springframework.context.annotation.Configuration;
   import org.springframework.context.annotation.Primary;
   
   import javax.sql.DataSource;
   
   @Configuration
   public class DataSourceConfig {
       @Bean(name = 'datasourceA')
       @Primary
       @ConfigurationProperties(prefix = 'spring.datasource.datasourceA')
       public DataSource datasourceA() {
           return DataSourceBuilder.create().build();
       }
   
       @Bean(name = 'datasourceB')
       @ConfigurationProperties(prefix = 'spring.datasource.datasourceB')
       public DataSource datasourceB() {
           return DataSourceBuilder.create().build();
       }
   }
   

3. 指定数据源:

   在服务 A 和服务 B 的类上分别使用 @Primary 和 @Qualifier 注解来指定使用的数据源：

   import org.springframework.beans.factory.annotation.Autowired;
   import org.springframework.beans.factory.annotation.Qualifier;
   import org.springframework.stereotype.Service;
   import org.springframework.transaction.annotation.Transactional;
   
   @Service
   public class ServiceA {
       private final RepositoryA repositoryA;
   
       @Autowired
       public ServiceA(@Qualifier('datasourceA') DataSource dataSourceA) {
           this.repositoryA = new RepositoryA(dataSourceA);
       }
   
       @Transactional(transactionManager = 'transactionManagerA')
       public void test1() {
           // 使用 datasourceA 进行操作
           repositoryA.test1();
       }
   }
   
   @Service
   public class ServiceB {
       private final RepositoryB repositoryB;
   
       @Autowired
       public ServiceB(@Qualifier('datasourceB') DataSource dataSourceB) {
           this.repositoryB = new RepositoryB(dataSourceB);
       }
   
       @Transactional(transactionManager = 'transactionManagerB')
       public void test2() {
           // 使用 datasourceB 进行操作
           repositoryB.test2();
       }
   }
   

   在上述代码中，RepositoryA 和 RepositoryB 分别使用不同的数据源进行操作。在 ServiceA 和 ServiceB 中，使用 @Autowired 注解来注入所需的数据源，并使用 @Transactional 注解来开启事务。

4. 调用服务:

   在测试方法中调用 serviceA.test1() 和 serviceB.test2() 即可：

   public void test() {
       serviceA.test1();
       serviceB.test2();
   }
   

注意：

• 以上代码仅供参考，实际应用中可能需要根据具体情况进行调整。
• 确保每个数据源都有对应的 transactionManager，并且配置正确。
• 在同一个方法中调用不同数据源的服务时，需要注意事务管理，确保操作的原子性和一致性。
• 可以使用 @Transactional 注解来指定事务管理器，或者在配置文件中配置默认的事务管理器。
• 对于复杂的场景，可以使用 Spring 的 TransactionDefinition 接口来定制事务行为。

总结：

Spring 多数据源配置和事务管理是实现多数据源操作的常用方法，可以方便地管理不同数据源之间的数据交互。在实际应用中，需要注意配置的正确性和事务管理的合理使用，以确保操作的正确性和一致性。