Springboot扩展点之BeanDefinitionRegistryPostProcessor,你学会了吗?

2023年 11月 27日 65.5k 0

前言

通过这篇文章来大家分享一下,另外一个Springboot的扩展点BeanDefinitionRegistryPostProcessor,一般称这类扩展点为容器级后置处理器,另外一类是Bean级的后置处理器;容器级的后置处理器会在Spring容器初始化后、刷新前这个时间执行一次,Bean级的后置处理器,则是在每一个Bean实例化前后都会执行。

图片图片

功能特性

  • postProcessBeanDefinitionRegistry()方法 可以通过BeanDefinitionRegistry对BeanDefintion进行增删改查;
  • 继承了BeanFactoryPostProcessor,BeanFactoryPostProcessor是容器级别的扩展接口,org.springframework.beans.factory.config.BeanFactoryPostProcessor#postProcessBeanFactory方法在容器实例化后、刷新容器前被执行,即在容器刷新前还可以对BeanDefintion再作一些操作;
  • public interface BeanDefinitionRegistryPostProcessor extends BeanFactoryPostProcessor {
       void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry registry) throws BeansException;
    }
    @FunctionalInterface
    public interface BeanFactoryPostProcessor {
       void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException;
    }

    总结起来就是,在所有的BeanDefinition加载完成之后,Bean真正被实例化之前,可以通过实现BeanDefinitionRegistryPostProcessor接口,对BeanDefinition再做一些定制化的操作,比如修改某个bean的BeanDefinition的属性、手动注册一些复杂的Bean。

    对于Spring原理不太熟悉的小伙伴心里看到这可能有点晕了,BeanDefinition是什么?BeanDefinitionRegistry又是什么?ConfigurableListableBeanFactory又又是什么?别着急,这里拐个弯简单的解释一下,方便下面的内容理解起来更顺畅。

    BeanDefinition

    大家都知道,Spring的核心之一是IOC(控制反转),Spring之所以可以实现bean控制权的反转,是因为Spring的容器功能,在bean纳入Spring容器管理前,所有bean会被抽象封装成一个BeanDefinition实例,然后会在不同的时机根据BeanDefinition实例信息对bean进行实例化。

    简单说,Dog.java描述狗这一类动物的属性和行为,BeanDefinition描述Dog.java这个类。

    BeanDefinitionRegistry

    BeanDefinitionRegistry从字面意思看是bean的定义信息的注册登记,其实这个类的功能和字面意思一样,就是对BeanDefinition进行管理(增删改查);

    public interface BeanDefinitionRegistry extends AliasRegistry {
        //注册beanDefinition
       void registerBeanDefinition(String beanName, BeanDefinition beanDefinition)
             throws BeanDefinitionStoreException;
        //移除指定的beanDefinition
       void removeBeanDefinition(String beanName) throws NoSuchBeanDefinitionException;
        //根据beanName查询beanDefinition
       BeanDefinition getBeanDefinition(String beanName) throws NoSuchBeanDefinitionException;
        //判断某个beanDefinition是否已经注册
       boolean containsBeanDefinition(String beanName);
        //获取所有已注册的beanDefinition
       String[] getBeanDefinitionNames();
        //获取所有已注册的beanDefinition的数量
       int getBeanDefinitionCount();
        //判断某个beanDefinition是否已经被使用
       boolean isBeanNameInUse(String beanName);
    
    
    }

    ConfigurableListableBeanFactory

    上面提到了Spring的容器,Spring的核心之一是IOC,那么Spring的容器设计就是核心中的核心了。Spring的容器有多种形态,最基础的形态就是BeanFactory,ConfigurableListableBeanFactory间接继承了BeanFactory,因此ConfigurableListableBeanFactory实现类除了有Spring基础版本容器的功能外,还有一些高级的功能,Springboot默认的实际实现是DefaultListableBeanFactory,有兴趣的小伙伴可以以此为入口深入探究一番,这里不展开细说了。

    图片图片

    自定义实现

    MyBeanDefinitionRegistryPostProcessor

    下面通过一个具体类MyBeanDefinitionRegistryPostProcessor实现BeanDefinitionRegistryPostProcessor接口,来探究BeanDefinitionRegistryPostProcessor实现类的初始化和执行过程。

  • 在postProcessBeanDefinitionRegistry()方法被调用的时候手工在Spring中注册了Dog类的BeanDefinition信息;
  • 在postProcessBeanFactory()方法被调用的时候,从Spring容器中取出Dog类的BeanDefinition信息和Dog类的实例;
  • @Data
    public class Dog {
        private String name;
        private String color;
    }
    @Component
    public class MyBeanDefinitionRegistryPostProcessor implements BeanDefinitionRegistryPostProcessor {
        @Override
        public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry registry) throws BeansException {
            //手工定义一个beanDefinition实例
            RootBeanDefinition beanDefinition = new RootBeanDefinition();
            //给beanDefinition填充属性
            beanDefinition.setBeanClass(Dog.class);
            MutablePropertyValues propertyValues = new MutablePropertyValues();
            PropertyValue propertyValue1 = new PropertyValue("name", "旺财");
            PropertyValue propertyValue2 = new PropertyValue("color", "黑色");
            propertyValues.addPropertyValue(propertyValue1);
            propertyValues.addPropertyValue(propertyValue2);
            beanDefinition.setPropertyValues(propertyValues);
            //注册手工定义的beanDefinition
            registry.registerBeanDefinition("dog", beanDefinition);
        }
        @Override
        public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
            System.out.println("-----------start------------");
            //根据类名取出手工注册的beanDefinition
            BeanDefinition beanDefinition = beanFactory.getBeanDefinition("dog");
            System.out.println(beanDefinition.getBeanClassName());
            //根据类从容器中取出手工注册的beanDefinition所描述的实例bean
            Dog dog = beanFactory.getBean(Dog.class);
            System.out.println(dog.getName());
            System.out.println(dog.getColor());
            System.out.println("-----------end------------");
        }
    }

    单元测试

    @Test
    public void test(){
        AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext("com.fanfu");
        Dog dog = ((Dog) context.getBean("dog"));
        System.out.println(dog.getName());
        System.out.println(dog.getColor());
    }

    UML类图

    通过BeanDefinitionRegistryPostProcessorUML类图可以看出BeanDefinitionRegistryPostProcessor继承了BeanFactoryPostProcessor,postProcessBeanDefinitionRegistry()方法属于BeanDefinitionRegistryPostProcessor,postProcessBeanFactory()属于BeanFactoryPostProcessor,所有实现了BeanDefinitionRegistryPostProcessor接口的实现类都需要实现这个方法,而作为Springboot的扩展点之一,其扩展的逻辑也在这两个方法中;

    图片图片

    初始化和执行时机

    通过自定义的MyBeanDefinitionRegistryPostProcessor类,实现BeanDefinitionRegistryPostProcessor接口,从项目启动开始,其执行过程如下:

  • 执行项目的主类,SpringApplication#run被调用;
  • 进入boot.SpringApplication#run方法后,刚开始是一些Spring容器初始化的配置操作,直到执行到SpringApplication#refreshContext,开始容器刷新,进入了关键阶段;
  • 在SpringApplication#refreshContext,实际的刷新逻辑是在AbstractApplicationContext#refresh方法中;
  • AbstractApplicationContext#refresh方法中,调用AbstractApplicationContext#invokeBeanFactoryPostProcessors开始初始化和执行实现BeanDefinitionRegistryPostProcessor接口的postProcessBeanDefinitionRegistry()和postProcessBeanFactory();
  • 进入AbstractApplicationContext#invokeBeanFactoryPostProcessors方法,发现又调用了PostProcessorRegistrationDelegate#invokeBeanFactoryPostProcessors();
  • 在PostProcessorRegistrationDelegate#invokeBeanFactoryPostProcessors()方法中,并不是直接就初始化和执行postProcessBeanDefinitionRegistry()和postProcessBeanFactory(),而是又进行了一系列的判断,其判断顺序是:1、通过AbstractApplicationContext#addBeanFactoryPostProcessor提前注册的BeanDefinitionRegistryPostProcessor实现类;2、实现了PriorityOrdered接口;3、是否实现了Ordered;4、剩下的其他BeanDefinitionRegistryPostProcessor实现类;自定义的MyBeanDefinitionRegistryPostProcessor就属于第4类,所以是所有实现里较晚才被执行的,如果想要提前被执行,可以考虑前面三种方式;
  • 在PostProcessorRegistrationDelegate#invokeBeanFactoryPostProcessors()方法中执行完MyBeanDefinitionRegistryPostProcessor#postProcessBeanDefinitionRegistry方法后,紧接着就开始执行MyBeanDefinitionRegistryPostProcessor#postProcessBeanFactory方法了;从整个调用过程看postProcessBeanDefinitionRegistry()是早于postProcessBeanFactory()方法执行;
  • 下面是我根据整个调用过程画的一个时序图,过程确实比较复杂,但是逻辑比较清晰,因此并不难理解,想要真的搞清楚整个过程,最好的方法就是照着这个图,亲自执行一遍,通过debug观察每一个关键节点的执行过程。

    图片图片

    内部实现类

    spring-boot-starter-web中内置的实现类有CachingMetadataReaderFactoryPostProcessor、ConfigurationClassPostProcessor、ConfigurationWarningsPostProcessor、EmbeddedDataSourceBeanFactoryPostProcessor、ImportsCleanupPostProcessor、TestRestTemplateRegistrar、WebTestClientRegistrar、WsdlDefinitionBeanFactoryPostProcessor,观察一下每个实现类会发现:都比较类似,这些内置实现类都是Springboot中的内部类,通过这些BeanDefinitionRegistryPostProcessor内部实现类向Spring容器中注册了一些特殊的BeanDefinition,如果展开详细再说一说这些Bean,怕是一天一夜也说不完,有兴趣的小伙伴可以深入了解一下,这里就不再展开了。

    总结

    通过梳理整个过程,其实最关键的就是一句话:在Spring容器初始后、未刷新前,即Bean已被扫描注册为BeanDefinition后,未正式实例化前,可以通过实现BeanDefinitionRegistryPostProcessor做一些额外的操作。

    相关文章

    JavaScript2024新功能:Object.groupBy、正则表达式v标志
    PHP trim 函数对多字节字符的使用和限制
    新函数 json_validate() 、randomizer 类扩展…20 个PHP 8.3 新特性全面解析
    使用HTMX为WordPress增效:如何在不使用复杂框架的情况下增强平台功能
    为React 19做准备:WordPress 6.6用户指南
    如何删除WordPress中的所有评论

    发布评论