1 前言

上一篇,我们分析了 SpringBoot 源码结构及各个模块 pom 之间的关系后,那么此篇开始就开始解开 SpringBoot 新特性之一—自动配置的神秘面纱了。因为 SpringBoot 自动配置原理是基于其大量的条件注解ConditionalOnXXX,因此,本节我们先来撸下 Spring 的条件注解的相关源码。

2 SpringBoot 的派生条件注解

我们都知道,SpringBoot 自动配置是需要满足相应的条件才会自动配置,因此 SpringBoot 的自动配置大量应用了条件注解ConditionalOnXXX。如下图:

3 SpringBoot自动配置的条件注解原理 - 图1

那么上图的条件注解如何使用呢?

举个栗子,我们来看下如何使用@ConditionalOnClass@ConditionalOnProperty这两个注解,先看下图代码:

3 SpringBoot自动配置的条件注解原理 - 图2 HelloWorldEnableAutoConfiguration这个自动配置类应用了@ConditionalOnClassConditionalOnProperty两个条件注解,那么只有在满足:classpath中存在HelloWorldComponent.class和配置了hello.world.namehello.world.age属性这两个条件的情况下才会创建HelloWorldComponent这个bean

其实 SpringBoot 的@ConditionalOnXXX等条件注解都是派生注解,那么什么是派生注解呢? 就拿上面的栗子来说,以@ConditionalOnClass(HelloWorldComponent.class)为例,我们打开ConditionalOnClass注解源码,如下:

  1. @Target({
  2. ElementType.TYPE,
  3. ElementType.METHOD
  4. })
  5. @Retention(RetentionPolicy.RUNTIME)
  6. @Documented
  7. @Conditional(OnClassCondition.class)
  8. public @interface ConditionalOnClass {
  9. Class << ? > [] value() default {};
  10. String[] name() default {};
  11. }

可以看到@ConditionalOnClass注解上面又标注了@Conditional(OnClassCondition.class)注解,因此@ConditionalOnClass@Conditional的派生注解,@Conditional(OnClassCondition.class)@ConditionalOnClass注解是等价的,即这两个注解标注在某个配置类上的效果是等价的。

而 SpringBoot 的自动配置原理正是建立在这些大量的派生条件注解@ConditionalOnXXX之上,而这些条件注解的原理跟 Spring 的 Condition 接口有关。因此我们先来研究下 Condition 接口的相关源码。

3 Condition 接口

3.1 Condition 接口源码分析

分析 Condition 接口源码前先看下如何自定义ConditionalOnXXX注解,举个栗子,比如自定义一个@ConditionalOnLinux注解,该注解只有在其属性environment是”linux”才会创建相关的 bean。定义了以下代码:

  1. /**
  2. * 实现spring 的Condition接口,并且重写matches()方法,如果@ConditionalOnLinux的注解属性environment是linux就返回true
  3. *
  4. */
  5. public class LinuxCondition implements Condition {
  6. @Override
  7. public boolean matches(ConditionContext context, AnnotatedTypeMetadata metadata) {
  8. // 获得注解@ConditionalOnLinux的所有属性
  9. List < AnnotationAttributes > allAnnotationAttributes = annotationAttributesFromMultiValueMap(
  10. metadata.getAllAnnotationAttributes(
  11. ConditionalOnLinux.class.getName()));
  12. for (AnnotationAttributes annotationAttributes: allAnnotationAttributes) {
  13. // 获得注解@ConditionalOnLinux的environment属性
  14. String environment = annotationAttributes.getString("environment");
  15. // 若environment等于linux,则返回true
  16. if ("linux".equals(environment)) {
  17. return true;
  18. }
  19. }
  20. return false;
  21. }
  22. }
  23. @Target({
  24. ElementType.TYPE,
  25. ElementType.METHOD
  26. })
  27. @Retention(RetentionPolicy.RUNTIME)
  28. @Documented
  29. @Conditional(LinuxCondition.class)
  30. public @interface ConditionalOnLinux {
  31. // 标注是哪个环境
  32. String environment() default "";
  33. }
  34. @Configuration
  35. public class ConditionConfig {
  36. // 只有`@ConditionalOnLinux`的注解属性`environment`是"linux"时才会创建bean
  37. @Bean
  38. @ConditionalOnLinux(environment = "linux")
  39. public Environment linuxEnvironment() {
  40. return new LinuxEnvironment();
  41. }
  42. }

上面的代码我们捋一下:

  1. LinuxCondition实现了Condition接口并实现了matches方法,而matches方法则判断@ConditionalOnLinux的注解属性environment是否”linux”,是则返回 true,否则 false。
  2. 然后我们再定义了一个注解@ConditionalOnLinux,这个注解是@Conditional的派生注解,与@Conditional(LinuxCondition.class)等价,注意@ConditionalOnLinux注解定义了一个属性environment。而我们最终可以利用LinuxConditionmatches方法中的参数AnnotatedTypeMetadata来获取@ConditionalOnLinux的注解属性environment的值,从而用来判断值是否为 linux”。
  3. 最后我们又定义了一个配置类ConditionConfig,在linuxEnvironment方法上标注了@ConditionalOnLinux(environment = "linux")。因此,这里只有 LinuxConditionmatches方法返回 true 才会创建bean

学会了如何自定义@ConditionalOnXXX注解后,我们现在再来看下Condition接口的源码:

  1. @FunctionalInterface
  2. public interface Condition {
  3. boolean matches(ConditionContext context, AnnotatedTypeMetadata metadata);
  4. }

Condition 接口主要有一个matches方法,该方法决定了是否要注册相应的bean对象。其中matches方法中有两个参数,参数类型分别是ConditionContextAnnotatedTypeMetadata,这两个参数非常重要。它们分别用来获取一些环境信息和注解元数据从而用在matches方法中判断是否符合条件。

ConditionContext,顾名思义,主要是跟Condition的上下文有关,主要用来获取Registry,BeanFactory,Environment,ResourceLoaderClassLoader等。那么获取这些用来干什么呢?举个栗子,比如OnResourceCondition需要靠ConditionContext来获取ResourceLoader来加载指定资源,OnClassCondition需要靠ConditionContext来获取ClassLoader来加载指定类等,下面看下其源码:

  1. public interface ConditionContext {
  2. BeanDefinitionRegistry getRegistry();
  3. @Nullable
  4. ConfigurableListableBeanFactory getBeanFactory();
  5. Environment getEnvironment();
  6. ResourceLoader getResourceLoader();
  7. @Nullable
  8. ClassLoader getClassLoader();
  9. }

AnnotatedTypeMetadata,这个跟注解元数据有关,利用AnnotatedTypeMetadata可以拿到某个注解的一些元数据,而这些元数据就包含了某个注解里面的属性,比如前面的栗子,利用AnnotatedTypeMetadata可以拿到@ConditionalOnLinux的注解属性environment的值。下面看下其源码:

  1. public interface AnnotatedTypeMetadata {
  2. boolean isAnnotated(String annotationName);
  3. Map<String, Object> getAnnotationAttributes(String annotationName);
  4. Map<String, Object> getAnnotationAttributes(String annotationName, boolean classValuesAsString);
  5. MultiValueMap<String, Object> getAllAnnotationAttributes(String annotationName);
  6. MultiValueMap<String, Object> getAllAnnotationAttributes(String annotationName, boolean classValuesAsString);
  7. }

回到刚才的栗子,我们知道@ConditionalOnLinux注解真正起作用的是Condition接口的具体实现类LinuxConditionmatches方法,那么这个matches方法是在何时被调用的呢?

通过 idea 调试看调用的栈帧,如下图:

3 SpringBoot自动配置的条件注解原理 - 图3

发现是在ConditionEvaluatorshouldSkip方法中调用了LinuxConditionmatches方法,自然我们再去看看ConditionEvaluatorshouldSkip的方法执行了什么逻辑。

  1. // 这个方法主要是如果是解析阶段则跳过,如果是注册阶段则不跳过
  2. public boolean shouldSkip(@Nullable AnnotatedTypeMetadata metadata, @Nullable ConfigurationPhase phase) {
  3. // 若没有被@Conditional或其派生注解所标注,则不会跳过
  4. if (metadata == null || !metadata.isAnnotated(Conditional.class.getName())) {
  5. return false;
  6. }
  7. // 没有指定phase,注意phase可以分为PARSE_CONFIGURATION或REGISTER_BEAN类型
  8. if (phase == null) {
  9. // 若标有@Component,@Import,@Bean或@Configuration等注解的话,则说明是PARSE_CONFIGURATION类型
  10. if (metadata instanceof AnnotationMetadata &&
  11. ConfigurationClassUtils.isConfigurationCandidate((AnnotationMetadata) metadata)) {
  12. return shouldSkip(metadata, ConfigurationPhase.PARSE_CONFIGURATION);
  13. }
  14. // 否则是REGISTER_BEAN类型
  15. return shouldSkip(metadata, ConfigurationPhase.REGISTER_BEAN);
  16. }
  17. List<Condition> conditions = new ArrayList<>();
  18. // TODO 获得所有标有@Conditional注解或其派生注解里面的Condition接口实现类并实例化成对象。
  19. // 比如@Conditional(OnBeanCondition.class)则获得OnBeanCondition.class,OnBeanCondition.class往往实现了Condition接口
  20. for (String[] conditionClasses : getConditionClasses(metadata)) {
  21. // 将类实例化成对象
  22. for (String conditionClass : conditionClasses) {
  23. Condition condition = getCondition(conditionClass, this.context.getClassLoader());
  24. conditions.add(condition);
  25. }
  26. }
  27. // 排序,即按照Condition的优先级进行排序
  28. AnnotationAwareOrderComparator.sort(conditions);
  29. for (Condition condition : conditions) {
  30. ConfigurationPhase requiredPhase = null;
  31. if (condition instanceof ConfigurationCondition) {
  32. // 从condition中获得对bean是解析还是注册
  33. requiredPhase = ((ConfigurationCondition) condition).getConfigurationPhase();
  34. }
  35. // 若requiredPhase为null或获取的阶段类型正是当前阶段类型且不符合condition的matches条件,则跳过
  36. if ((requiredPhase == null || requiredPhase == phase) && !condition.matches(this.context, metadata)) {
  37. return true;
  38. }
  39. }
  40. return false;
  41. }

shouldSkip这个方法执行的逻辑主要是如果是解析阶段则跳过,如果是注册阶段则不跳过;如果是在注册阶段即REGISTER_BEAN阶段的话,此时会得到所有的Condition接口的具体实现类并实例化这些实现类,然后再执行下面关键的代码进行判断是否需要跳过。

  1. if ((requiredPhase == null || requiredPhase == phase) && !condition.matches(this.context, metadata)) {
  2. return true;
  3. }

上面代码最重要的逻辑是调用了Condition接口的具体实现类的matches方法,若matches返回false,则跳过,不进行注册bean的操作;若matches返回true,则不跳过,进行注册bean的操作;

好了,Condition的源码分析就到此为止,再往上翻调用方法的话应该就是 Spring 加载bean定义的相关源码了,不属于这里的分析范围。

3.2 Spring 的内置 Condition 接口实现类

前面我们学会了如何自定义条件注解及Condition的源码分析,那么我们不禁好奇,Spring 究竟内置了哪些Condition接口的实现类呢?

那么看下 Spring 的Condition接口的具体实现类的类图:

3 SpringBoot自动配置的条件注解原理 - 图4

发现 Spring 内置的Condition接口的具体实现类虽然有多个,但只有ProfileCondition不是测试相关的,因此可以说真正的内置的Condition接口的具体实现类只有ProfileCondition一个,非常非常少,这跟 SpringBoot 的大量派生条件注解形成了鲜明的对比。ProfileCondition大家都知道,是跟环境有关,比如我们平时一般有dev,testprod环境,而ProfileCondition就是判断我们项目配置了哪个环境的。下面是ProfileCondition的源码,很简单,这里就不分析了。

  1. class ProfileCondition implements Condition {
  2. @Override
  3. public boolean matches(ConditionContext context, AnnotatedTypeMetadata metadata) {
  4. MultiValueMap<String, Object> attrs = metadata.getAllAnnotationAttributes(Profile.class.getName());
  5. if (attrs != null) {
  6. for (Object value : attrs.get("value")) {
  7. if (context.getEnvironment().acceptsProfiles(Profiles.of((String[]) value))) {
  8. return true;
  9. }
  10. }
  11. return false;
  12. }
  13. return true;
  14. }
  15. }

4 SpringBootCondition 源码解析

前面看到 Spring 对Condition的内置注解可以说只有ProfileCondition一个,但是我们都知道,SpringBoot 则内置了大量的条件注解ConditionalOnXXX。在分析前,我们先来看一下SpringBootCondition的整体类图来个整体的理解,如下图:

3 SpringBoot自动配置的条件注解原理 - 图5

可以看到SpringBootCondition作为 SpringBoot 条件注解的基类,处于整个类图的中心,它实现了Condition接口,然后又有很多具体的子类OnXXXCondition,这些OnXXXCondition其实就是@ConditionalOnXXX的条件类。

我们先来看下SpringBootCondition这个父类是主要做了哪些事情,抽象了哪些共有的逻辑?

SpringBootConditon实现了Condition接口,作为 SpringBoot 众多条件注解OnXXXCondtion的父类,它的作用主要就是打印一些条件注解评估报告的日志,比如打印哪些配置类是符合条件注解的,哪些是不符合的。打印的日志形式如下图:

3 SpringBoot自动配置的条件注解原理 - 图6

3 SpringBoot自动配置的条件注解原理 - 图7 因为SpringBootConditon实现了Condition接口,也实现了matches方法,因此该方法同样也是被ConditionEvaluatorshouldSkip方法中调用,因此我们就以SpringBootConditonmatches方法为入口去进行分析。直接上代码:

  1. // SpringBootCondition.java
  2. public final boolean matches(ConditionContext context,
  3. AnnotatedTypeMetadata metadata) {
  4. // 得到metadata的类名或方法名
  5. String classOrMethodName = getClassOrMethodName(metadata);
  6. try {
  7. // 判断每个配置类的每个条件注解@ConditionalOnXXX是否满足条件,然后记录到ConditionOutcome结果中
  8. // 注意getMatchOutcome是一个抽象模板方法,交给OnXXXCondition子类去实现
  9. ConditionOutcome outcome = getMatchOutcome(context, metadata);
  10. // 打印condition评估的日志,哪些条件注解@ConditionalOnXXX是满足条件的,哪些是不满足条件的,这些日志都打印出来
  11. logOutcome(classOrMethodName, outcome);
  12. // 除了打印日志外,这些是否匹配的信息还要记录到ConditionEvaluationReport中
  13. recordEvaluation(context, classOrMethodName, outcome);
  14. // 最后返回@ConditionalOnXXX是否满足条件
  15. return outcome.isMatch();
  16. }
  17. catch (NoClassDefFoundError ex) {
  18. throw new IllegalStateException(
  19. "Could not evaluate condition on " + classOrMethodName + " due to "
  20. + ex.getMessage() + " not "
  21. + "found. Make sure your own configuration does not rely on "
  22. + "that class. This can also happen if you are "
  23. + "@ComponentScanning a springframework package (e.g. if you "
  24. + "put a @ComponentScan in the default package by mistake)",
  25. ex);
  26. }
  27. catch (RuntimeException ex) {
  28. throw new IllegalStateException(
  29. "Error processing condition on " + getName(metadata), ex);
  30. }
  31. }

上面代码的注释已经非常详细,我们知道了SpringBootCondition抽象了所有其具体实现类OnXXXCondition的共有逻辑—condition评估信息打印,最重要的是封装了一个模板方法getMatchOutcome(context, metadata),留给各个OnXXXCondition具体子类去覆盖实现属于自己的判断逻辑,然后再返回相应的匹配结果给SpringBootCondition用于日志打印。

因此我们知道了SpringBootCondition其实就是用来打印condition评估信息的,对于其他枝节方法我们不必追究过深,免得丢了主线。我们现在的重点是放在交给OnXXXCondition子类实现的模板方法上getMatchOutcome(context, metadata);,因为这个方法将会由很多OnXXXCondition覆盖重写判断逻辑,这里是我们接下来分析的重点。

因为SpringBootCondition有众多具体实现类,下面只挑OnResourceConditionOnBeanConditionOnWebApplicationCondition进行讲解,而AutoConfigurationImportFilter跟自动配置有关,则留到自动配置源码解析的时候再进行分析。

4.1 OnResourceCondition 源码分析

现在先来看下一个逻辑及其简单的注解条件类OnResourceConditionOnResourceCondition继承了SpringBootCondition父类,覆盖了其getMatchOutcome方法,用于@ConditionalOnResource注解指定的资源存在与否。OnResourceCondition的判断逻辑非常简单,主要拿到@ConditionalOnResource注解指定的资源路径后,然后用ResourceLoader根据指定路径去加载看资源存不存在。下面直接看代码:

先来看下@ConditionalOnResource的代码,

  1. @Target({ ElementType.TYPE, ElementType.METHOD })
  2. @Retention(RetentionPolicy.RUNTIME)
  3. @Documented
  4. @Conditional(OnResourceCondition.class)
  5. public @interface ConditionalOnResource {
  6. /**
  7. * The resources that must be present.
  8. * @return the resource paths that must be present.
  9. */
  10. String[] resources() default {};
  11. }

再来看OnResourceCondition的代码:

  1. @Order(Ordered.HIGHEST_PRECEDENCE + 20)
  2. class OnResourceCondition extends SpringBootCondition {
  3. private final ResourceLoader defaultResourceLoader = new DefaultResourceLoader();
  4. @Override
  5. public ConditionOutcome getMatchOutcome(ConditionContext context,
  6. AnnotatedTypeMetadata metadata) {
  7. // 获得@ConditionalOnResource注解的属性元数据
  8. MultiValueMap<String, Object> attributes = metadata
  9. .getAllAnnotationAttributes(ConditionalOnResource.class.getName(), true);
  10. // 获得资源加载器,若ConditionContext中有ResourceLoader则用ConditionContext中的,没有则用默认的
  11. ResourceLoader loader = (context.getResourceLoader() != null)
  12. ? context.getResourceLoader() : this.defaultResourceLoader;
  13. List<String> locations = new ArrayList<>();
  14. // 将@ConditionalOnResource中定义的resources属性值取出来装进locations集合
  15. collectValues(locations, attributes.get("resources"));
  16. Assert.isTrue(!locations.isEmpty(),
  17. "@ConditionalOnResource annotations must specify at "
  18. + "least one resource location");
  19. // missing集合是装不存在指定资源的资源路径的
  20. List<String> missing = new ArrayList<>();
  21. // 遍历所有的资源路径,若指定的路径的资源不存在则将其资源路径存进missing集合中
  22. for (String location : locations) {
  23. // 这里针对有些资源路径是Placeholders的情况,即处理 ${}
  24. String resource = context.getEnvironment().resolvePlaceholders(location);
  25. if (!loader.getResource(resource).exists()) {
  26. missing.add(location);
  27. }
  28. }
  29. // 如果存在某个资源不存在,那么则报错
  30. if (!missing.isEmpty()) {
  31. return ConditionOutcome.noMatch(ConditionMessage
  32. .forCondition(ConditionalOnResource.class)
  33. .didNotFind("resource", "resources").items(Style.QUOTE, missing));
  34. }
  35. // 所有资源都存在,那么则返回能找到就提的资源
  36. return ConditionOutcome
  37. .match(ConditionMessage.forCondition(ConditionalOnResource.class)
  38. .found("location", "locations").items(locations));
  39. }
  40. // 将@ConditionalOnResource中定义的resources属性值取出来装进locations集合
  41. private void collectValues(List<String> names, List<Object> values) {
  42. for (Object value : values) {
  43. for (Object item : (Object[]) value) {
  44. names.add((String) item);
  45. }
  46. }
  47. }
  48. }

可以看到OnResourceConditiongetMatchOutcome方法非常简单,这里不再详述。

4.2 OnBeanCondition 源码分析

OnBeanCondition同样继承了FilteringSpringBootCondition父类,覆盖了父类FilteringSpringBootConditiongetOutcomes方法。而FilteringSpringBootCondition又是SpringBootCondition的子类,FilteringSpringBootCondition跟自动配置类过滤有关,这里先不分析。值得注意的是**OnBeanCondition**同样重写了**SpringBootCondition****getMatchOutcome**方法,用来判断 Spring 容器中是否存在指定条件的bean。同时是OnBeanCondition@ConditionalOnBean,@ConditionalOnSingleCandidateConditionalOnMissingBean的条件类。

同样,先来看OnBeanCondition复写父类SpringBootConditiongetMatchOutcome方法的代码:

  1. @Override
  2. public ConditionOutcome getMatchOutcome(ConditionContext context,
  3. AnnotatedTypeMetadata metadata) {
  4. ConditionMessage matchMessage = ConditionMessage.empty();
  5. // (1),配置类(metadata)标注@ConditionalOnBean注解的情况
  6. if (metadata.isAnnotated(ConditionalOnBean.class.getName())) {
  7. // 将@ConditionalOnBean注解属性封装进BeanSearchSpec对象中
  8. // 注意BeanSearchSpec是一个静态内部类,用来存储@ConditionalOnBean和@ConditionalOnMissingBean注解的属性值
  9. BeanSearchSpec spec = new BeanSearchSpec(context, metadata,
  10. ConditionalOnBean.class);
  11. // 调用getMatchingBeans得到符合条件的bean
  12. MatchResult matchResult = getMatchingBeans(context, spec);
  13. // 如果不匹配
  14. if (!matchResult.isAllMatched()) {
  15. String reason = createOnBeanNoMatchReason(matchResult);
  16. return ConditionOutcome.noMatch(ConditionMessage
  17. .forCondition(ConditionalOnBean.class, spec).because(reason));
  18. }
  19. // 如果匹配
  20. matchMessage = matchMessage.andCondition(ConditionalOnBean.class, spec)
  21. .found("bean", "beans")
  22. .items(Style.QUOTE, matchResult.getNamesOfAllMatches());
  23. }
  24. // (2),配置类(metadata)标注@ConditionalOnSingleCandidate注解的情况
  25. if (metadata.isAnnotated(ConditionalOnSingleCandidate.class.getName())) {
  26. BeanSearchSpec spec = new SingleCandidateBeanSearchSpec(context, metadata,
  27. ConditionalOnSingleCandidate.class);
  28. MatchResult matchResult = getMatchingBeans(context, spec);
  29. if (!matchResult.isAllMatched()) {
  30. return ConditionOutcome.noMatch(ConditionMessage
  31. .forCondition(ConditionalOnSingleCandidate.class, spec)
  32. .didNotFind("any beans").atAll());
  33. }
  34. else if (!hasSingleAutowireCandidate(context.getBeanFactory(),
  35. matchResult.getNamesOfAllMatches(),
  36. spec.getStrategy() == SearchStrategy.ALL)) {
  37. return ConditionOutcome.noMatch(ConditionMessage
  38. .forCondition(ConditionalOnSingleCandidate.class, spec)
  39. .didNotFind("a primary bean from beans")
  40. .items(Style.QUOTE, matchResult.getNamesOfAllMatches()));
  41. }
  42. matchMessage = matchMessage
  43. .andCondition(ConditionalOnSingleCandidate.class, spec)
  44. .found("a primary bean from beans")
  45. .items(Style.QUOTE, matchResult.getNamesOfAllMatches());
  46. }
  47. // (3),配置类(metadata)标注@ConditionalOnMissingBean注解的情况
  48. if (metadata.isAnnotated(ConditionalOnMissingBean.class.getName())) {
  49. BeanSearchSpec spec = new BeanSearchSpec(context, metadata,
  50. ConditionalOnMissingBean.class);
  51. MatchResult matchResult = getMatchingBeans(context, spec);
  52. if (matchResult.isAnyMatched()) {
  53. String reason = createOnMissingBeanNoMatchReason(matchResult);
  54. return ConditionOutcome.noMatch(ConditionMessage
  55. .forCondition(ConditionalOnMissingBean.class, spec)
  56. .because(reason));
  57. }
  58. matchMessage = matchMessage.andCondition(ConditionalOnMissingBean.class, spec)
  59. .didNotFind("any beans").atAll();
  60. }
  61. // 最终返回matchMessage
  62. return ConditionOutcome.match(matchMessage);
  63. }

我们可以看到OnBeanCondition类覆盖的getMatchOutcome方法分别处理了标注@ConditionalOnBean,@ConditionalOnSingleCandidate@ConditionalOnMissingBean注解的情况,分别对应上面代码注释的(1),(2)(3)处。

现在我们只看针对@ConditionalOnBean注解的处理逻辑,从上面代码中可以看到若配置类(metadata)标注@ConditionalOnBean注解的话,主要做了以下事情:

  1. 将该注解属性提取出来封装进BeanSearchSpec对象中;
  2. 然后调用getMatchingBeans(context, spec)方法来获取是否有匹配的bean
  3. 最后返回bean的匹配情况;

可以看到最重要的逻辑是第 2 步,那么我们再来看下getMatchingBeans方法,直接上代码:

  1. protected final MatchResult getMatchingBeans(ConditionContext context,
  2. BeanSearchSpec beans) {
  3. // 获得Spring容器的beanFactory
  4. ConfigurableListableBeanFactory beanFactory = context.getBeanFactory();
  5. // 判断bean的搜索策略是否是SearchStrategy.ANCESTORS策略
  6. if (beans.getStrategy() == SearchStrategy.ANCESTORS) {
  7. BeanFactory parent = beanFactory.getParentBeanFactory();
  8. Assert.isInstanceOf(ConfigurableListableBeanFactory.class, parent,
  9. "Unable to use SearchStrategy.PARENTS");
  10. beanFactory = (ConfigurableListableBeanFactory) parent;
  11. }
  12. // MatchResult用来存储bean的匹配结果
  13. MatchResult matchResult = new MatchResult();
  14. // 如果bean的搜索策略不是SearchStrategy.CURRENT的话,则置considerHierarchy为true
  15. boolean considerHierarchy = beans.getStrategy() != SearchStrategy.CURRENT;
  16. // 获取TypeExtractor,TypeExtractor是用来判断bean的类型的
  17. TypeExtractor typeExtractor = beans.getTypeExtractor(context.getClassLoader());
  18. // 获取是否有被忽略bean类型,若有的话将该bean类型的名称装进beansIgnoredByType集合
  19. // 这里主要是针对@ConditionalOnMissingBean的ignored属性
  20. List<String> beansIgnoredByType = getNamesOfBeansIgnoredByType(
  21. beans.getIgnoredTypes(), typeExtractor, beanFactory, context,
  22. considerHierarchy);
  23. // 遍历bean的所有类型
  24. for (String type : beans.getTypes()) {
  25. // 调用getBeanNamesForType方法根据bean类型得到所有符合条件的bean类型,并放到typeMatches集合
  26. Collection<String> typeMatches = getBeanNamesForType(beanFactory, type,
  27. typeExtractor, context.getClassLoader(), considerHierarchy);
  28. // 移除掉Ignored的类型
  29. typeMatches.removeAll(beansIgnoredByType);
  30. // 若typeMatches为空,那么则说明正在遍历的这个type类型不符合匹配条件,此时用matchResult记录一下这个不符合条件的类型
  31. if (typeMatches.isEmpty()) {
  32. matchResult.recordUnmatchedType(type);
  33. }
  34. // 若typeMatches不为空,那么则说明正在遍历的这个type类型符合匹配条件,此时用matchResult记录一下这个符合条件的类型
  35. else {
  36. matchResult.recordMatchedType(type, typeMatches);
  37. }
  38. }
  39. // 这里针对@ConditionalOnBean等注解的annotation属性的处理
  40. for (String annotation : beans.getAnnotations()) {
  41. List<String> annotationMatches = Arrays
  42. .asList(getBeanNamesForAnnotation(beanFactory, annotation,
  43. context.getClassLoader(), considerHierarchy));
  44. annotationMatches.removeAll(beansIgnoredByType);
  45. if (annotationMatches.isEmpty()) {
  46. matchResult.recordUnmatchedAnnotation(annotation);
  47. }
  48. else {
  49. matchResult.recordMatchedAnnotation(annotation, annotationMatches);
  50. }
  51. }
  52. // 这里针对@ConditionalOnBean等注解的name属性的处理
  53. for (String beanName : beans.getNames()) {
  54. // beansIgnoredByType集合不包含beanName且beanFactory包含这个bean,则匹配
  55. if (!beansIgnoredByType.contains(beanName)
  56. && containsBean(beanFactory, beanName, considerHierarchy)) {
  57. matchResult.recordMatchedName(beanName);
  58. }
  59. // 否则,不匹配
  60. else {
  61. matchResult.recordUnmatchedName(beanName);
  62. }
  63. }
  64. // 最后返回匹配结果
  65. return matchResult;
  66. }

上面的逻辑主要是从 spring 容器中搜索有无指定条件的bean,搜索 Spring 容器搜索 bean 的话有三种搜索策略,分别是CURRENT,ANCESTORSALL,分表表示只从当前的context中搜索bean,只从父context中搜索bean和从整个context中搜索bean;定义了搜索策略后,然后再根据BeanSearchSpec对象封装的注解属性分别取指定的容器中查找有无符合条件的bean,然后再进行一些过滤。比如@ConditionalOnMissingBean注解有定义ignored属性值,那么从容器中搜索到有符合条件的bean时,此时还要移除掉ignored指定的bean

好了,上面就已经分析了OnBeanCondition这个条件类了,我们坚持主线优先的原则,具体的细节代码不会深究。

4.3 OnWebApplicationCondition

OnWebApplicationCondition同样继承了FilteringSpringBootCondition父类,覆盖了父类FilteringSpringBootConditiongetOutcomes方法。而FilteringSpringBootCondition又是SpringBootCondition的子类,FilteringSpringBootCondition跟自动配置类过滤有关,这里先不分析。值得注意的是**OnWebApplicationCondition**同样重写了**SpringBootCondition****getMatchOutcome**方法,用来判断当前应用是否 web 应用。同时是OnWebApplicationCondition@ConditionalOnWebApplication的条件类。

同样,先来看OnWebApplicationCondition重写SpringBootConditiongetMatchOutcome方法:

  1. public ConditionOutcome getMatchOutcome(ConditionContext context,
  2. AnnotatedTypeMetadata metadata) {
  3. // 配置类是否标注有@ConditionalOnWebApplication注解
  4. boolean required = metadata
  5. .isAnnotated(ConditionalOnWebApplication.class.getName());
  6. // 调用isWebApplication方法返回匹配结果
  7. ConditionOutcome outcome = isWebApplication(context, metadata, required);
  8. // 若有标注@ConditionalOnWebApplication但不符合条件,则返回不匹配
  9. if (required && !outcome.isMatch()) {
  10. return ConditionOutcome.noMatch(outcome.getConditionMessage());
  11. }
  12. // 若没有标注@ConditionalOnWebApplication但符合条件,则返回不匹配
  13. if (!required && outcome.isMatch()) {
  14. return ConditionOutcome.noMatch(outcome.getConditionMessage());
  15. }
  16. // 这里返回匹配的情况,TODO 不过有个疑问:如果没有标注@ConditionalOnWebApplication注解,又不符合条件的话,也会执行到这里,返回匹配?
  17. return ConditionOutcome.match(outcome.getConditionMessage());
  18. }

上面代码的逻辑很简单,主要是调用isWebApplication方法来判断当前应用是否是 web 应用。因此,我们再来看下isWebApplication方法:

  1. private ConditionOutcome isWebApplication(ConditionContext context,
  2. AnnotatedTypeMetadata metadata, boolean required) {
  3. // 调用deduceType方法判断是哪种类型,其中有SERVLET,REACTIVE和ANY类型,其中ANY表示了SERVLET或REACTIVE类型
  4. switch (deduceType(metadata)) {
  5. // SERVLET类型
  6. case SERVLET:
  7. return isServletWebApplication(context);
  8. // REACTIVE类型
  9. case REACTIVE:
  10. return isReactiveWebApplication(context);
  11. default:
  12. return isAnyWebApplication(context, required);
  13. }
  14. }

isWebApplication方法中,首先从@ConditionalOnWebApplication注解中获取其定义了什么类型,然后根据不同的类型进入不同的判断逻辑。这里我们只看下SERVLET的情况判断处理,看代码:

  1. private ConditionOutcome isServletWebApplication(ConditionContext context) {
  2. ConditionMessage.Builder message = ConditionMessage.forCondition("");
  3. // 若classpath中不存在org.springframework.web.context.support.GenericWebApplicationContext.class,则返回不匹配
  4. if (!ClassNameFilter.isPresent(SERVLET_WEB_APPLICATION_CLASS,
  5. context.getClassLoader())) {
  6. return ConditionOutcome.noMatch(
  7. message.didNotFind("servlet web application classes").atAll());
  8. }
  9. // 若classpath中存在org.springframework.web.context.support.GenericWebApplicationContext.class,那么又分为以下几种匹配的情况
  10. // session
  11. if (context.getBeanFactory() != null) {
  12. String[] scopes = context.getBeanFactory().getRegisteredScopeNames();
  13. if (ObjectUtils.containsElement(scopes, "session")) {
  14. return ConditionOutcome.match(message.foundExactly("'session' scope"));
  15. }
  16. }
  17. // ConfigurableWebEnvironment
  18. if (context.getEnvironment() instanceof ConfigurableWebEnvironment) {
  19. return ConditionOutcome
  20. .match(message.foundExactly("ConfigurableWebEnvironment"));
  21. }
  22. // WebApplicationContext
  23. if (context.getResourceLoader() instanceof WebApplicationContext) {
  24. return ConditionOutcome.match(message.foundExactly("WebApplicationContext"));
  25. }
  26. // 若以上三种都不匹配的话,则说明不是一个servlet web application
  27. return ConditionOutcome.noMatch(message.because("not a servlet web application"));
  28. }

对于是SERVLET的情况,首先根据classpath中是否存在org.springframework.web.context.support.GenericWebApplicationContext.class,如果不存在该类,则直接返回不匹配;若存在的话那么又分为以下几种匹配的情况:

  • session
  • ConfigurableWebEnvironment
  • WebApplicationContext

若上面三种情况都不匹配,则说明不是一个 servlet web application。

4.4 其他

由于 springboot 的OnXXXCondition类实现太多,不可能每个条件类都分析一遍,因此上面只分析了OnResourceCondition,OnBeanConditiononWebApplicationCondition的源码。我们分析源码不可能把所有代码都通读一遍的,阅读源码的话,只要理解了某个模块的类之间的关系及挑几个有代表性的类分析下就行,不可能一网打尽。

若有时间的话,推荐看下几个我们常用的条件类的源码:OnPropertyCondition,OnClassConditionOnExpressionCondition等。

5 如何扩展 SpringBootCondition

前文我们知道了如何扩展 Spring 的Condition接口,那么我们该如何扩展 SpringBoot 的SpringBootCondition类呢?

推荐阅读springboot 之使用 SpringBootCondition获得答案