根据条件加载配置 - ImportSelector

根据条件加载配置 - ImportSelector

通过ImportSelector可以帮助我们从多个配置类中选择你需要的配置类进行加载。

例如典型的自动配置类的加载过程中,Spring boot autoconfiguration有118个自动配置类,但是最终只有22个配置类被加载,它就是通过ImportSelector实现的,其实现类为AutoConfigurationImportSelector。

大多数的条件配置可以通过在定义@Configuration类时指定执行条件(比如@ConditionalOnClass或者@ConditionalOnMissingBean等)来实现。如果自动配置类不在ComponentScan的包路径下,自动配置类会定义在META-INF/spring.factories文件中,如下图:
在这里插入图片描述

1. 自定义ImportSelector

public class BusinessLogicAImportSelector implements ImportSelector, EnvironmentAware, BeanFactoryAware, BeanClassLoaderAware, ResourceLoaderAware {
    private Logger logger = LoggerFactory.getLogger(BusinessLogicAImportSelector.class);

    private BeanFactory beanFactory;
    private ClassLoader classLoader;
    private Environment environment;
    private ResourceLoader resourceLoader;

    @Override
    public String[] selectImports(AnnotationMetadata importingClassMetadata) {
        if (!importingClassMetadata.hasAnnotation(Contry.class.getName())) {
            return new String[0];
        }

        MultiValueMap<String, Object> contry = importingClassMetadata.getAllAnnotationAttributes(Contry.class.getName());
        if (contry.get("name").equals("TH")) {
            logger.info("当前版本是泰国版,返回泰国版相关的配置!");
            return new String[]{THConfiguration.class.getName()};
        } else {
            logger.info("当前版本是中国版,返回中国版相关的配置!");
            return new String[]{CNConfiguration.class.getName()};
        }
    }

    @Override
    public void setBeanClassLoader(ClassLoader classLoader) {
        this.classLoader = classLoader;
    }

    @Override
    public void setBeanFactory(BeanFactory beanFactory) throws BeansException {
        this.beanFactory = beanFactory;
    }

    @Override
    public void setEnvironment(Environment environment) {
        this.environment = environment;
    }

    @Override
    public void setResourceLoader(ResourceLoader resourceLoader) {
        this.resourceLoader = resourceLoader;
    }
}

2. 配置ImportSelector

@Configuration
@Contry(name = "TH")
@Import(BusinessLogicAImportSelector.class)
public class BusinessLogicConfiguration {
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值