Java实现多数据源配置

在Java开发中，尤其是大型企业级应用，通常需要连接多个数据库以处理不同的数据业务需求。在传统的单一数据源配置中，连接数据库相对简单，但当需求演变为多数据源时，就需要一种灵活的解决方案。本文将介绍如何在Spring Boot应用中实现多数据源配置，并提供代码示例。

多数据源配置的背景

多个数据源的配置能够帮助我们按需连接不同的数据库，例如，主数据库处理核心业务数据，而从数据库可以用来支持报表或审计等功能。为了实现这一点，我们需要配置多个数据源连接，并通过Spring管理这些连接。

代码示例

1. 引入依赖

在pom.xml中添加所需的依赖：

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-jpa</artifactId>
</dependency>
<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
</dependency>

2. 配置数据源

在application.yml文件中配置两个数据源：

spring:
  datasource:
    main:
      url: jdbc:mysql://localhost:3306/main_db
      username: root
      password: password
      driver-class-name: com.mysql.cj.jdbc.Driver
    secondary:
      url: jdbc:mysql://localhost:3306/secondary_db
      username: root
      password: password
      driver-class-name: com.mysql.cj.jdbc.Driver

3. 创建数据源配置类

首先创建两个数据源的配置类，以便Spring能正确识别它们。

@Configuration
@EnableTransactionManagement
public class DataSourceConfig {

    @Primary
    @Bean(name = "mainDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.main")
    public DataSource mainDataSource() {
        return DataSourceBuilder.create().build();
    }

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

    @Bean(name = "mainEntityManagerFactory")
    public LocalContainerEntityManagerFactoryBean mainEntityManagerFactory(
            EntityManagerFactoryBuilder builder,
            @Qualifier("mainDataSource") DataSource dataSource) {
        return builder
                .dataSource(dataSource)
                .packages("com.example.main.model") // 主数据包
                .persistenceUnit("main")
                .build();
    }

    @Bean(name = "secondaryEntityManagerFactory")
    public LocalContainerEntityManagerFactoryBean secondaryEntityManagerFactory(
            EntityManagerFactoryBuilder builder,
            @Qualifier("secondaryDataSource") DataSource dataSource) {
        return builder
                .dataSource(dataSource)
                .packages("com.example.secondary.model") // 从数据包
                .persistenceUnit("secondary")
                .build();
    }
}

4. 使用数据源

最后，在服务中使用这些配置：

@Service
public class MainService {

    @Autowired
    @Qualifier("mainEntityManagerFactory")
    private EntityManagerFactory mainEntityManagerFactory;

    @Autowired
    @Qualifier("secondaryEntityManagerFactory")
    private EntityManagerFactory secondaryEntityManagerFactory;

    public void performOperations() {
        // 使用 mainEntityManagerFactory 处理主数据
        // 使用 secondaryEntityManagerFactory 处理次数据
    }
}

流程图

下面是多数据源配置的基本流程图：

flowchart TD
    A[应用启动] --> B{选择数据源}
    B -->|主数据源| C[主数据源配置]
    B -->|次数据源| D[次数据源配置]
    C --> E[连接到主数据库]
    D --> F[连接到次数据库]
    E --> G[执行操作]
    F --> H[执行操作]

结论

在Java中实现多数据源配置虽然增加了复杂性，但通过Spring Boot的灵活配置和自动化管理，可以快速、有效地连接多个数据库。借助上述配置和代码示例，开发者可以根据需求自由切换数据源，从而实现更复杂的数据交互和业务逻辑处理。这一配置方式不仅适合于大型企业应用，也可以在高并发、高可用的环境中使用。希望这篇文章能为您的多数据源开发提供帮助！