Spring Boot启动前注册Bean并注入配置类的方案问询

可行方案1：通过ApplicationContextInitializer提前注册Bean #

ApplicationContextInitializer是Spring启动阶段最早的扩展点之一，能在上下文刷新前完成Bean注册，确保后续配置类可以正常注入。

步骤与代码示例 #

1. 实现ApplicationContextInitializer接口，在initialize方法中调用外部服务获取凭证并注册Bean：

public class MongoCredentialsInitializer implements ApplicationContextInitializer<ConfigurableApplicationContext> {
    @Override
    public void initialize(ConfigurableApplicationContext context) {
        // 调用外部服务获取用户名和密码
        String username = fetchAuthUsernameFromExternalService();
        char[] password = fetchAuthPasswordFromExternalService().toCharArray();
        MongoCredentials credentials = MongoCredentials.createCredential(username, "target-db", password);
        
        // 将凭证对象注册为Spring Bean
        BeanDefinitionRegistry registry = (BeanDefinitionRegistry) context.getBeanFactory();
        GenericBeanDefinition beanDef = new GenericBeanDefinition();
        beanDef.setBeanClass(MongoCredentials.class);
        beanDef.setInstanceSupplier(() -> credentials);
        registry.registerBeanDefinition("mongoCredentials", beanDef);
    }

    // 模拟外部服务调用逻辑
    private String fetchAuthUsernameFromExternalService() {
        return "db-admin";
    }

    private String fetchAuthPasswordFromExternalService() {
        return "secure-pass-123";
    }
}

2. 注册Initializer到Spring Boot：

   • 方法一：在src/main/resources/META-INF/spring.factories中添加配置：

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

   • 方法二：在启动类上显式指定：

   @SpringBootApplication(initializers = MongoCredentialsInitializer.class)
   public class YourApplication {
       public static void main(String[] args) {
           SpringApplication.run(YourApplication.class, args);
       }
   }
   

3. 在AppConfig中注入并使用：

@Configuration
public class AppConfig {
    @Autowired
    private MongoCredentials mongoCredentials;

    @Bean
    public MongoTemplate mongoTemplate() {
        MongoClientSettings clientSettings = MongoClientSettings.builder()
                .credential(mongoCredentials)
                .applyConnectionString(new ConnectionString("mongodb://localhost:27017"))
                .build();
        MongoClient mongoClient = MongoClients.create(clientSettings);
        return new MongoTemplate(mongoClient, "target-db");
    }
}

可行方案2：显式定义MongoCredentials Bean并控制加载顺序 #

直接定义MongoCredentials的@Bean，通过@DependsOn显式声明依赖关系，确保凭证Bean优先于MongoTemplate加载。

代码示例 #

1. 单独配置凭证Bean：

@Configuration
public class MongoAuthConfig {
    @Bean
    public MongoCredentials mongoCredentials() {
        String username = fetchAuthUsernameFromExternalService();
        char[] password = fetchAuthPasswordFromExternalService().toCharArray();
        return MongoCredentials.createCredential(username, "target-db", password);
    }

    // 模拟外部服务调用
    private String fetchAuthUsernameFromExternalService() {
        return "db-admin";
    }

    private String fetchAuthPasswordFromExternalService() {
        return "secure-pass-123";
    }
}

2. 在AppConfig中声明依赖并注入：

@Configuration
@DependsOn("mongoCredentials") // 确保凭证Bean先初始化
public class AppConfig {
    @Autowired
    private MongoCredentials mongoCredentials;

    @Bean
    public MongoTemplate mongoTemplate() {
        // 同方案1的MongoTemplate创建逻辑
    }
}

替代方案：使用MongoClientSettingsCustomizer #

无需单独注册MongoCredentials Bean，直接通过自定义MongoClientSettings的方式注入凭证，适配Spring Boot自动配置逻辑。

代码示例 #

@Component
public class MongoAuthCustomizer implements MongoClientSettingsCustomizer {
    @Override
    public void customize(MongoClientSettings.Builder settingsBuilder) {
        // 调用外部服务获取凭证
        String username = fetchAuthUsernameFromExternalService();
        char[] password = fetchAuthPasswordFromExternalService().toCharArray();
        MongoCredentials credentials = MongoCredentials.createCredential(username, "target-db", password);
        settingsBuilder.credential(credentials);
    }

    // 模拟外部服务调用
    private String fetchAuthUsernameFromExternalService() {
        return "db-admin";
    }

    private String fetchAuthPasswordFromExternalService() {
        return "secure-pass-123";
    }
}

此时只需在application.yml/properties中配置基础MongoDB连接信息（如spring.data.mongodb.uri=mongodb://localhost:27017），Spring Boot会自动将自定义的凭证配置应用到MongoClient，进而创建可用的MongoTemplate。

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