使用Spring Boot简单实现多数据源配置与整合

2023-06-05 02:23:05

Spring Boot 中的多数据源配置和整合

概述

在实际项目中，操作多个数据源的情况并不少见。Spring Boot 提供了简便的机制来配置和整合多个数据源，从而提高数据管理的灵活性和效率。本文将逐步指导您如何在 Spring Boot 项目中实现多数据源配置和整合。

添加依赖项

首先，在项目中添加必要的依赖项：

dependencies {
    implementation 'org.springframework.boot:spring-boot-starter-data-jpa'
    implementation 'org.springframework.boot:spring-boot-starter-jdbc'
    runtimeOnly 'mysql:mysql-connector-java'
    runtimeOnly 'org.postgresql:postgresql'
}

配置数据源

在 application.properties 或 application.yml 文件中配置数据源连接信息：

# 主数据源
spring.datasource.primary.url=jdbc:mysql://localhost:3306/primary_db
spring.datasource.primary.username=root
spring.datasource.primary.password=root

# 次数据源
spring.datasource.secondary.url=jdbc:postgresql://localhost:5432/secondary_db
spring.datasource.secondary.username=postgres
spring.datasource.secondary.password=postgres

创建数据源配置类

为每个数据源创建配置类：

@Configuration
@EnableTransactionManagement
public class PrimaryDataSourceConfig {

    @Bean
    @Primary
    public DataSource primaryDataSource() {
        return new HikariDataSource(...); // 配置主数据源连接信息
    }
}

@Configuration
@EnableTransactionManagement
public class SecondaryDataSourceConfig {

    @Bean
    public DataSource secondaryDataSource() {
        return new HikariDataSource(...); // 配置次数据源连接信息
    }
}

配置数据源事务管理器

创建两个数据源的事务管理器：

@Configuration
public class TransactionManagerConfig {

    @Bean
    @Primary
    public PlatformTransactionManager primaryTransactionManager() {
        return new JpaTransactionManager(...); // 指定主数据源
    }

    @Bean
    public PlatformTransactionManager secondaryTransactionManager() {
        return new JpaTransactionManager(...); // 指定次数据源
    }
}

配置 JPA 实体管理器工厂

创建两个 JPA 实体管理器工厂：

@Configuration
public class EntityManagerFactoryConfig {

    @Bean
    @Primary
    public LocalContainerEntityManagerFactoryBean primaryEntityManagerFactory() {
        return new LocalContainerEntityManagerFactoryBean(...); // 指定主数据源和相关配置
    }

    @Bean
    public LocalContainerEntityManagerFactoryBean secondaryEntityManagerFactory() {
        return new LocalContainerEntityManagerFactoryBean(...); // 指定次数据源和相关配置
    }
}

配置事务注解支持

启用事务注解支持：

@Configuration
public class TransactionalConfig {

    @Bean
    public TransactionalEventListener transactionalEventListener() {
        return new TransactionalEventListenerFactory().getTransactionalEventListener();
    }
}

在 DAO 层使用数据源

在 DAO 层中使用 Spring Data JPA 或 JDBC 来操作数据：

@Repository
public class UserRepository {

    @Autowired
    private JdbcTemplate primaryJdbcTemplate;

    @Autowired
    private JdbcTemplate secondaryJdbcTemplate;

    public List<User> findAllFromPrimary() {
        return primaryJdbcTemplate.query(...); // 查询主数据源
    }

    public List<User> findAllFromSecondary() {
        return secondaryJdbcTemplate.query(...); // 查询次数据源
    }
}