Spring Boot自定义Starter自动配置：如何确保条件在DataSource创建后执行？

问题核心：@Conditional的matches方法是在Bean定义解析阶段执行的，此时DataSource还未完成实例化，直接尝试获取DataSource实例必然失败。你试过的@AutoConfigureOrder和@AutoConfiguration(after = DataSourceAutoConfiguration.class)没生效，大概率是因为时机不对或者逻辑没踩对Spring的生命周期节点。

下面是几个可行的解决方案：

方案1：将检查逻辑移到Bean初始化阶段（推荐） #

放弃在Condition阶段做检查，改用@ConditionalOnBean(DataSource.class)确保自动配置在DataSource存在后加载，然后把PostgreSQL的校验逻辑放到Bean的初始化方法中，这样能直接拿到已实例化的DataSource，不符合要求时可抛出异常终止上下文启动。

示例代码：

@AutoConfiguration(after = DataSourceAutoConfiguration.class)
@ConditionalOnBean(DataSource.class)
public class OutboxStarterAutoconfiguration {

    @Bean
    public OutboxService outboxService(DataSource dataSource) {
        // 初始化前先校验DataSource是否为PostgreSQL
        validatePostgreSQLDataSource(dataSource);
        return new OutboxService(dataSource);
    }

    private void validatePostgreSQLDataSource(DataSource dataSource) {
        try (Connection conn = dataSource.getConnection()) {
            String productName = conn.getMetaData().getDatabaseProductName();
            if (!"PostgreSQL".equalsIgnoreCase(productName)) {
                throw new IllegalStateException("必须使用PostgreSQL数据源");
            }
        } catch (SQLException e) {
            throw new RuntimeException("无法验证数据源类型", e);
        }
    }
}

如果需要更早触发校验，也可以用@PostConstruct注解：

@Service
public class DataSourceValidator {
    private final DataSource dataSource;

    public DataSourceValidator(DataSource dataSource) {
        this.dataSource = dataSource;
    }

    @PostConstruct
    public void validate() {
        // 同上校验逻辑
    }
}

方案2：在Condition中通过Bean定义/环境变量判断 #

如果必须在Condition阶段完成检查，不要尝试获取DataSource实例，而是通过Bean定义信息或环境配置来判断是否为PostgreSQL数据源：

public class PostgreSQLDataSourceCondition implements Condition {
    @Override
    public boolean matches(ConditionContext context, AnnotatedTypeMetadata metadata) {
        // 方式1：从环境变量读取数据源配置
        Environment env = context.getEnvironment();
        String driverClassName = env.getProperty("spring.datasource.driver-class-name");
        String url = env.getProperty("spring.datasource.url");
        
        if ((driverClassName != null && driverClassName.contains("postgresql")) 
            || (url != null && url.startsWith("jdbc:postgresql:"))) {
            return true;
        }

        // 方式2：检查已注册的DataSource Bean定义（适用于自定义DataSource Bean的场景）
        String[] dataSourceBeanNames = BeanFactoryUtils.beanNamesForTypeIncludingAncestors(
                context.getBeanFactory(), DataSource.class, true, false);
        for (String beanName : dataSourceBeanNames) {
            BeanDefinition bd = context.getRegistry().getBeanDefinition(beanName);
            // 检查Bean定义中的属性（比如驱动类）
            if (bd instanceof AbstractBeanDefinition) {
                AbstractBeanDefinition abd = (AbstractBeanDefinition) bd;
                Object driverProp = abd.getPropertyValues().get("driverClassName");
                if (driverProp != null && driverProp.toString().contains("postgresql")) {
                    return true;
                }
            }
        }
        return false;
    }
}

然后在自动配置类上使用：

@AutoConfiguration(after = DataSourceAutoConfiguration.class)
@Conditional(PostgreSQLDataSourceCondition.class)
public class OutboxStarterAutoconfiguration {
    // ... 业务逻辑
}

方案3：确保自动配置的顺序生效 #

如果之前用@AutoConfiguration(after = DataSourceAutoConfiguration.class)没生效，检查以下两点：

1. 确认你的自动配置类已注册到META-INF/spring/org.springframework.boot.autoconfigure.AutoConfiguration.imports文件中，文件内容是自动配置类的全限定名，每行一个。
2. 不要在ContextInitAutoconfiguration上过早添加条件，否则可能导致其在DataSourceAutoConfiguration之前被处理。把条件移到OutboxStarterAutoconfiguration上，并确保它明确依赖DataSourceAutoConfiguration。

内容的提问来源于stack exchange，提问作者fabio trabattoni