entitymanagerfactory 多数据源

EntityManagerFactory 是 Java Persistence API (JPA) 中的一个关键接口，用于创建 EntityManager 实例。EntityManager 是 JPA 的主要接口，用于与持久化上下文进行交互。当涉及到多数据源时，配置和管理多个 EntityManagerFactory 实例变得尤为重要。

基础概念

EntityManagerFactory: 这是一个重量级的对象，通常在应用程序启动时创建，并在整个应用程序生命周期中共享。它负责创建 EntityManager 实例。
多数据源: 指的是应用程序需要连接到多个数据库实例。这在需要处理不同业务逻辑或数据隔离的场景中很常见。

类型与应用场景

读写分离: 主数据库用于写操作，从数据库用于读操作，适用于读多写少的场景。
分库分表: 将数据分散到多个数据库或表中，用于处理大规模数据和高并发访问。
多租户架构: 每个租户有自己的数据库，实现数据隔离和安全性。

配置多数据源

在 Spring Boot 中配置多数据源通常涉及以下步骤：

定义多个数据源:

@Configuration
public class DataSourceConfig {

    @Bean
    @Primary
    @ConfigurationProperties("spring.datasource.primary")
    public DataSource primaryDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    @ConfigurationProperties("spring.datasource.secondary")
    public DataSource secondaryDataSource() {
        return DataSourceBuilder.create().build();
    }
}

创建多个 EntityManagerFactory:

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

    @Primary
    @Bean
    public LocalContainerEntityManagerFactoryBean primaryEntityManagerFactory(
            EntityManagerFactoryBuilder builder, @Qualifier("primaryDataSource") DataSource dataSource) {
        return builder
                .dataSource(dataSource)
                .packages("com.example.primary.entities")
                .persistenceUnit("primary")
                .build();
    }

    @Primary
    @Bean
    public PlatformTransactionManager primaryTransactionManager(
            @Qualifier("primaryEntityManagerFactory") EntityManagerFactory entityManagerFactory) {
        return new JpaTransactionManager(entityManagerFactory);
    }
}

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

    @Bean
    public LocalContainerEntityManagerFactoryBean secondaryEntityManagerFactory(
            EntityManagerFactoryBuilder builder, @Qualifier("secondaryDataSource") DataSource dataSource) {
        return builder
                .dataSource(dataSource)
                .packages("com.example.secondary.entities")
                .persistenceUnit("secondary")
                .build();
    }

    @Bean
    public PlatformTransactionManager secondaryTransactionManager(
            @Qualifier("secondaryEntityManagerFactory") EntityManagerFactory entityManagerFactory) {
        return new JpaTransactionManager(entityManagerFactory);
    }
}

可能遇到的问题及解决方法

事务管理问题: 在多数据源环境下，确保事务管理的正确性非常重要。使用 @Transactional 注解时需要指定事务管理器。
实体类冲突: 如果不同数据源中有相同名称的实体类，可能会导致冲突。可以通过包路径或不同的持久化单元名称来区分。
连接池配置: 确保每个数据源都有合适的连接池配置，以避免资源耗尽或性能瓶颈。

示例代码

假设我们有两个数据源，分别对应两个不同的数据库，以下是如何在 Spring Boot 中配置和使用它们的示例：

@SpringBootApplication
public class MultiDataSourceApplication {

    public static void main(String[] args) {
        SpringApplication.run(MultiDataSourceApplication.class, args);
    }
}

@Entity
@Table(name = "user")
public class User {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
    private String name;
    // getters and setters
}

@Repository
public interface UserRepository extends JpaRepository<User, Long> {
}

@Service
public class UserService {

    @Autowired
    private UserRepository userRepository;

    @Transactional("primaryTransactionManager")
    public User saveUser(User user) {
        return userRepository.save(user);
    }
}

通过上述配置和代码，可以有效地管理和使用多个数据源，确保应用程序的灵活性和可扩展性。