神祕又強大的@SpringBootApplication註解

一、前言

大部分的配置都可以用Java類+註解來代替,而在SpringBoot項目中見的最多的莫過於@SpringBootApplication註解了,它在每個SpringBoot的啓動類上都有標註。

這個註解對SpringBoot的啓動和自動配置到底有什麼樣的影響呢?本文將爲各位大佬解析它的源碼,揭開@SpringBootApplication註解神祕的面紗。

二、正文

對SpringBoot工程的自動配置很感興趣,於是學習其源碼並整理了其中一些內容,如果有錯誤請大家指正~話不多說,直接上源碼;

@SpringBootApplication註解的源碼如下:

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
@SpringBootConfiguration
@EnableAutoConfiguration
@ComponentScan(excludeFilters = {
        @Filter(type = FilterType.CUSTOM, classes = TypeExcludeFilter.class),
        @Filter(type = FilterType.CUSTOM, classes = AutoConfigurationExcludeFilter.class) })
public @interface SpringBootApplication {
...
}

可以看到這是一個複合註解,一共包括7個不同的註解,下面對這7個不同的註解進行分析。

2.1 註解

2.1.1 註解1:@Target({ElementType.TYPE})

用來表示註解作用範圍,TYPE表示作用範圍爲類或接口。

2.1.2 註解2:@Retention(RetentionPolicy.RUNTIME)

2.1.3 註解3:@Documented

表明這個註釋是由 javadoc記錄的。

2.1.4 註解4:@Inherited

放在註解上,當父類加了@SpringBootApplication註解時,子類也會繼承這個註解(對接口的實現類無效)。

2.1.5 註解5:@SpringBootConfiguration

底層仍是@Configuration註解, 源碼如下:

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Configuration
public @interface SpringBootConfiguration {
}

2.1.6 註解6:@ComponetScan

@ComponentScan這個註解在Spring中很重要,它對應XML配置中的元素@ComponentScan的功能其實就是自動掃描並加載符合條件的組件(比如@Component和@Repository等)或者bean定義,最終將這些bean定義加載到IoC容器中。

可以通過basePackages等屬性來細粒度的定製@ComponentScan自動掃描的範圍,如果不指定,則默認Spring框架實現會從聲明@ComponentScan所在類的package進行掃描。所以SpringBoot的啓動類最好是放在root package下,因爲默認不指定basePackages。

2.2 註解:@EnableAutoConfiguration

個人感覺@EnableAutoConfiguration這個Annotation最爲重要它的作用可以概括爲:藉助@Import的幫助,將所有符合自動配置條件的bean定義加載到IoC容器。

其源碼如下:

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
@AutoConfigurationPackage
@Import(AutoConfigurationImportSelector.class)
public @interface EnableAutoConfiguration {
  String ENABLED_OVERRIDE_PROPERTY = "spring.boot.enableautoconfiguration";
  
  Class<?>[] exclude() default {};
  
  String[] excludeName() default {};
}

這裏需要關注@AutoConfigurationPackage和@Import(AutoConfigurationImportSelector.class)兩個註解。

2.2.1 註釋:@AutoConfigurationPackage

源碼如下:

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
@AutoConfigurationPackage
@Import(AutoConfigurationImportSelector.class)
public @interface EnableAutoConfiguration {
  String ENABLED_OVERRIDE_PROPERTY = "spring.boot.enableautoconfiguration";
  
  Class<?>[] exclude() default {};
  
  String[] excludeName() default {};
}

可以發現這個註解的核心其實也是Import註解,表示對於標註該註解的類的包,應當使用AutoConfigurationPackages註冊。接着看Registrar這個類:

static class Registrar implements ImportBeanDefinitionRegistrar, DeterminableImports {
​
    @Override
      //metadata是我們註解所在的元信息
    public void registerBeanDefinitions(AnnotationMetadata metadata, BeanDefinitionRegistry registry) {
            //將我們註解所在包下所有的組件進行註冊
      register(registry, new PackageImport(metadata).getPackageName());
    }
​
    @Override
    public Set<Object> determineImports(AnnotationMetadata metadata) {
      return Collections.singleton(new PackageImport(metadata));
    }
}

這個類中的核心方法是register方法:

private static final String BEAN = AutoConfigurationPackages.class.getName();
  
  public static void register(BeanDefinitionRegistry registry, String... packageNames) {
    if (registry.containsBeanDefinition(BEAN)) {
      BeanDefinition beanDefinition = registry.getBeanDefinition(BEAN);
      ConstructorArgumentValues constructorArguments = beanDefinition.getConstructorArgumentValues();
      constructorArguments.addIndexedArgumentValue(0, addBasePackages(constructorArguments, packageNames));
​    }
    else {
      GenericBeanDefinition beanDefinition = new GenericBeanDefinition();
      beanDefinition.setBeanClass(BasePackages.class);
      beanDefinition.getConstructorArgumentValues().addIndexedArgumentValue(0, packageNames);
      beanDefinition.setRole(BeanDefinition.ROLE_INFRASTRUCTURE);
      registry.registerBeanDefinition(BEAN, beanDefinition);
    }
}

register方法的邏輯非常清晰:如果這個bean已經被註冊,就獲取它的構造函數參數值,並將包名添加進去;否則就創建一個新的bean定義並進行註冊。通過@AutoConfigurationPackage這個註解,可以將註解所在包下所有的組件進行註冊。

2.2.2 註解:@Import(AutoConfigurationImportSelector.class)

這個註解導入了AutoConfigurationImportSelector這個類這個類的核心方法是selectImports方法,實現ImportSelector接口。方法基於我們在pom.xml文件中配置的jar包和組件進行導入。所以方法返回的是一個Class全路徑的String數組,返回的Class會被Spring容器管理。方法源碼如下:

@Override
public String[] selectImports(AnnotationMetadata annotationMetadata) {
  if (!isEnabled(annotationMetadata)) {
    return NO_IMPORTS;
  }
  AutoConfigurationMetadata autoConfigurationMetadata = AutoConfigurationMetadataLoader
      .loadMetadata(this.beanClassLoader);
  AutoConfigurationEntry autoConfigurationEntry = getAutoConfigurationEntry(autoConfigurationMetadata,
      annotationMetadata);
  return StringUtils.toStringArray(autoConfigurationEntry.getConfigurations());
}

這個方法的結構也很清晰,首先通過isEnabled方法判斷是否需要進行導入,如果需要導入的話,通過loadMetadata方法獲取配置信息,並通過getAutoConfigurationEntry進行自動裝配。isEnabled方法源碼如下:

protected boolean isEnabled(AnnotationMetadata metadata) {
  if (getClass() == AutoConfigurationImportSelector.class) {
    return getEnvironment().getProperty(EnableAutoConfiguration.ENABLED_OVERRIDE_PROPERTY, Boolean.class, true);
  }
  return true;
}

這個方法通過EnableAutoConfiguration.ENABLED_OVERRIDE_PROPERTY這個配置項進行判斷是否需要自動配置,默認爲true。loadMetadata方法源碼如下:


protected static final String PATH = "META-INF/" + "spring-autoconfigure-metadata.properties";
​
  public static AutoConfigurationMetadata loadMetadata(ClassLoader classLoader) {
    return loadMetadata(classLoader, PATH);
  }
​
  static AutoConfigurationMetadata loadMetadata(ClassLoader classLoader, String path) {
    try {
      Enumeration<URL> urls = (classLoader != null) ? classLoader.getResources(path)
          : ClassLoader.getSystemResources(path);
      Properties properties = new Properties();
      while (urls.hasMoreElements()) {
        properties.putAll(PropertiesLoaderUtils.loadProperties(new UrlResource(urls.nextElement())));
      }
      return loadMetadata(properties);
    }
    catch (IOException ex) {
      throw new IllegalArgumentException("Unable to load @ConditionalOnClass location [" + path + "]", ex);
    }
  }
  static AutoConfigurationMetadata loadMetadata(Properties properties) {
    return new PropertiesAutoConfigurationMetadata(properties);
  }

可以看到這個方法會加載META-INF/spring-autoconfigure-metadata.properties下的所有配置信息幷包裝成AutoConfigurationMetadata對象返回。

注:spring-autoconfigure-metadata.properties文件在spring-boot-autoconfigure-2.1.9.RELEASE.jar/META-INF下。

getAutoConfigurationEntry方法源碼如下:

protected AutoConfigurationEntry getAutoConfigurationEntry(AutoConfigurationMetadata autoConfigurationMetadata,
      AnnotationMetadata annotationMetadata) {
    if (!isEnabled(annotationMetadata)) {
      return EMPTY_ENTRY;
    }
    AnnotationAttributes attributes = getAttributes(annotationMetadata);
    List<String> configurations = getCandidateConfigurations(annotationMetadata, attributes);
    configurations = removeDuplicates(configurations);
    Set<String> exclusions = getExclusions(annotationMetadata, attributes);
    checkExcludedClasses(configurations, exclusions);
    configurations.removeAll(exclusions);
    configurations = filter(configurations, autoConfigurationMetadata);
    fireAutoConfigurationImportEvents(configurations, exclusions);
    return new AutoConfigurationEntry(configurations, exclusions);
}

這個方法是AutoConfiguration的主流程方法,可以將這個方法的每一行看做一個步驟,那麼處理步驟如下:

1. 加載配置了@EnableAutoConfiguration註解的屬性值getAttribute方法:

protected AnnotationAttributes getAttributes(AnnotationMetadata metadata) {
    String name = getAnnotationClass().getName();
    AnnotationAttributes attributes = AnnotationAttributes.fromMap(metadata.getAnnotationAttributes(name, true));
    Assert.notNull(attributes, () -> "No auto-configuration attributes found. Is " + metadata.getClassName()
        + " annotated with " + ClassUtils.getShortName(name) + "?");
    return attributes;
}

2.得到META-INF/spring.factories文件中以@EnableAutoConfiguration完全限定類名做key的value,getCandidateConfigurations方法:

​protected List<String> getCandidateConfigurations(AnnotationMetadata metadata, AnnotationAttributes attributes) {
    List<String> configurations = SpringFactoriesLoader.loadFactoryNames(getSpringFactoriesLoaderFactoryClass(),
        getBeanClassLoader());
    Assert.notEmpty(configurations, "No auto configuration classes found in META-INF/spring.factories. If you "
        + "are using a custom packaging, make sure that file is correct.");
    return configurations;
}
protected Class<?> getSpringFactoriesLoaderFactoryClass() {
    return EnableAutoConfiguration.class;
}

其中,SpringFactoriesLoader.loadFactoryNames()這個方法的作用是使用給定的類加載器從META-INF/spring.factories加載給定類型的工廠實現的完全限定類名;

3.去重;

4.得到需要排除的類的類名,這些類可以在@EnableAutoConfiguration註解中配置;

5.檢查這兩個集合;

6.把需要排除的類移除;

7.根據OnBeanCondition、OnClassCondition等條件進行過濾(有興趣可以深入瞭解);

8.廣播事件,得到AutoConfigurationImportListener所有實現類,然後生成事件進行廣播;

9.把需要裝配和排除的類完全限定名封裝成了AutoConfigurationEntry對象返回。

因此,@EnableAutoConfiguration可以簡單總結爲:從classpath中搜尋所有的META-INF/spring.factories配置文件,並將其中EnableAutoConfiguration對應的配置項通過反射實例化爲對應的標註了@Configuration的IoC容器配置類,並加載到IoC容器。

三、小結

通過以上分析可知@SpringBootApplication註解的運作是通過@SpringApplicationConfiguration聲明被標註類爲配置類,從而被AnnotationConfigApplicationContext掃描並初始化Spring容器。

通過@EnableAutoConfiguration來掃描,過濾並加載所需要的組件;通過@ComponentScan掃描並註冊所有標註了@Component及其子註解的類;這些註解的共同運作實現了springboot工程強大的自動配置能力。

以上就是本次總結的全部內容,希望能對大家有所幫助。如果有疏漏錯誤之處,還請大家不吝指正。

作者:vivo 互聯網開發團隊-Peng peng

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章