com.github.datalking.context.support.PostProcessorRegistrationDelegate Maven / Gradle / Ivy
package com.github.datalking.context.support;
import com.github.datalking.aop.aspectj.AnnotationAwareAspectJAutoProxyCreator;
import com.github.datalking.beans.factory.config.BeanFactoryPostProcessor;
import com.github.datalking.beans.factory.config.BeanPostProcessor;
import com.github.datalking.beans.factory.config.ConfigurableListableBeanFactory;
import com.github.datalking.beans.factory.support.BeanDefinitionRegistry;
import com.github.datalking.beans.factory.support.BeanDefinitionRegistryPostProcessor;
import com.github.datalking.common.Ordered;
import com.github.datalking.common.PriorityOrdered;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
/**
* beanFactoryPostProcessor相关功能执行的代理类
*
* 一是触发beanFactoryPostProcessor，二是注册beanFactoryPostProcessor
*
* @author yaoo on 4/13/18
*/
public class PostProcessorRegistrationDelegate {
public static void invokeBeanFactoryPostProcessors(
Collection postProcessors,
ConfigurableListableBeanFactory beanFactory) {
for (BeanFactoryPostProcessor postProcessor : postProcessors) {
postProcessor.postProcessBeanFactory(beanFactory);
}
}
public static void invokeBeanFactoryPostProcessors(
ConfigurableListableBeanFactory beanFactory,
List beanFactoryPostProcessors) {
/// 如果beanFactory是可注册BeanDefinition的
if (beanFactory instanceof BeanDefinitionRegistry) {
BeanDefinitionRegistry registry = (BeanDefinitionRegistry) beanFactory;
// 获取BeanDefinitionRegistryPostProcessor类型的bean名称
String[] postProcessorNames = beanFactory.getBeanNamesForType(BeanDefinitionRegistryPostProcessor.class);
// 实例化默认的后置处理器，包括ConfigurationClassPostProcessor
List priorityOrderedPostProcessors = new ArrayList<>();
for (String ppName : postProcessorNames) {
priorityOrderedPostProcessors.add((BeanDefinitionRegistryPostProcessor) beanFactory.getBean(ppName));
}
// 扫描@Configuration、@Bean、@ComponentScan
// 注册AnnotationAwareAspectJAutoProxyCreator的BeanDefinition
invokeBeanDefinitionRegistryPostProcessors(priorityOrderedPostProcessors, registry);
}
/// 如果beanFactory不可注册BeanDefinition
else {
invokeBeanFactoryPostProcessors(beanFactoryPostProcessors, beanFactory);
}
}
private static void invokeBeanDefinitionRegistryPostProcessors(
Collection postProcessors,
BeanDefinitionRegistry registry) {
for (BeanDefinitionRegistryPostProcessor postProcessor : postProcessors) {
postProcessor.postProcessBeanDefinitionRegistry(registry);
}
}
public static void registerBeanPostProcessors(ConfigurableListableBeanFactory beanFactory,
AbstractApplicationContext applicationContext) {
// 获取BeanPostProcessor类型的bean名
String[] postProcessorNames = beanFactory.getBeanNamesForType(BeanPostProcessor.class);
List priorityOrderedPostProcessors = new ArrayList<>();
List orderedPostProcessorNames = new ArrayList<>();
List nonOrderedPostProcessorNames = new ArrayList<>();
for (String ppName : postProcessorNames) {
if (beanFactory.isTypeMatch(ppName, PriorityOrdered.class)) {
//BeanPostProcessor pp = beanFactory.getBean(ppName, BeanPostProcessor.class);
BeanPostProcessor pp = (BeanPostProcessor) beanFactory.getBean(ppName);
priorityOrderedPostProcessors.add(pp);
} else if (beanFactory.isTypeMatch(ppName, Ordered.class)) {
orderedPostProcessorNames.add(ppName);
} else {
nonOrderedPostProcessorNames.add(ppName);
}
}
registerBeanPostProcessors(beanFactory, priorityOrderedPostProcessors);
// 下面实例化 AutoProxyCreator
List orderedPostProcessors = new ArrayList<>();
for (String ppName : orderedPostProcessorNames) {
BeanPostProcessor pp = (BeanPostProcessor) beanFactory.getBean(ppName);
orderedPostProcessors.add(pp);
// 特殊初始化AnnotationAwareAspectJAutoProxyCreator的bean
if (pp instanceof AnnotationAwareAspectJAutoProxyCreator){
((AnnotationAwareAspectJAutoProxyCreator)pp).setBeanFactory(beanFactory);
}
}
registerBeanPostProcessors(beanFactory, orderedPostProcessors);
List nonOrderedPostProcessors = new ArrayList();
for (String ppName : nonOrderedPostProcessorNames) {
BeanPostProcessor pp = (BeanPostProcessor) beanFactory.getBean(ppName);
nonOrderedPostProcessors.add(pp);
}
registerBeanPostProcessors(beanFactory, nonOrderedPostProcessors);
}
private static void registerBeanPostProcessors(ConfigurableListableBeanFactory beanFactory,
List postProcessors) {
for (BeanPostProcessor postProcessor : postProcessors) {
beanFactory.addBeanPostProcessor(postProcessor);
}
}
}