spring boot 自动配置原理

350 阅读5分钟
本文已参与「新人创作礼」活动,一起开启掘金创作之路。
spring boot 自动配置原理

自动配置原理

@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 {
	......
}

可以看到,@SpringBootApplication 是一个组合注解。除了最上面的几个元注解以以外,还有三个 Spring 的注解:

@SpringBootConfiguration,表示被注解的元素为一个 Spring Boot 配置类
@EnableAutoConfiguration,负责开启自动配置的注解,这一小节最靓的仔
@ComponentScan,用于配置扫描的包路径

关键点

我们重点关注 @EnableAutoConfiguration,我们继续深入源码:

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
@AutoConfigurationPackage
@Import(AutoConfigurationImportSelector.class)
public @interface EnableAutoConfiguration {
	......
}

@Import 注解和 AutoConfigurationImportSelector 类是我们需要特别关注的。自动配置的工作就是在 AutoConfigurationImportSelector 类中完成的。通过 getAutoConfigurationEntry 方法得到一个需要自动配置的列表:

protected AutoConfigurationEntry getAutoConfigurationEntry(
    AutoConfigurationMetadata autoConfigurationMetadata,
    AnnotationMetadata annotationMetadata) {
    if (!isEnabled(annotationMetadata)) {
        return EMPTY_ENTRY;
    }
    AnnotationAttributes attributes = getAttributes(annotationMetadata);
    // 得到一个包含 118 个元素的列表
    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);
}

getCandidateConfigurations 方法会获取到 Spring 预设的自动配置列表,共 118 个,这个列表的名单就放在 spring-boot-autoconfigure-x.x.x.RELEASE.jar 包中的 /META-INF/spring.factories 文件中。如果我们手动配置了需要关闭的自动配置,那么会通过 configurations.removeAll(exclusions) 将其从列表中移除。最后通过 filter 方法过滤掉不需要自动配置的项,最终会得到一个包含所有需要自动配置项的列表。

拆解

获取预设自动配置列表

AutoConfigurationImportSelector 类通过调用 SpringFactoriesLoader 的 loadFactoryNames() 方法来读取 spring.factories 文件中的 key(org.springframework.boot.autoconfigure.EnableAutoConfiguration)来加载 Spring 预设的自动配置列表。

读取 spring.factories 文件的源码示例:

public final class SpringFactoriesLoader {
    // spring.factories 文件路径
    public static final String FACTORIES_RESOURCE_LOCATION = "META-INF/spring.factories";
    ......
        
    public static List<String> loadFactoryNames(Class<?> factoryClass, @Nullable ClassLoader classLoader) {
		String factoryClassName = factoryClass.getName();
        // 调用真正读取 spring.factories 文件的方法
		return loadSpringFactories(classLoader).getOrDefault(factoryTypeName, Collections.emptyList());
	}

	private static Map<String, List<String>> loadSpringFactories(@Nullable ClassLoader classLoader) {
		......

		try {
            // 读取 spring.factories 文件
			Enumeration<URL> urls = (classLoader != null ?
					classLoader.getResources(FACTORIES_RESOURCE_LOCATION) :
					ClassLoader.getSystemResources(FACTORIES_RESOURCE_LOCATION));
			......
		}
		......
	}
    ......
}

处理 @Import

分析完 AutoConfigurationImportSelector 类,下面来分析 @Import 注解。正是因为这个注解,ImportSelector 才能处理自动配置的逻辑。@Import 的处理逻辑是由 ConfigurationClassParser 类完成的,入口是 doProcessConfigurationClass() 方法:

class ConfigurationClassParser {
    ......
    protected final SourceClass doProcessConfigurationClass(ConfigurationClass configClass, SourceClass sourceClass)
			throws IOException {

		......

		// 处理 @PropertySource 注解
		for (AnnotationAttributes propertySource : AnnotationConfigUtils.attributesForRepeatable(
				sourceClass.getMetadata(), PropertySources.class,
				org.springframework.context.annotation.PropertySource.class)) {
			......
		}

		// 处理 @ComponentScan 注解
		Set<AnnotationAttributes> componentScans = AnnotationConfigUtils.attributesForRepeatable(
				sourceClass.getMetadata(), ComponentScans.class, ComponentScan.class);
		......

		// 处理 @Import 注解
		processImports(configClass, sourceClass, getImports(sourceClass), filter, true);

		// 处理 @ImportResource 注解
		AnnotationAttributes importResource =
				AnnotationConfigUtils.attributesFor(sourceClass.getMetadata(), ImportResource.class);

		......
	}
    
    ......
        
	private void processImports(ConfigurationClass configClass, SourceClass currentSourceClass,
			Collection<SourceClass> importCandidates, Predicate<String> exclusionFilter,
			boolean checkForCircularImports) {
		......

		if (checkForCircularImports && isChainedImportOnStack(configClass)) {
			......
		}
		else {
			try {
				for (SourceClass candidate : importCandidates) {
                    // 处理 ImportSelector
					if (candidate.isAssignable(ImportSelector.class)) {
						Class<?> candidateClass = candidate.loadClass();
						ImportSelector selector = ParserStrategyUtils.instantiateClass(candidateClass, ImportSelector.class,
								this.environment, this.resourceLoader, this.registry);
                        Predicate<String> selectorFilter = selector.getExclusionFilter();
                        if (selectorFilter != null) {
                            exclusionFilter = exclusionFilter.or(selectorFilter);
                        }
						if (selector instanceof DeferredImportSelector) {
							this.deferredImportSelectorHandler.handle(configClass, (DeferredImportSelector) selector);
						}
						else {
							String[] importClassNames = selector.selectImports(currentSourceClass.getMetadata());
							Collection<SourceClass> importSourceClasses = asSourceClasses(importClassNames, exclusionFilter);
                            // 递归调用
							processImports(configClass, currentSourceClass, importSourceClasses, exclusionFilter, false);
						}
					}
					else if (candidate.isAssignable(ImportBeanDefinitionRegistrar.class)) {
						......
					}
					else {
						processConfigurationClass(candidate.asConfigClass(configClass), exclusionFilter);
					}
				}
			}
			......
		}
	}    
}

doProcessConfigurationClass() 方法调用 processImports() 方法来处理 @Import 注解。processImports() 方法中又分情况对 @Import 注解中的值执行了不同的处理逻辑。当值为 ImportSelector.class(AutoConfigurationImportSelector 是 ImportSelector 的实现类 ) 类型时,还会通过递归再次调用自己。

这段逻辑相对比较复杂,不仅有自我递归,还有多个方法间的 “串联递归”。上面代码中,进入最后一个 else 下面的 processConfigurationClass() 方法,你会看到它调用了最上面的 doProcessConfigurationClass() 方法,这样就形成了一个调用环: doProcessConfigurationClass() ->processImports()-> processConfigurationClass() -> doProcessConfigurationClass()。

调用链

整理了一份从 Spring 容器启动一直到自动配置功能完成的方法调用链,可以在看源码或者 Debug 的时候作为参考:

1. SpringApplication
   1. run()
   2. refreshContext()
   3. refresh()
2. AbstractApplicationContext
   1. refresh()
   2. invokeBeanFactoryPostProcessors()
3. PostProcessorRegistrationDelegate
   1. invokeBeanFactoryPostProcessors()
4. ConfigurationClassPostProcessor
   1. postProcessBeanDefinitionRegistry()
   2. processConfigBeanDefinitions()
5. ConfigurationClassParser
   1. parse()
6. ConfigurationClassParser.DeferredImportSelectorHandler
   1. process()
7. ConfigurationClassParser.DeferredImportSelectorGroupingHandler
   1. processGroupImports()
8. ConfigurationClassParser.DeferredImportSelectorGrouping
   1. getImports()
9. AutoConfigurationImportSelector.AutoConfigurationGroup
   1. process()
10. AutoConfigurationImportSelector
    1. getAutoConfigurationEntry()

第一层级为类,其中还有几个是内部类,用 “.” 和主类隔开了,第二层级为该类下的方法。从上到下按顺序依次调用。

在 2.1.0 之后的版本中 ConfigurationClassParser.processImports() 的代码如下:

if (selector instanceof DeferredImportSelector) {
    this.deferredImportSelectorHandler.handle(configClass, (DeferredImportSelector) selector);
}
else {
    String[] importClassNames = selector.selectImports(currentSourceClass.getMetadata());
    ......
}

而 AutoConfigurationImportSelector 就是 DeferredImportSelector 的实现类,所以根本不会走 else 中的逻辑。

自动配置的入口就是 refreshContext() 方法。

按需配置

Spring Boot 的自动配置再一次践行了约定优于配置的原则。它的自动配置并不是将所有预设列表全部加载进来,是非常智能的 “按需配置”。能做到这一点要归功于 @Conditional 注解和 Condition 接口。它们使得各种配置只有在符合一定的条件时才会被加载。

在讲自动配置原理的时候,我们了解到 AutoConfigurationImportSelector.getAutoConfigurationEntry() 方法中 configurations = filter (configurations, autoConfigurationMetadata) 就是用来过滤那些不符合条件配置的。

以 DataSourceAutoConfiguration 为例来具体分析一下,先来看一下源码:

@Configuration(proxyBeanMethods = false)
@ConditionalOnClass({ DataSource.class, EmbeddedDatabaseType.class })
@EnableConfigurationProperties(DataSourceProperties.class)
@Import({ DataSourcePoolMetadataProvidersConfiguration.class, DataSourceInitializationConfiguration.class })
public class DataSourceAutoConfiguration {
    ......
}

DataSourceAutoConfiguration 通过 @ConditionalOnClass({ DataSource.class, EmbeddedDatabaseType.class }) 告诉 Spring,只有当 classpath 下存在 DataSource.class 和 EmbeddedDatabaseType.class 时,DataSourceAutoConfiguration 才会被加载。

@ConditionalOnClass 是 @Conditional 的衍生注解,由 @Conditional 和 OnClassCondition 类组成,源码如下:

@Target({ ElementType.TYPE, ElementType.METHOD })
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Conditional(OnClassCondition.class)
public @interface ConditionalOnClass {
	......
}

OnClassCondition 是一个实现了 Condition 接口的类,@ConditionalOnClass 表示 classpath 里有指定的类时加载配置。它是 @Conditional 众多衍生注解中的一个,Spring Boot 提供了一些基于 @Conditional 的衍生注解:

注解说明
@ConditionalOnBean当容器里有指定 Bean 时
@ConditionalOnMissingBean当容器里没有指定 Bean 时
@ConditionalOnClassclasspath 里有指定的类时
@ConditionalOnMissingClassclasspath 里没有指定的类时
@ConditionalOnExpression给定的 Spring Expression Language(SpEL)表达式计算结果为 true 时
@ConditionalOnJavaJVM 的版本匹配特定值或者一个范围时
@ConditionalOnJndi参数中给定的 JNDI 位置至少存在一个时(如果没有给参数,则要有 JNDI InitialContext)
@ConditionalOnProperty指定的属性为指定的值时
@ConditionalOnResourceclasspath 里有指定的资源时
@ConditionalOnWebApplication当前应用是 Web 应用时
@ConditionalOnNotWebApplication当前应用不是 Web 应用时

这些注解都是基于 @Conditional,可以覆盖到我们大多数的使用场景,如果以上情况不能满足你的需求,还可以通过自己实现 Condition 接口来完成自定义的需求。