Spring中@Condition底层实现原理

Love The Way You Lie 2024-04-05 07:22 67阅读 0赞

> @Condition是Spring中很关键的一个注解，它能够帮助我们只注册符合条件的Bean，我们可以通过配置、容器中的Bean、Java版本等筛选符合bean注册。

### 1.注解逻辑 ###

以ConditionOnBean为例

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

本质上引入了OnBeanCondition类来进行筛选，而这个类是继承于Condition类来实现的。

@FunctionalInterface
    public interface Condition {
        
    	/**
    	*	该方法用于判断该注解条件是否符合，符合返回true
    	**/
    	boolean matches(ConditionContext context, AnnotatedTypeMetadata metadata);
    
    }

#### 2.SpringBoot扩展 ####

SpringBoot为了使这个注解能够更加全面，定义了抽象类SpringBootCondition，所有的条件注解都是继承于此类，但SB提供的所有注解都没有实现matches方法，只有在SpringBootCondition中有，那么我们可以得出其实是使用了**模版方法**。

@Override
    public final boolean matches(ConditionContext context, AnnotatedTypeMetadata metadata) {
        
    	String classOrMethodName = getClassOrMethodName(metadata);
    	try {
        
    		// 本质通过子类getMatchOutcome来实现具体方法逻辑
    		ConditionOutcome outcome = getMatchOutcome(context, metadata);
    		logOutcome(classOrMethodName, outcome);
    		recordEvaluation(context, classOrMethodName, outcome);
    		return outcome.isMatch();
    	}
    	catch (NoClassDefFoundError ex) {
        
    		xxx
    	}
    	catch (RuntimeException ex) {
        
    		xxx
    	}
    }

#### 3.具体的getMatchOutcome方法 ####

@Override
    public ConditionOutcome getMatchOutcome(ConditionContext context, AnnotatedTypeMetadata metadata) {
        
    	ConditionMessage matchMessage = ConditionMessage.empty();
    	MergedAnnotations annotations = metadata.getAnnotations();
    	// 如果是ConditionOnBean则会执行以下判断逻辑
    	if (annotations.isPresent(ConditionalOnBean.class)) {
        
    		Spec<ConditionalOnBean> spec = new Spec<>(context, metadata, annotations, ConditionalOnBean.class);
    		// 关键方法用于获取匹配结果
    		MatchResult matchResult = getMatchingBeans(context, spec);
    		if (!matchResult.isAllMatched()) {
        
    			String reason = createOnBeanNoMatchReason(matchResult);
    			return ConditionOutcome.noMatch(spec.message().because(reason));
    		}
    		matchMessage = spec.message(matchMessage).found("bean", "beans").items(Style.QUOTE,
    				matchResult.getNamesOfAllMatches());
    	}
    	// 省略其他条件注解判断逻辑
    }

定位到getMatchingBeans方法，知道本质就是通过去容器获取该类型的beandefinition名称，后续会对获取到的名称做筛选，筛选出满足的条件

// 核心逻辑
    Collection<String> typeMatches = getBeanNamesForType(classLoader, considerHierarchy, beanFactory, type, parameterizedContainers);

同理我们可以看其他条件注解  
@ConditionalOnClass  
@ConditionalOnProperty等具体实现也是获取环境配置或者class.forName去判断是否有这个类。

### 2.调用入口 ###

#### 1.BeanFactoryPostProcessor ####

> condition和BeanFactoryPostProcessor有什么关系呢？  
> 我们都知道@condition注解的作用就是避免注册某些我们不需要的bean。那我们入手就在bean的扫描过程。那么我们定义到自动配置类扫描代码处，即ConfigurationClassPostProcessor类

我们定位到配置类注册逻辑处，postProcessBeanDefinitionRegistry() -> processConfigBeanDefinitions()

可以发现内部有parse类就是用于解析配置。最后跟踪到processConfigBeanDefinitions处会发现关键调用入口

if (this.conditionEvaluator.shouldSkip(configClass.getMetadata(), ConfigurationPhase.PARSE_CONFIGURATION)) {
        
    		return;
    }

最后我们跟踪到ConditionEvaluator就能发现本质就是调用了matches方法进行匹配

public boolean shouldSkip(@Nullable AnnotatedTypeMetadata metadata, @Nullable ConfigurationPhase phase) {
        
    	if (metadata == null || !metadata.isAnnotated(Conditional.class.getName())) {
        
    		return false;
    	}
    
    	if (phase == null) {
        
    		if (metadata instanceof AnnotationMetadata &&
    				ConfigurationClassUtils.isConfigurationCandidate((AnnotationMetadata) metadata)) {
        
    			return shouldSkip(metadata, ConfigurationPhase.PARSE_CONFIGURATION);
    		}
    		return shouldSkip(metadata, ConfigurationPhase.REGISTER_BEAN);
    	}
    	// 解析注解condition
    	List<Condition> conditions = new ArrayList<>();
    	for (String[] conditionClasses : getConditionClasses(metadata)) {
        
    		for (String conditionClass : conditionClasses) {
        
    			Condition condition = getCondition(conditionClass, this.context.getClassLoader());
    			conditions.add(condition);
    		}
    	}
    
    	AnnotationAwareOrderComparator.sort(conditions);
    
    	for (Condition condition : conditions) {
        
    		ConfigurationPhase requiredPhase = null;
    		if (condition instanceof ConfigurationCondition) {
        
    			requiredPhase = ((ConfigurationCondition) condition).getConfigurationPhase();
    		}
    		// matches方法进行判断
    		if ((requiredPhase == null || requiredPhase == phase) && !condition.matches(this.context, metadata)) {
        
    			return true;
    		}
    	}
    
    	return false;
    }

**看到这里会不会有个疑问，明明是按顺序解析Bean信息，如果使用了@ConditionOnBean，但解析该配置类时并未引入相同类型的Bean，解析完再解析其他同类型的Bean，该条件注解不就不起作用了吗？**

是的，解析顺序在条件注解判断时候也很重要，像ConditionOnBean判断容器中是否有该bean，但该bean还未注册到容器中，相当于判断就失效了，所有bean的解析顺序也很重要，所以我们会发现一般我们会先注册自己代码中的bean，然后扫描自动配置的bean。而同一配置类内我们可以用@AutoConfigureBefore等注解指定扫描顺序

### 3.注解扫描顺序 ###

我们查看AutoConfigurationSorter中的getInPriorityOrder方法

1.  按照字母顺序排序
2.  按照@AutoConfigureOrder排序
3.  按照@AutoConfigureBefore @AutoConfigureAfter排序

List<String> getInPriorityOrder(Collection<String> classNames) {
        
    	AutoConfigurationClasses classes = new AutoConfigurationClasses(this.metadataReaderFactory,
    			this.autoConfigurationMetadata, classNames);
    	List<String> orderedClassNames = new ArrayList<>(classNames);
    	// 按字母顺序排序
    	Collections.sort(orderedClassNames);
    	// 然后按照order顺序排序
    	orderedClassNames.sort((o1, o2) -> {
        
    		int i1 = classes.get(o1).getOrder();
    		int i2 = classes.get(o2).getOrder();
    		return Integer.compare(i1, i2);
    	});
    	// 最后按照 @AutoConfigureBefore @AutoConfigureAfter排序
    	orderedClassNames = sortByAnnotation(classes, orderedClassNames);
    	return orderedClassNames;
    }

经过上面的排序后就能保证配置类的加载顺序，保证Bean创建流程。

### 总结 ###

@Condition注解帮助我们选择适当的bean进行注册，而不同的bean注册时候又有错综复杂的依赖关系，在SpringBoot中都帮我们很好解决了。