5

Spring 条件注解没生效?咋回事

 9 months ago
source link: http://www.javaboy.org/2023/0911/spring-condition.html
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.
neoserver,ios ssh client
3 个月前 14 分钟 读完 (大约 2058 个字)

Spring 条件注解没生效?咋回事

条件注解相信各位小伙伴都用过,Spring 中的多环境配置 profile 底层就是通过条件注解来实现的,松哥在之前的 Spring 视频中也有和大家详细介绍过条件注解的使用,感兴趣的小伙伴戳这里:Spring源码应该怎么学?

从 Spring4.0 开始,Spring 提供了一个更加细粒度的条件注解: ConfigurationCondition。从名字上就可以看出来这个是搭配 @Configuration 注解一起使用的,ConfigurationCondition 提供了一种更加细粒度的条件匹配,可以在配置或者 Bean 注册的时候去评估条件注解是否满足。

也就是说,当一个类上存在条件注解的时候,我们可以有两个评估条件注解是否满足的时机:

  1. 在配置的时候去评估。
  2. 在 Bean 注册的时候评估。

在配置的时候评估,可能会导致当前类都不会被加载,在 Bean 注册的时候再去评估,意味着当前类就会被加载。

1. ConfigurationCondition

我们先来看下这个类的定义:

public interface ConfigurationCondition extends Condition {
ConfigurationPhase getConfigurationPhase();
enum ConfigurationPhase {
PARSE_CONFIGURATION,
REGISTER_BEAN
}
}

大家看到,这里其实就是定义了两个枚举值,然后提供了一个方法返回枚举值。

  • PARSE_CONFIGURATION:这个表示 Condition 条件应该在解析 @Configuration 类时进行评估,如果评估不通过,则不会将 @Configuration 添加到容器中。
  • REGISTER_BEAN:这个表示添加常规 Bean 的时候去评估 Condition 条件(常规 Bean 就是指非配置类,例如添加搭配 @Bean 注解使用的条件注解),这个条件不会阻止注册 @Configuration 类到容器中。

其实道理很好懂,就是加载配置类的时候就根据条件注解判断要不要加载配置类,还是等到注册 Bean 的时候再去看条件注解是否满足条件。

2. 案例分析

松哥通过一个简单案例来和小伙伴们演示一下。

假设我现在有如下条件:

public class MyCondition implements ConfigurationCondition {
@Override
public ConfigurationPhase getConfigurationPhase() {
return ConfigurationPhase.PARSE_CONFIGURATION;
}

@Override
public boolean matches(ConditionContext context, AnnotatedTypeMetadata metadata) {
return context.getBeanFactory().containsBean("a");
}
}

这个条件我没有直接实现 Condition 接口,而是实现类 ConfigurationCondition 接口,在这个接口中,getConfigurationPhase 方法返回了 PARSE_CONFIGURATION,表示在加载配置类的时候就去评估条件是否满足,matches 方法则是去判断容器中是否存在一个名为 a 的 Bean。

现在我有两个配置类,分别是 A 和 B,如下:

@Configuration
public class A {
}
@Configuration
@Conditional(MyCondition.class)
public class B {
}

A 配置类正常加载,B 配置类有一个加载条件,就是得 A 存在,B 才会加载。

现在,在容器中加载 B 和 A 两个配置,如下:

AnnotationConfigApplicationContext ctx = new AnnotationConfigApplicationContext();
ctx.register(B.class,A.class);
ctx.refresh();
String[] beanDefinitionNames = ctx.getBeanDefinitionNames();
for (String beanDefinitionName : beanDefinitionNames) {
System.out.println(beanDefinitionName);
}

大家注意,加载的时候,我先加载了 B,后加载了 A,这点很重要,加载 B 的时候,由于此时容器中还不存在一个名为 a 的 Bean,而我们的评估时机是在处理配置类的时候,因此就会导致 B 配置类不会被加载,最终打印出来的 BeanName 就没有 b。

但是,如果我们将 MyCondition 中,条件注解的评估时机改为 ConfigurationPhase.REGISTER_BEAN,那么就表示在系统启动的时候,并不会去评估条件注解是否满足,而是会将 @Configuration 配置类进行解析,此时启动系统,就会发现最终打印出来的 beanName 里既有 a 又有 b。

3. 源码分析

接下来我们再来从源码的角度来分析一下上述行为。

在 Spring 中,提供了一个专门的内部类 ConditionEvaluator 来处理要不要跳过条件注解,该类中有一个名为 shouldSkip 的方法,用来处理此事:

public boolean shouldSkip(AnnotatedTypeMetadata metadata) {
return shouldSkip(metadata, null);
}
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 annotationMetadata &&
ConfigurationClassUtils.isConfigurationCandidate(annotationMetadata)) {
return shouldSkip(metadata, ConfigurationPhase.PARSE_CONFIGURATION);
}
return shouldSkip(metadata, ConfigurationPhase.REGISTER_BEAN);
}
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 configurationCondition) {
requiredPhase = configurationCondition.getConfigurationPhase();
}
if ((requiredPhase == null || requiredPhase == phase) && !condition.matches(this.context, metadata)) {
return true;
}
}
return false;
}

第一个方法不用多说,我们来看第二个重载方法,重载方法多了一个参数 ConfigurationPhase,这个就表示配置的阶段,也就是条件注解生效的阶段。

首先会去判断当前注解是否是一个条件注解,如果不是条件注意,那么就不能跳过,要继续后面的解析(继续后面的解析时 Bean 将会被注册),如果是条件注解,则继续后面的判断。继续判断,如果没有传递 phase 进来,说明没有指定应该在哪个阶段去评估条件注解,那么这个时候就去判断,如果当前注解是一个配置类上的注解,那么就设置 phase 为 PARSE_CONFIGURATION,然后继续调用 shouldSkip 方法,否则就设置 phase 为 REGISTER_BEAN 然后继续调用 shouldSkip 方法。

那么什么样的情况会被认为是一个配置类上的注解呢?如果当前类上添加的注解时 @Component、@ComponentScan、@Import、@ImportResource 以及这四种注解衍生出来的注解,亦或者当前类中有 @Bean 注解标记的方法,那么当前类就是一个配置类,就会设置 phase 为 PARSE_CONFIGURATION。

第二次进入 shouldSkip 方法的时候,就已经有明确的 phase 了。这次进来后,把所有的条件注解的条件收集起来,存入到 conditions 集合中,然后再对该集合进行排序。然后遍历该集合。遍历的时候就去判断这个条件注解是不是 ConfigurationCondition 类型的,如果是,则提取出来其中的 phase 为 requiredPhase,这个就表示这个条件注意希望自己被处理的阶段,接下来去判断,如果 requiredPhase 为空,说明条件并未指定自己的执行时间,那么就执行 matches 方法进行条件评估;如果 requiredPhase 不为空,并且和传入的 phase 相等,那么也是当前评估。其实这个判断核心逻辑就是以参数传入进来的 phase 为准,要么条件没有设置评估时机,要么设置了,但是得和参数传进来的 phase 一致,只有满足这两个条件,才会当场进行评估。

这就是系统条件注解的评估逻辑。

对于配置类来说,是在 AnnotatedBeanDefinitionReader#doRegisterBean 方法中调用评估逻辑的:

private <T> void doRegisterBean(Class<T> beanClass, @Nullable String name,
@Nullable Class<? extends Annotation>[] qualifiers, @Nullable Supplier<T> supplier,
@Nullable BeanDefinitionCustomizer[] customizers) {
AnnotatedGenericBeanDefinition abd = new AnnotatedGenericBeanDefinition(beanClass);
if (this.conditionEvaluator.shouldSkip(abd.getMetadata())) {
return;
}
//...
}

调用的时候并未明确指定 phase,所以会在进入到 shouldSkip 方法后,自行分析是哪个阶段评估条件注解。

对于 @Bean 注解标记的类来说,是在 ConfigurationClassBeanDefinitionReader#loadBeanDefinitionsForBeanMethod 方法中调用评估逻辑的:

private void loadBeanDefinitionsForBeanMethod(BeanMethod beanMethod) {
ConfigurationClass configClass = beanMethod.getConfigurationClass();
MethodMetadata metadata = beanMethod.getMetadata();
String methodName = metadata.getMethodName();
if (this.conditionEvaluator.shouldSkip(metadata, ConfigurationPhase.REGISTER_BEAN)) {
configClass.skippedBeanMethods.add(methodName);
return;
}
//...
}

这个调用的时候,就传入了 phase 了,直接指定了是在 Bean 初始化的时候评估。

好啦,这就是条件注解条件评估时机的两种情况。在 Spring Boot 中定义的条件注解里,有不少都用到了 ConfigurationCondition,而不是传统的 Condition,感兴趣的小伙伴可以自行查看哦~

# Spring

喜欢这篇文章吗?扫码关注公众号【江南一点雨】【江南一点雨】专注于 SPRING BOOT+微服务以及前后端分离技术,每天推送原创技术干货,关注后回复 JAVA,领取松哥为你精心准备的 JAVA 干货!

javaboy.jpg

About Joyk


Aggregate valuable and interesting links.
Joyk means Joy of geeK