config center

package com.config.enhance.config; import com.config.enhance.context.EnhanceContext; import com.config.enhance.resolver.EnhancePropertyResolver; import com.config.enhance.resolver.MutablePlaceholdersResolver; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.BeansException; import org.springframework.beans.factory.BeanFactory; import org.springframework.beans.factory.BeanFactoryAware; import org.springframework.beans.factory.InitializingBean; import org.springframework.beans.factory.config.ConfigurableListableBeanFactory; import org.springframework.boot.context.properties.bind.Binder; import org.springframework.boot.context.properties.bind.PlaceholdersResolver; import org.springframework.context.weaving.LoadTimeWeaverAware; import org.springframework.instrument.classloading.LoadTimeWeaver; import java.lang.reflect.Field; import java.lang.reflect.Method; import java.util.List; /** * @author caojiancheng * @date 2023-11-26 18:03 */ @Slf4j public class ConfigurationPropertiesBinderEnhancer implements LoadTimeWeaverAware, InitializingBean, BeanFactoryAware { private ConfigurableListableBeanFactory beanFactory; @Override public void afterPropertiesSet() throws Exception { EnhanceContext enhanceContext = EnhanceContext.getInstance(); List<EnhancePropertyResolver> propertyResolvers = enhanceContext.getPropertyResolvers(); if (propertyResolvers == null || propertyResolvers.isEmpty()) { log.info("enhancePropertyResolvers is empty"); return; } //解析@ConfigurationProperties enhanceConfigurationPropertiesBinder(propertyResolvers); } @Override public void setLoadTimeWeaver(LoadTimeWeaver loadTimeWeaver) { } private void enhanceConfigurationPropertiesBinder(List<EnhancePropertyResolver> enhancePropertyResolvers) { try { EnhanceContext enhanceContext = EnhanceContext.getInstance(); enhanceContext.setEnhanceBinderStatus(EnhanceContext.ENHANCE_BINDER_FAIL); //springboot 2.2.0及以上使用，之前的版本没有该bean String binderBeanName = "org.springframework.boot.context.internalConfigurationPropertiesBinder"; if (!beanFactory.containsBeanDefinition(binderBeanName)) { log.warn("not internalConfigurationPropertiesBinder,use EnhanceBindingPostProcessor"); return; } Object configurationPropertiesBinder = beanFactory.getBean(binderBeanName); //获取configurationPropertiesBinder的 binder 对象 Field binderField = configurationPropertiesBinder.getClass().getDeclaredField("binder"); binderField.setAccessible(true); Binder binder = (Binder) binderField.get(configurationPropertiesBinder); if (binder == null) { //如果binder为空，要先获取springboot定义的binder Method getBinder = configurationPropertiesBinder.getClass().getDeclaredMethod("getBinder"); getBinder.setAccessible(true); binder = (Binder) getBinder.invoke(configurationPropertiesBinder); } //获取springboot原生binder的解析方法，解析${xxx}的 Field placeholdersResolverField = binder.getClass().getDeclaredField("placeholdersResolver"); placeholdersResolverField.setAccessible(true); PlaceholdersResolver springbootResolver = (PlaceholdersResolver) placeholdersResolverField.get(binder); //将springboot和自定义解析方法组合到一起，先执行springboot的解析，对解析的结果再执行自定义的解析 MutablePlaceholdersResolver mutablePlaceholdersResolver = new MutablePlaceholdersResolver(springbootResolver); enhancePropertyResolvers.forEach(mutablePlaceholdersResolver::addPlaceholdersResolver); //将新的解析器设置回binder中 placeholdersResolverField.set(binder, mutablePlaceholdersResolver); //如果设置成功，设置 enhanceContext.setEnhanceBinderStatus(EnhanceContext.ENHANCE_BINDER_SUC); } catch (Exception e) { log.warn("enhanceConfigurationPropertiesBinder error,use EnhanceBindingPostProcessor", e); } } @Override public void setBeanFactory(BeanFactory beanFactory) throws BeansException { this.beanFactory = (ConfigurableListableBeanFactory) beanFactory; } }