1. 程式人生 > 其它 >Spring原始碼 10 IOC refresh方法6

Spring原始碼 10 IOC refresh方法6

Spring IOC 的核心是 AbstractApplicationContextrefresh 方法。
其中一共有 13 個主要方法,這裡分析第 6 個:registerBeanPostProcessors

1 AbstractApplicationContext

1-1 註冊 Bean 增強器

registerBeanPostProcessors(beanFactory)
protected void registerBeanPostProcessors(ConfigurableListableBeanFactory beanFactory) {
    // 註冊 Bean 增強器
    PostProcessorRegistrationDelegate.registerBeanPostProcessors(beanFactory, this);
}

1-2 註冊 Bean 增強器

PostProcessorRegistrationDelegate.registerBeanPostProcessors(beanFactory, this)

2 PostProcessorRegistrationDelegate

public static void registerBeanPostProcessors(ConfigurableListableBeanFactory beanFactory, AbstractApplicationContext applicationContext) {
   String[] postProcessorNames = beanFactory.getBeanNamesForType(BeanPostProcessor.class, true, false);
   /**
    * BeanPostProcessorChecker 是一個普通的資訊列印,可能會有些情況
    * 當 Spring 的配置中的後處理器還沒有被註冊就已經開始了 bean 的初始化時
    * 便會打印出 BeanPostProcessorChecker 中設定的資訊
    */
   int beanProcessorTargetCount = beanFactory.getBeanPostProcessorCount() + 1 + postProcessorNames.length;
   beanFactory.addBeanPostProcessor(new BeanPostProcessorChecker(beanFactory, beanProcessorTargetCount));
   // 使用 PriorityOrdered 保證順序
   List<BeanPostProcessor> priorityOrderedPostProcessors = new ArrayList<>();
   List<BeanPostProcessor> internalPostProcessors = new ArrayList<>();
   // 使用 Ordered 保證順序
   List<String> orderedPostProcessorNames = new ArrayList<>();
   // 無序 BeanPostProcessor
   List<String> nonOrderedPostProcessorNames = new ArrayList<>();
   for (String ppName : postProcessorNames) {
      if (beanFactory.isTypeMatch(ppName, PriorityOrdered.class)) {
         BeanPostProcessor pp = beanFactory.getBean(ppName, BeanPostProcessor.class);
         priorityOrderedPostProcessors.add(pp);
         if (pp instanceof MergedBeanDefinitionPostProcessor) {
            internalPostProcessors.add(pp);
         }
      } else if (beanFactory.isTypeMatch(ppName, Ordered.class)) {
         orderedPostProcessorNames.add(ppName);
      } else {
         nonOrderedPostProcessorNames.add(ppName);
      }
   }
   // 第 1 步,排序增強器。註冊所有實現 PriorityOrdered 的 BeanPostProcessor
   sortPostProcessors(priorityOrderedPostProcessors, beanFactory);
   // 註冊 Bean 增強器
   registerBeanPostProcessors(beanFactory, priorityOrderedPostProcessors);
   // 第 2 步,註冊所有實現 Ordered 的 BeanPostProcessor
   List<BeanPostProcessor> orderedPostProcessors = new ArrayList<>(orderedPostProcessorNames.size());
   for (String ppName : orderedPostProcessorNames) {
      BeanPostProcessor pp = beanFactory.getBean(ppName, BeanPostProcessor.class);
      orderedPostProcessors.add(pp);
      if (pp instanceof MergedBeanDefinitionPostProcessor) {
         internalPostProcessors.add(pp);
      }
   }
   sortPostProcessors(orderedPostProcessors, beanFactory);
   registerBeanPostProcessors(beanFactory, orderedPostProcessors);
   // 第 3 步,註冊所有無序的 BeanPostProcessor
   List<BeanPostProcessor> nonOrderedPostProcessors = new ArrayList<>(nonOrderedPostProcessorNames.size());
   for (String ppName : nonOrderedPostProcessorNames) {
      BeanPostProcessor pp = beanFactory.getBean(ppName, BeanPostProcessor.class);
      nonOrderedPostProcessors.add(pp);
      if (pp instanceof MergedBeanDefinitionPostProcessor) {
         internalPostProcessors.add(pp);
      }
   }
   registerBeanPostProcessors(beanFactory, nonOrderedPostProcessors);
   /**
    * 第 4 步,註冊所有 MergedBeanDefinitionPostProcessor 型別的 BeanPostProcessor,並非重複註冊
    * 在 beanFactory.addBeanPostProcessor 中會先移除已經存在的 BeanPostProcessor
    */
   sortPostProcessors(internalPostProcessors, beanFactory);
   registerBeanPostProcessors(beanFactory, internalPostProcessors);
   // 新增 ApplicationListener 探測器
   beanFactory.addBeanPostProcessor(new ApplicationListenerDetector(applicationContext));
}

2-1 排序增強器

sortPostProcessors(priorityOrderedPostProcessors, beanFactory)
private static void sortPostProcessors(List<?> postProcessors, ConfigurableListableBeanFactory beanFactory) {
   if (postProcessors.size() <= 1) {
      return;
   }
   Comparator<Object> comparatorToUse = null;
   if (beanFactory instanceof DefaultListableBeanFactory) {
      comparatorToUse = ((DefaultListableBeanFactory) beanFactory).getDependencyComparator();
   }
   if (comparatorToUse == null) {
      comparatorToUse = OrderComparator.INSTANCE;
   }
   postProcessors.sort(comparatorToUse);
}

2-2 註冊 Bean 增強器

registerBeanPostProcessors(beanFactory, priorityOrderedPostProcessors)
private static void registerBeanPostProcessors(
      ConfigurableListableBeanFactory beanFactory, List<BeanPostProcessor> postProcessors) {

   if (beanFactory instanceof AbstractBeanFactory) {
      // Bulk addition is more efficient against our CopyOnWriteArrayList there
      ((AbstractBeanFactory) beanFactory).addBeanPostProcessors(postProcessors);
   }
   else {
      for (BeanPostProcessor postProcessor : postProcessors) {
         beanFactory.addBeanPostProcessor(postProcessor);
      }
   }
}