ImportBeanDefinitionRegistrar+SPI简化Spring开发

最新推荐文章于 2025-02-28 22:34:54 发布

置顶 ascii_he

最新推荐文章于 2025-02-28 22:34:54 发布

阅读量417

点赞数

分类专栏： Java 文章标签： spring spi java

本文链接：https://blog.csdn.net/pretty_he720/article/details/105016326

版权

Java 专栏收录该内容

11 篇文章

订阅专栏

本文介绍如何使用ImportBeanDefinitionRegistrar和SPI机制简化Spring配置，通过自定义注解@EnableAutoRegistrar触发动态扩展，实现在传统Spring项目中去XML化及定制化功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

相信使用Spring的小伙伴会在开发中遇到配置各式各样的bean的情况，例如配置数据库，配置redis，配置AOP等。springboot已经很大程度上帮我们简化了配置，但实际业务开发中是否有其无法满足的地方呢？或者在传统spring项目中如何简化配置？如一些定制化的功能，再如去xml化。本文尝试使用ImportBeanDefinitionRegistrar+SPI的方式来实现。后文会有更多的应用实例。

前置要求：需要你对spring的ImportBeanDefinitionRegistrar以及java的SPI机制有一定的了解，不太了解的可以先搜搜相关资源。两者是动态扩展的利器，建议多研究。

首先定义一个注解EnableAutoRegistrar，这样在@Configuration的类上引入@EnableAutoRegistrar注解来触发我们的动态扩展。

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)
@Documented
@Import(ConfigurationRegistrar.class)
public @interface EnableAutoRegistrar {
}

这里使用spring的Import机制，会自动注入ConfigurationRegistrar的bean，在ConfigurationRegistrar中我们实现ImportBeanDefinitionRegistrar接口以及其它一些aware接口来获取上下文并实现动态定义bean。

public class ConfigurationRegistrar implements ImportBeanDefinitionRegistrar,
        ResourceLoaderAware, BeanFactoryAware, EnvironmentAware {

    private ConfigurationRegisterHandler handler = ServiceBootstrap.loadPrimary(ConfigurationRegisterHandler.class);

    @Setter
    private ResourceLoader resourceLoader;
    @Setter
    private BeanFactory beanFactory;
    @Setter
    private Environment environment;

    @Override
    public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) {
        handler.registerBeanDefinitions(new RegisterBeanDefinitionContext()
                .setImportingClassMetadata(importingClassMetadata)
                .setRegistry(registry)
                .setResourceLoader(resourceLoader)
                .setBeanFactory(beanFactory)
                .setEnvironment(environment));
    }
}

这里我们并没有直接写动态定义bean的处理实际逻辑，为方便扩展，这里用的是SPI机制，从META-INF/services下加载ConfigurationRegisterHandler来委托其处理。
ConfigurationRegisterHandler定义：

public interface ConfigurationRegisterHandler extends Ordered {

    /**
     * 处理注册
     */
    void registerBeanDefinitions(RegisterBeanDefinitionContext context);

}

Ordered是为方便排序按顺序处理
其实现：

public class DefaultConfigurationRegisterHandler implements ConfigurationRegisterHandler {

    private static final String HANDLER_DIR = "META-INF/coding4j/";

    @Override
    public void registerBeanDefinitions(RegisterBeanDefinitionContext context) {
        final Multimap<String, ConfigurationRegisterHandler> handlerMap =
                ServiceBootstrap.loadAllOrdered(HANDLER_DIR, ConfigurationRegisterHandler.class);
        if (handlerMap != null) {
            // 同一类的只取第一个Handler处理，所有Handler会再排序处理
            handlerMap.keySet().stream()
                    .map(k -> handlerMap.get(k).iterator().next())
                    .sorted(Comparator.comparingInt(x -> x.getOrder()))
                    .forEach(x -> x.registerBeanDefinitions(context));
        }
    }

    @Override
    public int getOrder() {
        return Ordered.LOWEST_PRECEDENCE;
    }
}