如何实现Spring中的多数据源配置？

在Spring中配置多个数据源主要分为以下几个步骤：

步骤1: 添加数据源依赖

确保你的项目中包含了所需数据库的依赖。通常在pom.xml中添加Spring Boot提供的starter依赖就可以自动包含数据库驱动和Spring的数据源支持。

步骤2: 配置数据源属性

在application.properties或application.yml中配置多个数据源属性，包括JDBC URL、用户和密码等。每个数据源都应该有不同的配置前缀。

application.yml示例：

app:
  datasource:
    primary:
      jdbc-url: jdbc:mysql://localhost:3306/db_primary
      username: primary
      password: primarypass
      driver-class-name: com.mysql.cj.jdbc.Driver
    secondary:
      jdbc-url: jdbc:mysql://localhost:3306/db_secondary
      username: secondary
      password: secondarypass
      driver-class-name: com.mysql.cj.jdbc.Driver

步骤3: 创建数据源配置类

创建两个或更多的配置类，每个配置类对应一个数据源。可以使用@Configuration注解标注配置类，并分别使用@Bean注解定义DataSource、EntityManagerFactory和TransactionManager的bean。你还可以使用@Primary注解标注主数据源。

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(
  basePackages = "com.example.primary.repos",
  entityManagerFactoryRef = "primaryEntityManagerFactory",
  transactionManagerRef = "primaryTransactionManager"
)
public class PrimaryDataSourceConfig {

    @Bean
    @Primary
    @ConfigurationProperties("app.datasource.primary")
    public DataSourceProperties primaryDataSourceProperties() {
        return new DataSourceProperties();
    }

    @Bean
    @Primary
    public DataSource primaryDataSource() {
        return primaryDataSourceProperties().initializeDataSourceBuilder().build();
    }

    // Define the EntityManagerFactory bean with primary data source
    // Define the TransactionManager bean with primary data source
}

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(
  basePackages = "com.example.secondary.repos",
  entityManagerFactoryRef = "secondaryEntityManagerFactory",
  transactionManagerRef = "secondaryTransactionManager"
)
public class SecondaryDataSourceConfig {

    @Bean
    @ConfigurationProperties("app.datasource.secondary")
    public DataSourceProperties secondaryDataSourceProperties() {
        return new DataSourceProperties();
    }

    @Bean
    public DataSource secondaryDataSource() {
        return secondaryDataSourceProperties().initializeDataSourceBuilder().build();
    }

    // Define the EntityManagerFactory bean with secondary data source
    // Define the TransactionManager bean with secondary data source
}

你需要为每个数据源分别配置它们的事务管理器和实体管理工厂，确保它们操作不同的数据库。

步骤4: 使用数据源

在你的Repository类中，使用@Qualifier注解指定使用哪个数据源。

@Repository
public class SomeRepository {

    private final JdbcTemplate jdbcTemplate;

    @Autowired
    public SomeRepository(@Qualifier("secondaryDataSource") DataSource dataSource) {
        this.jdbcTemplate = new JdbcTemplate(dataSource);
    }

    // ...使用jdbcTemplate进行操作...
}

对于JPA存储库，Spring Data会自动使用@EnableJpaRepositories注解中指定的entityManagerFactoryRef和transactionManagerRef来正确配置存储库。

步骤5: 操作多数据源

现在可以在应用中使用定义的数据源了。Spring 会根据你在代码中的配置选择合适的数据源进行操作。你可以在事务方法中使用@Transactional注解，并通过transactionManager属性来指定事务管理器。

总结

配置多数据源让你的应用程序可以灵活地处理多个数据库。不过，这需要你手动配置和管理多个事务管理器和实体管理工厂。在服务层中操作时，需要清楚地了解哪个服务方法应该使用哪个数据源。通过Spring的配置和抽象，你可以同时使用JPA、JDBC模板或其他数据访问技术管理不同的数据源。