Spring Boot 2.7.2升级后：无法及时从AWS参数存储获取数据库凭证

一、Spring Boot 2.7.x相关变更说明 #

Spring Boot 2.7.x在自动配置顺序、环境初始化逻辑上有针对性调整，直接影响你的场景：

• 自动配置优先级提升：DataSourceAutoConfiguration的初始化优先级在2.7.x版本中有所提高，默认会优先于仅通过@Autowired注入到启动类的自定义Bean完成初始化，导致数据库连接池提前尝试创建连接。
• 环境初始化流程优化：2.7.x对Environment加载流程做了调整，若自定义凭证加载逻辑是在Bean实例化阶段（比如启动类的@Autowired依赖触发）执行，会滞后于DataSource的自动配置流程。

二、强制数据库连接等待凭证获取的解决方案 #

1. 改用ApplicationContextInitializer提前加载凭证 #

将凭证加载逻辑实现为ApplicationContextInitializer，在Spring上下文初始化前就把参数注入到环境中，确保DataSource初始化时能读取到有效凭证：

public class AwsParameterStoreInitializer implements ApplicationContextInitializer<ConfigurableApplicationContext> {

    @Override
    public void initialize(ConfigurableApplicationContext applicationContext) {
        ConfigurableEnvironment environment = applicationContext.getEnvironment();
        // 读取配置中定义的参数存储路径
        String usernameParam = environment.getProperty("spring.datasource.username.param");
        String passwordParam = environment.getProperty("spring.datasource.password.param");
        
        // 调用AWS SDK从Parameter Store获取凭证
        AWSParameterStoreClient client = AWSParameterStoreClientBuilder.defaultClient();
        String username = client.getParameter(usernameParam).getParameter().getValue();
        String password = client.getParameter(passwordParam).getParameter().getValue();
        
        // 将凭证添加到环境变量的最前面，确保优先级最高
        Map<String, Object> credentialProperties = new HashMap<>();
        credentialProperties.put("spring.datasource.username", username);
        credentialProperties.put("spring.datasource.password", password);
        environment.getPropertySources().addFirst(new MapPropertySource("aws-credentials", credentialProperties));
    }
}

随后在src/main/resources/META-INF/spring.factories中注册该初始化器：

org.springframework.context.ApplicationContextInitializer=com.yourpackage.AwsParameterStoreInitializer

2. 配置数据库连接池延迟初始化 #

如果使用Spring Boot默认的Hikari连接池，可通过配置让连接池在第一次实际使用时才初始化连接，而非启动阶段：

spring.datasource.hikari.initialization-fail-timeout=-1
spring.datasource.hikari.minimum-idle=0

同时给你的凭证加载Bean添加@Order(Ordered.HIGHEST_PRECEDENCE)注解，确保它在应用启动早期完成初始化。

3. 自定义DataSource并指定依赖关系 #

通过自定义DataSource Bean，明确依赖你的凭证加载Bean，强制凭证加载完成后再初始化DataSource：

@Configuration
public class DataSourceConfig {

    @Autowired
    private AwsCredentialLoader credentialLoader; // 你的自定义凭证加载类

    @Bean
    @DependsOn("awsCredentialLoader")
    public DataSource dataSource(Environment environment) {
        HikariDataSource dataSource = new HikariDataSource();
        dataSource.setJdbcUrl(environment.getProperty("spring.datasource.url"));
        dataSource.setUsername(environment.getProperty("spring.datasource.username"));
        dataSource.setPassword(environment.getProperty("spring.datasource.password"));
        // 补充其他连接池配置项
        return dataSource;
    }
}

注意：这种方式需要关闭默认的DataSource自动配置，在启动类添加@SpringBootApplication(exclude = DataSourceAutoConfiguration.class)。

4. 使用Spring Cloud AWS官方集成 #

引入Spring Cloud AWS Parameter Store依赖，让框架自动在环境初始化阶段加载参数，无需自定义逻辑：

<dependency>
    <groupId>io.awspring.cloud</groupId>
    <artifactId>spring-cloud-starter-aws-parameter-store-config</artifactId>
    <!-- 适配Spring Boot 2.7.x，建议使用3.x版本 -->
    <version>3.0.0</version>
</dependency>

配置参数存储前缀后，框架会自动将参数映射到Spring环境中，比如参数路径/config/datasource/username会对应spring.datasource.username。

三、排查建议 #

• 开启Bean加载日志：添加日志配置logging.level.org.springframework.beans.factory=DEBUG，查看控制台输出的Bean初始化顺序，对比你的凭证加载Bean和dataSourceBean的执行时间点。
• 标记关键节点日志：在凭证加载类的核心方法（比如@PostConstruct方法）中添加时间戳日志，同时记录数据库连接失败的异常时间，明确时序差异。
• 核对官方变更说明：仔细查阅Spring Boot 2.7官方发布说明中关于自动配置、环境初始化、Bean依赖的变更内容，确认是否有相关调整影响你的逻辑。
• 测试Bean优先级：给凭证加载Bean添加@Order(Ordered.HIGHEST_PRECEDENCE)或@Priority(0)注解，测试是否能优先于DataSource初始化。

内容的提问来源于stack exchange，提问作者Barbara Atkins