如何基于数据库值条件化加载Spring Boot中的RabbitMQ Exchange Bean？

可以通过Spring的条件注解机制实现这类动态Bean加载，以下是两种可行方案：

方案一：自定义Condition类（容器启动早期判断） #

这种方式直接在Spring容器初始化阶段执行数据库查询，不依赖Spring管理的DataSource Bean，适合需要尽早判断的场景：

1. 实现自定义Condition

import org.springframework.context.annotation.Condition;
import org.springframework.context.annotation.ConditionContext;
import org.springframework.core.type.AnnotatedTypeMetadata;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;

public class DatabaseExchangeCondition implements Condition {
    @Override
    public boolean matches(ConditionContext context, AnnotatedTypeMetadata metadata) {
        // 从环境变量读取数据库配置
        var env = context.getEnvironment();
        String url = env.getProperty("spring.datasource.url");
        String username = env.getProperty("spring.datasource.username");
        String password = env.getProperty("spring.datasource.password");
        String driver = env.getProperty("spring.datasource.driver-class-name");

        try {
            // 加载驱动并建立连接查询
            Class.forName(driver);
            try (Connection conn = DriverManager.getConnection(url, username, password)) {
                String sql = "select value from config where key = 'ETL'";
                try (PreparedStatement stmt = conn.prepareStatement(sql)) {
                    ResultSet rs = stmt.executeQuery();
                    if (rs.next()) {
                        return "CREATE".equals(rs.getString("value"));
                    }
                }
            }
        } catch (Exception e) {
            // 异常场景默认不创建Bean
            return false;
        }
        return false;
    }
}

2. 在Bean上使用@Conditional注解

@Bean
@Conditional(DatabaseExchangeCondition.class)
DirectExchange exchange() {
    return new DirectExchange("my.controller.exchange", true, false);
}

方案二：使用@ConditionalOnExpression（依赖Spring数据源） #

如果项目中已经由Spring管理DataSource和JdbcTemplate，可以通过这种更简洁的方式实现：

1. 创建配置读取类

import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Component;
import org.springframework.dao.EmptyResultDataAccessException;

@Component
public class ConfigChecker {
    private final JdbcTemplate jdbcTemplate;

    public ConfigChecker(JdbcTemplate jdbcTemplate) {
        this.jdbcTemplate = jdbcTemplate;
    }

    public boolean shouldCreateExchange() {
        try {
            String value = jdbcTemplate.queryForObject(
                "select value from config where key = 'ETL'", 
                String.class
            );
            return "CREATE".equals(value);
        } catch (EmptyResultDataAccessException e) {
            // 无匹配记录时默认不创建
            return false;
        }
    }
}

2. 在Bean上使用@ConditionalOnExpression注解

@Bean
@ConditionalOnExpression("@configChecker.shouldCreateExchange()")
DirectExchange exchange() {
    return new DirectExchange("my.controller.exchange", true, false);
}

注意事项 #

• 方案一中需要确保数据库驱动已经引入，且配置信息能在容器启动早期被读取到
• 方案二中要保证ConfigChecker Bean在条件判断前完成初始化，Spring Boot会自动处理依赖关系，但复杂场景下可配合@DependsOn注解明确依赖顺序
• 两种方案都建议在异常场景返回false，避免因数据库问题导致容器启动失败

内容的提问来源于stack exchange，提问作者Unknown User