springBoot启动的时候动态选择装载某些bean

佚名 6年前 (2019-04-12) 随笔 544人围观抢沙发百度已收录

最近有这样一个场景，我们使用了elasticjob lite框架，希望某些job在指定服务器不启动。让spring动态的来装载所需要的job及相关bean

这个时候可以使用@Conditional家族注解，该注解spring4.X后出现

SRE实战互联网时代守护先锋，助力企业售后服务体系运筹帷幄！一键直达领取阿里云限量特价优惠。

@Conditional源码如下，接受一个参数，实现了Condition接口的Class

Condition接口里面就一个方法

ConditionContext可以获取到spring比较多有用信息，具体翻看api，AnnotatedTypeMetadata能获取到标注@Conditional注解的类或方法的信息，有几个实现类，想获取这些信息需要强转到子类获取，matches()方法返回true他就会装载这个bean，返回false就不会去装载，当时我把他和@Bean注解还有@Service注解一起配合使用

以下是自定义Condition类代码

import org.springframework.context.annotation.Condition;
import org.springframework.context.annotation.ConditionContext;
import org.springframework.core.env.Environment;
import org.springframework.core.type.AnnotatedTypeMetadata;
import org.springframework.core.type.classreading.AnnotationMetadataReadingVisitor;
import org.springframework.core.type.classreading.MethodMetadataReadingVisitor;

public class MyCondition implements Condition {

    @Override
    public boolean matches(ConditionContext context, AnnotatedTypeMetadata metadata) {
        Environment env = context.getEnvironment();
        String enableGroup = env.getProperty("simpleJob.enableGroup");
        boolean result = true;
        if (metadata instanceof MethodMetadataReadingVisitor){
            MethodMetadataReadingVisitor mmrv = (MethodMetadataReadingVisitor) metadata;
            String methodName = mmrv.getMethodName();
            result = methodName.toUpperCase().startsWith(enableGroup);
        }else if (metadata instanceof AnnotationMetadataReadingVisitor){
            AnnotationMetadataReadingVisitor amrv = (AnnotationMetadataReadingVisitor) metadata;
            String className = amrv.getClassName();
            String simpleClassName = className.substring(className.lastIndexOf("."));
            result = simpleClassName.toUpperCase().startsWith(enableGroup);
        }
        return result;
    }
}