SpringBoot自动装配

Springboot自动装配-自定义starter实践与踩坑

什么是spring-boot的starter

简单来说就是:
引入相应的dependency就能自动向SpringApplicationContext中注入相关功能的bean。

其实道理很简单,将制定的自动装配类配置到META-INF下,再配合条件@Conditional就能达到看起来的效果即自动的注入,省去了大量配置

比如:

<dependency>
	<groupId>com.risk</groupId>
    <artifactId>distribute-design-spring-boot-starter</artifactId>
    <version>0.0.1-SNAPSHOT</version>
</dependency>

上面的jar包是我们自定义的starter。
他看起来是一个普通的jar包(有点像公司的二方包),

先说步骤
再说注意事项或者叫做所谓的坑(实际是理解不深刻)。

自定义starter步骤

我们以maven项目为例
分成如下的三个步骤:

  1. resource/META-INF/目录下建立文件spring.factories,并填写自动装配类
org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
  com.risk.distribute.design.lock.DistributeLockAutoConfiguration

其中DistributeLockAutoConfiguration@Configuration注解标识的

  1. 打包成jar,在工程中引入打包成的jar

3 定义配置类

@Configuration //①来标识它是一个配置类
@ConditionalOnProperty(value = "distribute.lock.enable",havingValue = "true",matchIfMissing = true)
@Import({ZookeeperDistributeLockConfiguration.class,
        RedisDistributeLockConfiguration.class})
@AutoConfigureAfter({RedisAutoConfiguration.class})//我们使用redisTemplate所以用的这个
public class DistributeLockAutoConfiguration {
	//②这个配置类是往BeanFactory注入bean
    @Bean
    public LockTemplate<?> lockTemplate(DistributeLock distributeLock){
        return new LockTemplate(distributeLock);
    }
}

最主要的是要有①和②两个部分,其它是稍微高级一点的功能,它看起来就是如此这样简单。

简单解释一些稍微高级一点的功能:
当环境中有这个属性

注解功能案例功能
@ConditionalOnPropertySpring的Environment中有某个属性,且值=xxx条件当环境中有distribute.lock.enable=true 条件成立或者没有这个属性则认为条件成立
Import往BeanFactory放入配置类,当然也得看配置类是否满足条件这里直接导入两个配置类
AutoConfigureAfter外部的starter对应的配置类排序RedisAutoConfiguration在前面,先被Spring加载

大家自己根据自己的业务写一个配置类的时候有没有遇到一些问题?

上面的示例 是最基本的 但是在实际过程中可能会遇到如下的一些问题

  • 当前项目@Configuration修饰某个配置类 然后@AutoConfigureAfter和@AutoConfigureBefore 不生效
  • 在当前项目中 @ConditionalOnBean 与starter中@ConditionOnMissingBean 然后创建bean,发现你的项目中永远获取不到starter中定义的bean,启动报错

带着这两个问题看下源码

自动装配源码分析

一切始于我们的启动类注解@SpringbootApplication注解

springApplication#run
	AnnotationConfigServletWebServertApplicationContext#refresh
	AbstractApplicationContext#invokeBeanFactoryPostProcessors
		ConfigurationClassPostProcessor#postProcessBeanDefinitionRegistry
		ConfigurationClassParser#parse

ConfigurationClassPostProcessor这个类BeanFactoryPostProcessor的子类 从名字可以看出他是专门处理 @Configuration标注的类的。

	public void processConfigBeanDefinitions(BeanDefinitionRegistry registry) {
		List<BeanDefinitionHolder> configCandidates = new ArrayList<>();
		String[] candidateNames = registry.getBeanDefinitionNames();

		for (String beanName : candidateNames) {
			BeanDefinition beanDef = registry.getBeanDefinition(beanName);
			if (ConfigurationClassUtils.isFullConfigurationClass(beanDef) ||
					ConfigurationClassUtils.isLiteConfigurationClass(beanDef)) {
				if (logger.isDebugEnabled()) {
					logger.debug("Bean definition has already been processed as a configuration class: " + beanDef);
				}
			}
			else if (ConfigurationClassUtils.checkConfigurationClassCandidate(beanDef, this.metadataReaderFactory)) {
			//① 当前类上有@Configuration注解的  通常情况下是我们的启动类@SpringBootApplication 它里边包含@Configuration注解
				configCandidates.add(new BeanDefinitionHolder(beanDef, beanName));
			}
		}

		// Return immediately if no @Configuration classes were found
		if (configCandidates.isEmpty()) {
			return;
		}

		// Sort by previously determined @Order value, if applicable
		configCandidates.sort((bd1, bd2) -> {
			int i1 = ConfigurationClassUtils.getOrder(bd1.getBeanDefinition());
			int i2 = ConfigurationClassUtils.getOrder(bd2.getBeanDefinition());
			return Integer.compare(i1, i2);
		});

		// Detect any custom bean name generation strategy supplied through the enclosing application context
		SingletonBeanRegistry sbr = null;
		if (registry instanceof SingletonBeanRegistry) {
			sbr = (SingletonBeanRegistry) registry;
			if (!this.localBeanNameGeneratorSet) {
				BeanNameGenerator generator = (BeanNameGenerator) sbr.getSingleton(CONFIGURATION_BEAN_NAME_GENERATOR);
				if (generator != null) {
					this.componentScanBeanNameGenerator = generator;
					this.importBeanNameGenerator = generator;
				}
			}
		}

		if (this.environment == null) {
			this.environment = new StandardEnvironment();
		}

		// ② 构建解析标注了@Configuration 的解析器
		ConfigurationClassParser parser = new ConfigurationClassParser(
				this.metadataReaderFactory, this.problemReporter, this.environment,
				this.resourceLoader, this.componentScanBeanNameGenerator, registry);

		Set<BeanDefinitionHolder> candidates = new LinkedHashSet<>(configCandidates);
		Set<ConfigurationClass> alreadyParsed = new HashSet<>(configCandidates.size());
		//③ 具体的解析逻辑,因为candidates为标注了@Configuration的类的BeanDefinition的集合(你项目中的),在解析的过程中还会产生
		//一些 @Configuration标注的类 也会加入到这个集合中 --比如SPI方式自动装配进来一些类(通常是Starter中的)所以用了while的方式
		do {
			parser.parse(candidates);
			parser.validate();

			Set<ConfigurationClass> configClasses = new LinkedHashSet<>(parser.getConfigurationClasses());
			configClasses.removeAll(alreadyParsed);

			// Read the model and create bean definitions based on its content
			if (this.reader == null) {
				this.reader = new ConfigurationClassBeanDefinitionReader(
						registry, this.sourceExtractor, this.resourceLoader, this.environment,
						this.importBeanNameGenerator, parser.getImportRegistry());
			}
			//④ 加载configClasses 主要关注 @Import @Bean部分 这些部分注册到BeanDefinitionRegistry 
			this.reader.loadBeanDefinitions(configClasses);
			alreadyParsed.addAll(configClasses);

			candidates.clear();
			if (registry.getBeanDefinitionCount() > candidateNames.length) {
				String[] newCandidateNames = registry.getBeanDefinitionNames();
				Set<String> oldCandidateNames = new HashSet<>(Arrays.asList(candidateNames));
				Set<String> alreadyParsedClasses = new HashSet<>();
				for (ConfigurationClass configurationClass : alreadyParsed) {
					alreadyParsedClasses.add(configurationClass.getMetadata().getClassName());
				}
				for (String candidateName : newCandidateNames) {
					if (!oldCandidateNames.contains(candidateName)) {
						BeanDefinition bd = registry.getBeanDefinition(candidateName);
						if (ConfigurationClassUtils.checkConfigurationClassCandidate(bd, this.metadataReaderFactory) &&
								!alreadyParsedClasses.contains(bd.getBeanClassName())) {
							candidates.add(new BeanDefinitionHolder(bd, candidateName));
						}
					}
				}
				candidateNames = newCandidateNames;
			}
		}
		while (!candidates.isEmpty());

		if (sbr != null && !sbr.containsSingleton(IMPORT_REGISTRY_BEAN_NAME)) {
			sbr.registerSingleton(IMPORT_REGISTRY_BEAN_NAME, parser.getImportRegistry());
		}

		if (this.metadataReaderFactory instanceof CachingMetadataReaderFactory) {
			((CachingMetadataReaderFactory) this.metadataReaderFactory).clearCache();
		}
	}

上面的代码分开来看

① 获取所有的有@Configuration标注的类的BeanDefinition ,并进行排序
BeanDefinitionRegistry 获取已经扫描并转变为BeanDefinition的集合,进行循环判断类上面是否有@Configuration注解正常来说只有我们的启动类@SpringBootApplication中包含@Configuration注解

② 构建解析标注了@Configuration 的解析器 ConfigurationClassParser

③ 具体的解析逻辑,parser#parse(configurationClasses) 因为candidates为标注了@Configuration的类的BeanDefinition的集合(你项目中的),在解析的过程中还会产生一些 @Configuration标注的类 也会加入到这个集合中 --比如SPI方式自动装配进来一些类(通常是Starter中的)所以用了while的方式

④ 加载configClasses 主要关注 @Import @Bean部分 这些部分注册到BeanDefinitionRegistry

第三步就是我们的重点了

重点类ConfigurationClassParser 简单记录下比较重要的成员属性

class ConfigurationClassParser {
	private final BeanDefinitionRegistry registry;
	private final ComponentScanAnnotationParser componentScanParser;
	private final ConditionEvaluator conditionEvaluator;
	private final Map<ConfigurationClass, ConfigurationClass;
	private final DeferredImportSelectorHandler deferredImportSelectorHandler = new DeferredImportSelectorHandler();

}

解析@Configuration标注配置类ConfigurationClassParser#parse(Set< BeanDefinitionHolder > configCandidates)

	public void parse(Set<BeanDefinitionHolder> configCandidates) {
		for (BeanDefinitionHolder holder : configCandidates) {
			BeanDefinition bd = holder.getBeanDefinition();
			//去掉了try catch
			parse(((AnnotatedBeanDefinition) bd).getMetadata(), holder.getBeanName());
			this.deferredImportSelectorHandler.process();
	}
	protected final void parse(AnnotationMetadata metadata, String beanName) throws IOException {
		processConfigurationClass(new ConfigurationClass(metadata, beanName));
		//每个@Configuraiton的元数据和名字包装成了ConfigurationClass类
	}

特别说明:到目前为止 这里的configCandidates是来自于【启动类和当前项目】中你手工添加的@Configuration的类

	protected void processConfigurationClass(ConfigurationClass configClass) throws IOException {
	//① 评估是否需要跳过@Configuration标注的类 依据 有@Conditional 
		if (this.conditionEvaluator.shouldSkip(configClass.getMetadata(), ConfigurationPhase.PARSE_CONFIGURATION)) {
			return;
		}

		ConfigurationClass existingClass = this.configurationClasses.get(configClass);
		if (existingClass != null) {
			if (configClass.isImported()) {
				if (existingClass.isImported()) {
					existingClass.mergeImportedBy(configClass);
				}
				// Otherwise ignore new imported config class; existing non-imported class overrides it.
				return;
			}
			else {
				// Explicit bean definition found, probably replacing an import.
				// Let's remove the old one and go with the new one.
				this.configurationClasses.remove(configClass);
				this.knownSuperclasses.values().removeIf(configClass::equals);
			}
		}

		// Recursively process the configuration class and its superclass hierarchy.
		SourceClass sourceClass = asSourceClass(configClass);
		do {
			sourceClass = doProcessConfigurationClass(configClass, sourceClass);
		}
		while (sourceClass != null);

		this.configurationClasses.put(configClass, configClass);
	}

① 评估是否需要跳过@Configuration标注的类 ,依据conditionEvaluator#shouldSkip()方法的实现逻辑

没有@Conditional 的就不跳过
有@Conditional的判断 将所有的@Conditional集合起来进行排序
只要有一个条件不通过跳过去 不对当前配置类记性解析了,通过就不跳过

第①处代码如下:

	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);
		}

		List<Condition> conditions = new ArrayList<>();
		for (String[] conditionClasses : getConditionClasses(metadata)) {
			for (String conditionClass : conditionClasses) {
				Condition condition = getCondition(conditionClass, this.context.getClassLoader());
				conditions.add(condition);
			}
		}
		//将@Configuration标注的类上 所有的@Conditional集合起来记性排序
		AnnotationAwareOrderComparator.sort(conditions);

		for (Condition condition : conditions) {
			ConfigurationPhase requiredPhase = null;
			if (condition instanceof ConfigurationCondition) {
				requiredPhase = ((ConfigurationCondition) condition).getConfigurationPhase();
			}
			if ((requiredPhase == null || requiredPhase == phase) && !condition.matches(this.context, metadata)) {//只要有一个条件不满足就skip调
				return true;
			}
		}
		return false;
	}

第②处代码:

	@Nullable
	protected final SourceClass doProcessConfigurationClass(ConfigurationClass configClass, SourceClass sourceClass)
			throws IOException {

		// 删除部分不重要的代码
		
		Set<AnnotationAttributes> componentScans = AnnotationConfigUtils.attributesForRepeatable(
				sourceClass.getMetadata(), ComponentScans.class, ComponentScan.class);
		if (!componentScans.isEmpty() &&
				!this.conditionEvaluator.shouldSkip(sourceClass.getMetadata(), ConfigurationPhase.REGISTER_BEAN)) {
			for (AnnotationAttributes componentScan : componentScans) {
				// The config class is annotated with @ComponentScan -> perform the scan immediately
				Set<BeanDefinitionHolder> scannedBeanDefinitions =
						this.componentScanParser.parse(componentScan, sourceClass.getMetadata().getClassName());
				// Check the set of scanned definitions for any further config classes and parse recursively if needed
				for (BeanDefinitionHolder holder : scannedBeanDefinitions) {
					BeanDefinition bdCand = holder.getBeanDefinition().getOriginatingBeanDefinition();
					if (bdCand == null) {
						bdCand = holder.getBeanDefinition();
					}
					if (ConfigurationClassUtils.checkConfigurationClassCandidate(bdCand, this.metadataReaderFactory)) {
						parse(bdCand.getBeanClassName(), holder.getBeanName());
					}
				}
			}
		}

		// Process any @Import annotations
		processImports(configClass, sourceClass, getImports(sourceClass), true);

		// Process any @ImportResource annotations
		AnnotationAttributes importResource =
				AnnotationConfigUtils.attributesFor(sourceClass.getMetadata(), ImportResource.class);
		if (importResource != null) {
			String[] resources = importResource.getStringArray("locations");
			Class<? extends BeanDefinitionReader> readerClass = importResource.getClass("reader");
			for (String resource : resources) {
				String resolvedResource = this.environment.resolveRequiredPlaceholders(resource);
				configClass.addImportedResource(resolvedResource, readerClass);
			}
		}

		// Process individual @Bean methods
		Set<MethodMetadata> beanMethods = retrieveBeanMethodMetadata(sourceClass);
		for (MethodMetadata methodMetadata : beanMethods) {
			configClass.addBeanMethod(new BeanMethod(methodMetadata, configClass));
		}

		// Process default methods on interfaces
		processInterfaces(configClass, sourceClass);

		// Process superclass, if any
		if (sourceClass.getMetadata().hasSuperClass()) {
			String superclass = sourceClass.getMetadata().getSuperClassName();
			if (superclass != null && !superclass.startsWith("java") &&
					!this.knownSuperclasses.containsKey(superclass)) {
				this.knownSuperclasses.put(superclass, configClass);
				// Superclass found, return its annotation metadata and recurse
				return sourceClass.getSuperClass();
			}
		}

		// No superclass -> processing is complete
		return null;
	}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值