java多数据源有一个错误能启动吗-CSDN博客

我整理的一些关于【Java】的项目学习资料（附讲解～～）和大家一起分享、学习一下：

https://d.51cto.com/f2PFnN

Java 多数据源配置与错误处理

在 Java 应用程序中，配置多个数据源是一个常见的需求，尤其是当我们需要从不同的数据库中读取数据时。不过，在多数据源配置中，当一个数据源出现错误时，我们需要确保应用程序仍然能够启动并继续运行。以下是实现这一目标的步骤和代码示例。

流程概览

以下是配置多数据源并处理错误的主要步骤：

步骤	描述
1	创建数据源配置类
2	配置主要数据源和备用数据源
3	使用Spring Boot的`@Primary`注解
4	处理数据源初始化错误
5	测试数据源连接

flowchart TD
    A[开始] --> B[创建数据源配置类]
    B --> C[配置主要数据源和备用数据源]
    C --> D[使用@Primary注解]
    D --> E[处理初始化错误]
    E --> F[测试数据源连接]
    F --> G[结束]

步骤详解

1. 创建数据源配置类

首先，我们需要创建一个配置类来管理多个数据源。

import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.autoconfigure.jdbc.DataSourceProperties;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.jdbc.DataSourceBuilder;
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 {
    
    // 创建主要数据源
    @Primary
    @Bean(name = "primaryDataSource")
    @ConfigurationProperties("spring.datasource.primary")
    public DataSource primaryDataSource() {
        return DataSourceBuilder.create().build();
    }

    // 创建备用数据源
    @Bean(name = "secondaryDataSource")
    @ConfigurationProperties("spring.datasource.secondary")
    public DataSource secondaryDataSource() {
        return DataSourceBuilder.create().build();
    }
}

解释：这里我们使用 Spring Boot 提供的 DataSourceBuilder 来构建两个数据源。@Primary 注解使得 primaryDataSource 成为默认数据源。

2. 配置主要数据源和备用数据源

在 application.yml 文件中配置数据源的信息。

spring:
  datasource:
    primary:
      url: jdbc:mysql://localhost:3306/primary_db
      username: root
      password: password
    secondary:
      url: jdbc:mysql://localhost:3306/secondary_db
      username: root
      password: another_password

解释：这里配置两个数据源的 URL、用户名和密码。

3. 使用 @Primary 注解

通过使用 @Primary 注解，Spring 会优先使用主要数据源。备用数据源仅在主要数据源初始化失败时使用。

4. 处理数据源初始化错误

我们可以通过实现 AbstractDataSource 或 DataSourceInitializer 来捕捉数据源初始化过程中的异常，并处理这些异常。

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.context.event.ApplicationReadyEvent;
import org.springframework.context.ApplicationListener;
import org.springframework.stereotype.Component;

@Component
public class DataSourceErrorListener implements ApplicationListener<ApplicationReadyEvent> {

    @Autowired
    @Qualifier("primaryDataSource")
    private DataSource primaryDataSource;

    @Override
    public void onApplicationEvent(ApplicationReadyEvent event) {
        try {
            // 尝试验证数据源连接
            primaryDataSource.getConnection().isValid(1);
        } catch (Exception e) {
            System.out.println("主要数据源无法启动，使用备用数据源。" + e.getMessage());
            // 这里可以添加更多处理逻辑
        }
    }
}

解释：在应用启动时，我们尝试获取主数据源的连接，如果失败则输出错误信息，随后可以执行备用数据源的逻辑。

5. 测试数据源连接

最后，我们可以编写单元测试来确保程序在执行时能够成功连接主要或备用的数据源。

import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;

import javax.sql.DataSource;
import java.sql.Connection;

@SpringBootTest
public class DataSourceTest {

    @Autowired
    private DataSource primaryDataSource;

    @Autowired
    private DataSource secondaryDataSource;

    @Test
    void testPrimaryDataSourceConnection() throws Exception {
        try (Connection connection = primaryDataSource.getConnection()) {
            assert connection.isValid(1);
        }
    }

    @Test
    void testSecondaryDataSourceConnection() throws Exception {
        try (Connection connection = secondaryDataSource.getConnection()) {
            assert connection.isValid(1);
        }
    }
}