Spring启动提前获取MongoTemplate致配置失效问题及方案咨询

根本原因 #

这不是Spring的Bug，完全是操作时机的问题。

MongoTemplate的初始化依赖Spring Boot对Mongo配置（application.yml/application.properties）的绑定，而你之前尝试的几种方式要么执行时机过早，要么配置类加载顺序不对，导致MongoTemplate在配置绑定完成前就被提前初始化，只能使用默认的localhost:27017配置：

• BeanDefinitionRegistryPostProcessor在BeanDefinition注册阶段执行，此时Mongo的自动配置类（MongoAutoConfiguration）还没完成配置绑定，MongoTemplate的BeanDefinition还是默认配置，甚至未完成初始化。
• 给BeanDefinitionRegistryPostProcessor注入MongoTemplate会得到null，因为这个处理器的执行早于大部分Bean的实例化阶段，MongoTemplate还未被创建。
• 直接在@Configuration方法参数注入得到默认实例，是因为你的自定义配置类加载顺序在MongoAutoConfiguration之前，此时MongoTemplate还没完成配置绑定。

正确解决方案 #

核心思路是：让自定义Bean的注册逻辑，在MongoTemplate完全初始化（绑定好yml配置）之后执行。推荐以下两种方法：

方法一：利用@AutoConfigureAfter控制配置加载顺序 #

在自定义Starter的配置类上添加@AutoConfigureAfter(MongoAutoConfiguration.class)，确保当前配置类在Mongo的自动配置完成后再执行，此时注入的MongoTemplate就是已经绑定好yml配置的实例。

示例代码：

@Configuration
// 确保在Mongo自动配置完成后再执行当前配置
@AutoConfigureAfter(MongoAutoConfiguration.class)
public class CustomMongoBeansAutoConfiguration {

    // 直接注入配置完成的MongoTemplate，创建自定义Bean
    @Bean
    public UserRepository userRepository(MongoTemplate mongoTemplate) {
        return new UserRepositoryImpl(mongoTemplate);
    }

    @Bean
    public OrderRepository orderRepository(MongoTemplate mongoTemplate) {
        return new OrderRepositoryImpl(mongoTemplate);
    }
}

方法二：动态注册Bean（批量创建场景） #

如果需要根据配置动态注册多个Bean，不要用BeanDefinitionRegistryPostProcessor，而是结合ApplicationContextAware和@AutoConfigureAfter，在ApplicationContext初始化完成前，使用已经配置好的MongoTemplate注册Bean。

示例代码：

@Configuration
@AutoConfigureAfter(MongoAutoConfiguration.class)
public class DynamicMongoBeanConfiguration {

    @Bean
    public ApplicationContextAware dynamicBeanRegistrar(MongoTemplate mongoTemplate) {
        return applicationContext -> {
            BeanDefinitionRegistry registry = (BeanDefinitionRegistry) applicationContext.getAutowireCapableBeanFactory();
            
            // 示例：从配置读取要创建的Bean列表，动态注册
            List<String> repoNames = Arrays.asList("productRepo", "categoryRepo");
            for (String repoName : repoNames) {
                BeanDefinition beanDef = BeanDefinitionBuilder
                        .genericBeanDefinition(GenericMongoRepository.class)
                        .addConstructorArgValue(mongoTemplate)
                        .addConstructorArgValue(repoName) // 假设自定义Repository需要名称参数
                        .getBeanDefinition();
                registry.registerBeanDefinition(repoName, beanDef);
            }
        };
    }
}

关键注意事项 #

1. 确保自定义Starter的META-INF/spring.factories文件中，将你的配置类加入自动配置列表：

org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
com.yourpackage.CustomMongoBeansAutoConfiguration,\
com.yourpackage.DynamicMongoBeanConfiguration

2. 不要在BeanDefinitionRegistryPostProcessor或ImportBeanDefinitionRegistrar中直接获取MongoTemplate，这些扩展点的执行时机过早，无法获取到配置完成的实例。

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