配置中心-Lion源码学习

一、源码入口

<dependency>
	<groupId>com.dianping.lion</groupId>
	<artifactId>spring-boot-starter-lion</artifactId>
	<version>2.2.4</version>
</dependency>

spring.factories

org.springframework.boot.autoconfigure.EnableAutoConfiguration=com.dianping.lion.springboot.LionConfigAutoConfiguration
org.springframework.context.ApplicationListener=com.dianping.lion.springboot.LionConfigExcludeListener,\
  com.dianping.lion.springboot.LionConfigToEnvironmentListener
org.springframework.boot.env.EnvironmentPostProcessor=com.dianping.lion.springboot.LionActiveEnvironmentPostProcessor

LionConfigAutoConfiguration

@Configuration
@EnableLionConfig
@AutoConfigureOrder(0)
public class LionConfigAutoConfiguration {
    public LionConfigAutoConfiguration() {
    }
}

EnableLionConfig

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Inherited
@Import(LionConfigRegister.class)
public @interface EnableLionConfig {

}

LionConfigRegister

public class LionConfigRegister implements ImportBeanDefinitionRegistrar {

    @Override
    public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) {
        LionRegisterUtils.registerInitConfigIfNecessary(registry);
        LionRegisterUtils.registerSpringLionConfigIfNecessary(registry);
    }

}

LionRegisterUtils#registerSpringLionConfigIfNecessary

public static void registerSpringLionConfigIfNecessary(BeanDefinitionRegistry registry) {
        if (!registry.containsBeanDefinition(SpringLionConfig.SPRING_LION_CONFIG_BEAN_NAME)) {
            registry.registerBeanDefinition(SpringLionConfig.SPRING_LION_CONFIG_BEAN_NAME, new RootBeanDefinition(SpringLionConfig.class));
        }
    }

自动装配了SpringLionConfig，该类就是进行Bean的属性字段进行解析赋值。

二、属性字段赋值

SpringLionConfig的继承关系：

由上图可知，SpringLionConfig实现了BeanPostProcessor接口，所以Lion是在Bean初始化之前对属性字段进行解析赋值。

SpringLionConfig#postProcessBeforeInitialization

public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
        if (!this.beanName.equals(beanName)) {
            findConfigData(bean, beanName);
        }
        return bean;
    }

SpringLionConfig#findConfigData

private void findConfigData(final Object bean, final String beanName) {
	final Map<String, Set<ConfigMetaData>> result = getConfigDataCache();
	final Map<String, Set<ConfigMetaData>> current = new HashMap<>();

	//如果是aop代理
	Class proxyClass = bean.getClass();
	if (isProxy(bean)) {
		proxyClass = getTargetClass(bean);
	}
	final Class targetClass = proxyClass;
	final String targetClassName = targetClass.getName();

	// 1、将属性字段上的注解@ConfigKey的value作为key去查找value
	// field
	ReflectionUtils.doWithFields(targetClass, new ReflectionUtils.FieldCallback() {
				@Override
				public void doWith(Field field) throws IllegalArgumentException, IllegalAccessException {

					if (field.getModifiers() == Modifier.FINAL) {
						LOGGER.error("[Field] field is final, can not use, {}.{}", new Object[]{targetClassName, field.getName()});
						return;
					}
					ConfigKey configKey = field.getAnnotation(ConfigKey.class);
					String key = configKey.value();
					if (StringUtils.isEmpty(key)) {
						LOGGER.warn("[Field] config key is null, {}.{}", new Object[]{targetClassName, field.getName()});
					} else {
						boolean needChange = configKey.needChange();
						if (!needChange) {
							LOGGER.warn("[Field] config key: {} at {}.{} is not need change.", new Object[]{key, targetClassName, field.getName()});
						}

						ConfigMetaData configMetaData = new ConfigMetaData(bean, beanName, field, true, needChange);

						final Set<ConfigMetaData> allConfigMetaDataSet = result.get(key);
						if (!CollectionUtils.isEmpty(allConfigMetaDataSet)) {
							allConfigMetaDataSet.add(configMetaData);
							result.put(key, allConfigMetaDataSet);
						} else {
							result.put(key, Sets.newHashSet(configMetaData));
						}

						final Set<ConfigMetaData> currentMetaDataSet = current.get(key);
						if (!CollectionUtils.isEmpty(currentMetaDataSet)) {
							currentMetaDataSet.add(configMetaData);
							current.put(key, currentMetaDataSet);
						} else {
							current.put(key, Sets.newHashSet(configMetaData));
						}
					}
				}
			},
			new ReflectionUtils.FieldFilter() {
				@Override
				public boolean matches(Field field) {
					return field.isAnnotationPresent(ConfigKey.class);
				}
			});

	// method上的注解解析省略
	// *********

	if (!CollectionUtils.isEmpty(current)) {
		// init
		for (Map.Entry<String, Set<ConfigMetaData>> entry : current.entrySet()) {
			String key = entry.getKey();
			Set<ConfigMetaData> configMetaDatas = entry.getValue();

			for (ConfigMetaData configMetaData : configMetaDatas) {
				Object object = configMetaData.getBean();
				Object member = configMetaData.getMember();

				if (configMetaData.isField()) {
				
					// 2、属性字段赋值
					setFieldVal(object, (Field) member, key);
				} else {
					invokeMethod(object, (Method) member, key);
				}

			}
		}
	}
}