如何手动注册Spring @Configuration类？按需动态加载带注解配置类

核心思路 #

通过Spring的BeanDefinitionRegistryPostProcessor扩展点，在Bean定义注册阶段动态读取DynamoDB配置，按需将DBConfig注册为配置类，确保@EnableTransactionManagement、@EnableJpaRepositories等注解正常生效。

步骤1：改造DBConfig类 #

移除@Configuration注解，保留所有功能型注解和Bean定义方法：

@EnableTransactionManagement
@EnableJpaRepositories(basePackages = "com.yourpackage.repositories")
// 移除@Configuration，避免启动时自动加载
public class DBConfig {

    @Bean
    public DataSource dataSource() {
        // 你的数据源配置逻辑
        return new HikariDataSource();
    }

    @Bean
    public LocalContainerEntityManagerFactoryBean entityManagerFactory(DataSource dataSource) {
        // JPA实体管理器配置
        LocalContainerEntityManagerFactoryBean em = new LocalContainerEntityManagerFactoryBean();
        em.setDataSource(dataSource);
        em.setPackagesToScan("com.yourpackage.entities");
        em.setJpaVendorAdapter(new HibernateJpaVendorAdapter());
        return em;
    }

    // 其他相关Bean定义...
}

步骤2：实现动态注册处理器 #

创建BeanDefinitionRegistryPostProcessor实现类，注入你的MyConfigLoader服务，读取DynamoDB配置后决定是否注册DBConfig：

@Component
public class DBConfigConditionalRegistrar implements BeanDefinitionRegistryPostProcessor {

    private final MyConfigLoader myConfigLoader;

    // 构造注入配置加载器
    public DBConfigConditionalRegistrar(MyConfigLoader myConfigLoader) {
        this.myConfigLoader = myConfigLoader;
    }

    @Override
    public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry registry) throws BeansException {
        // 从DynamoDB读取dbRequired标识
        boolean dbRequired = myConfigLoader.isDbRequired();
        
        if (dbRequired) {
            // 检查是否已注册，避免重复
            String beanName = AnnotationBeanNameGenerator.INSTANCE.generateBeanName(
                new AnnotatedGenericBeanDefinition(DBConfig.class), registry);
            if (!registry.containsBeanDefinition(beanName)) {
                // 使用AnnotatedBeanDefinitionReader注册配置类，自动处理所有Spring注解
                AnnotatedBeanDefinitionReader reader = new AnnotatedBeanDefinitionReader(registry);
                reader.register(DBConfig.class);
            }
        }
    }

    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
        // 无需额外操作，空实现即可
    }
}

关键说明 #

• AnnotatedBeanDefinitionReader会完整解析类上的所有Spring注解，包括@EnableTransactionManagement、@EnableJpaRepositories这类导入型注解，确保它们的功能正常生效。
• MyConfigLoader需确保能在BeanDefinitionRegistryPostProcessor执行时正常初始化：如果它依赖DynamoDB客户端等Bean，需保证这些Bean在该阶段已可用（可通过@Lazy(false)或提前初始化配置）。
• 加入重复注册检查，避免多次调用导致的Bean定义冲突。

备选方案：ApplicationContextInitializer #

如果需要在Spring上下文初始化更早阶段处理，可使用ApplicationContextInitializer：

public class DBConfigInitializer implements ApplicationContextInitializer<ConfigurableApplicationContext> {

    @Override
    public void initialize(ConfigurableApplicationContext context) {
        // 实例化或获取MyConfigLoader
        MyConfigLoader configLoader = new MyConfigLoader(); // 或从context获取已注册的Bean
        boolean dbRequired = configLoader.isDbRequired();
        
        if (dbRequired) {
            AnnotatedBeanDefinitionReader reader = new AnnotatedBeanDefinitionReader(context);
            reader.register(DBConfig.class);
        }
    }
}

然后在启动类中添加初始器：

public static void main(String[] args) {
    SpringApplication app = new SpringApplication(YourApplication.class);
    app.addInitializers(new DBConfigInitializer());
    app.run(args);
}

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