根据条件加载配置 - 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 {
}