Spring中BeanDefinitionRegistryPostProcessor接口 -【Spring底层原理】

}

// 待注入的bean

public class User {

}

// BeanDefinitionRegistryPostProcessor实现类

@Component

public class MyBeanDefinitionRegistryPostProcessor implements BeanDefinitionRegistryPostProcessor {

public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry beanDefinitionRegistry) throws BeansException {

System.out.println(“MyBeanDefinitionRegistryPostProcessor#postProcessBeanDefinitionRegistry”);

System.out.println(“bean的数量：”+ beanDefinitionRegistry.getBeanDefinitionCount());

// 通过bean的构建器生成rootbean

AbstractBeanDefinition beanDefinition = BeanDefinitionBuilder.rootBeanDefinition(User.class).getBeanDefinition();

beanDefinitionRegistry.registerBeanDefinition(“hello”,beanDefinition);

}

public void postProcessBeanFactory(ConfigurableListableBeanFactory configurableListableBeanFactory) throws BeansException {

System.out.println(“MyBeanDefinitionRegistryPostProcessor#postProcessBeanFactory”);

System.out.println(“bean的数量：” + configurableListableBeanFactory.getBeanDefinitionCount());

}

}

// 配置类

@Configuration

@ComponentScan(“postprocessor”)

public class AppConfig {

}

可以看到输出结果如下，从中我们可以看出：

• 先执行postProcessBeanDefinitionRegistry方法，获取到IOC容器中bean的数量为7

• 然后给Registry里面注册一个hello

• 再执行postProcessBeanFactory方法，由于上面又注册了一个，所以这里获取到IOC容器中bean的数量为8

• 等BeanDefinitionRegistryPostProcessor里面的方法都执行完后，再执行BeanFactoryPostP

《一线大厂Java面试题解析+后端开发学习笔记+最新架构讲解视频+实战项目源码讲义》

【docs.qq.com/doc/DSmxTbFJ1cmN1R2dB】 完整内容开源分享

rocessor里面的方法

三、源码分析

同样，我们通过Debug的方式来分析源码，在postProcessBeanDefinitionRegistry方法处打个断点，启动断点调试，根据方法调用栈来进行跟踪：

通过查看方法调用栈，可以作出如下分析：

• 启动IOC容器调用refresh方法

• 再依次调用invokeBeanFactoryPostProcessors()——>invokeBeanDefinitionRegistryPostProcessors()方法

• 在invokeBeanDefinitionRegistryPostProcessors方法中获取到currentRegistryProcessors参数

public static void invokeBeanFactoryPostProcessors(ConfigurableListableBeanFactory beanFactory, List beanFactoryPostProcessors) {

…

// 获取到BeanDefinitionRegistryPostProcessor组件

postProcessorNames = beanFactory.getBeanNamesForType(BeanDefinitionRegistryPostProcessor.class, true, false);

var16 = postProcessorNames;

var9 = postProcessorNames.length;

for(var10 = 0; var10 < var9; ++var10) {

ppName = var16[var10];

if (!processedBeans.contains(ppName) && beanFactory.isTypeMatch(ppName, Ordered.class)) {

// 获取currentRegistryProcessors参数

currentRegistryProcessors.add(beanFactory.getBean(ppName, BeanDefinitionRegistryPostProcessor.class));

processedBeans.add(ppName);

}

}

…

while(reiterate) {

reiterate = false;

postProcessorNames = beanFactory.getBeanNamesForType(BeanDefinitionRegistryPostProcessor.class, true, false);

String[] var19 = postProcessorNames;

var10 = postProcessorNames.length;

for(int var26 = 0; var26 < var10; ++var26) {

String ppName = var19[var26];

if (!processedBeans.contains(ppName)) {

currentRegistryProcessors.add(beanFactory.getBean(ppName, BeanDefinitionRegistryPostProcessor.class));

processedBeans.add(ppName);

reiterate = true;

}

}

sortPostProcessors(currentRegistryProcessors, beanFactory);

registryProcessors.addAll(currentRegistryProcessors);

// 1.在这个方法里面触发postProcessBeanDefinnitionRegistry

invokeBeanDefinitionRegistryPostProcessors(currentRegistryProcessors, registry, beanFactory.getApplicationStartup());

currentRegistryProcessors.clear();

}

// 2.在这个方法里面触发postProcessBeanFactory方法

invokeBeanFactoryPostProcessors((Collection)registryProcessors, (ConfigurableListableBeanFactory)beanFactory);

invokeBeanFactoryPostProcessors((Collection)regularPostProcessors, (ConfigurableListableBeanFactory)beanFactory);

…

// 3.在这个方法里面触发postProcessorBeanFactory方法

invokeBeanFactoryPostProcessors((Collection)nonOrderedPostProcessors, (ConfigurableListableBeanFactory)beanFactory);

…

}

从源码可以看到：

• 先获取BeanDefinitionRegistryPostProcessor组件

• 再按照一定规则（如优先级）加入到currentRegistryProcessors数组中

• currentRegistryProcessors作为参数传给invokeBeanDefinitionRegistryPostProcessors方法，最后在这个方法执行

• 下图可以看到此时已经将自己写的MyBeanDefinitionRegistryPostProcessor组件给获取到了

【1】触发BeanDefinitionRegistryPostProcessor#postProcessBeanDefinitionRegistry

进入invokeBeanDefinitionRegistryPostProcessors方法，可以看到，在这里获取到组件并触发了postProcessBeanDefinitionRegistry：

private static void invokeBeanDefinitionRegistryPostProcessors(Collection<? extends BeanDefinitionRegistryPostProcessor> postProcessors, BeanDefinitionRegistry registry, ApplicationStartup applicationStartup) {

Iterator var3 = postProcessors.iterator();

while(var3.hasNext()) {

BeanDefinitionRegistryPostProcessor postProcessor = (BeanDefinitionRegistryPostProcessor)var3.next();

StartupStep var10000 = applicationStartup.start(“spring.context.beandef-registry.post-process”);

postProcessor.getClass();

StartupStep postProcessBeanDefRegistry = var10000.tag(“postProcessor”, postProcessor::toString);

// 触发postProcessBeanDefinnitionRegistry

postProcessor.postProcessBeanDefinitionRegistry(registry);

postProcessBeanDefRegistry.end();

}

}

【2】触发BeanDefinitionRegistryPostProcessor组件的BeanFactoryPostProcessor#postProcessBeanFactory

在invokeBeanDefinitionRegistryPostProcessors方法执行完后，通过invokeBeanFactoryPostProcessors方法执行postProcessBeanFactory，点进源码进行查看：

private static void invokeBeanFactoryPostProcessors(Collection<? extends BeanFactoryPostProcessor> postProcessors, ConfigurableListableBeanFactory beanFactory) {

Iterator var2 = postProcessors.iterator();

while(var2.hasNext()) {

BeanFactoryPostProcessor postProcessor = (BeanFactoryPostProcessor)var2.next();

StartupStep var10000 = beanFactory.getApplicationStartup().start(“spring.context.bean-factory.post-process”);

postProcessor.getClass();

StartupStep postProcessBeanFactory = var10000.tag(“postProcessor”, postProcessor::toString);

// 执行postProcessBeanFactory

postProcessor.postProcessBeanFactory(beanFactory);

postProcessBeanFactory.end();

}

}

【3】触发其他组件的BeanFactoryPostProcessor#postProcessBeanFactory

private static void invokeBeanFactoryPostProcessors(Collection<? extends BeanFactoryPostProcessor> postProcessors, ConfigurableListableBeanFactory beanFactory) {

Iterator var2 = postProcessors.iterator();