config-center

package com.config.center; import com.alibaba.fastjson.JSON; import com.config.center.model.*; import com.config.center.utils.DataTransUtil; import com.config.center.utils.HttpUtil; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Value; import org.springframework.beans.factory.config.ConfigurableBeanFactory; import org.springframework.boot.context.properties.ConfigurationProperties; import org.springframework.boot.context.properties.bind.Bindable; import org.springframework.boot.context.properties.bind.Binder; import org.springframework.boot.context.properties.source.ConfigurationPropertySource; import org.springframework.core.annotation.AnnotationUtils; import org.springframework.core.env.ConfigurableEnvironment; import org.springframework.core.env.MapPropertySource; import org.springframework.core.env.MutablePropertySources; import java.lang.reflect.Field; import java.nio.charset.StandardCharsets; import java.util.*; import java.util.function.Function; import java.util.stream.Collectors; /** * @author caojiancheng * @date 2024-02-20 18:21 */ @Slf4j public class ConfigCenterClient { /** * 服务端地址 */ private String url; //配置map private Map<Long, ConfigBO> configMap; private static volatile ConfigCenterClient client; private List<Object> refreshBeanList; private List<RefreshFieldValueBO> refreshFieldValueList; public static final String PROPERTY_SOURCE_NAME = "configCenter"; private ConfigCenterClient(String url) { this.url = url; this.refreshBeanList = new ArrayList<>(); this.refreshFieldValueList = new ArrayList<>(); //将配置中心的配置转换成property格式，即user.name=xxx List<ConfigVO> configList = getAllValidConfig(); this.configMap = configList2ConfigMap(configList); } public static ConfigCenterClient getInstance(String url) { return Optional.ofNullable(client).orElseGet(() -> { synchronized (ConfigCenterClient.class) { if (client == null) { client = new ConfigCenterClient(url); } return client; } }); } public List<ConfigVO> getAllValidConfig() { HttpRespBO httpRespBO = HttpUtil.httpGet(url + "/config/get"); return httpResp2ConfigVOList(httpRespBO); } public void addRefreshField(String key, RefreshFieldBO refreshFieldBO) { configMap.values().stream().map(ConfigBO::getConfigDataList).filter(Objects::nonNull) .flatMap(List::stream).filter(configDataBO -> configDataBO.getKey().equals(key)) .findFirst().ifPresent(configDataBO -> configDataBO.addRefreshField(refreshFieldBO)); } public void addRefreshFieldValue(Object bean, Field field, String value) { refreshFieldValueList.add(new RefreshFieldValueBO(bean, field, value)); } public void addRefreshBeanList(Object bean) { refreshBeanList.add(bean); } public Map<String, Object> getConfigProperty() { return configMap.values().stream().map(ConfigBO::getConfigDataList).filter(Objects::nonNull) .flatMap(List::stream).collect(Collectors.toMap(ConfigDataBO::getKey, ConfigDataBO::getValue, (k1, k2) -> k1)); } public void startShortPolling() { polling("/config/change/get", () -> { try { Thread.sleep(1000); } catch (InterruptedException e) { e.printStackTrace(); } }, 2000); } public void startLongPolling() { polling("/config/change/get/long", null, 30000); } public void polling(String uri, Runnable runnable, int readTimeout) { Thread thread = new Thread(() -> { while (!Thread.interrupted()) { try { Optional.ofNullable(runnable).ifPresent(Runnable::run); Map<Long, List<ConfigDataBO>> refreshConfigMap = new HashMap<>(); configMap.values().forEach(configBO -> { Optional.ofNullable(configBO.getConfigDataList()).ifPresent(cdList -> cdList.stream() .filter(cd -> cd.getRefreshFieldList() != null && !cd.getRefreshFieldList().isEmpty()) .forEach(refreshConfigMap.computeIfAbsent(configBO.getId(), k1 -> new ArrayList<>())::add)); }); if (refreshConfigMap.isEmpty()) { return; } Map<String, Integer> configIdMap = refreshConfigMap.keySet().stream() .collect(Collectors.toMap(String::valueOf, configId -> configMap.get(configId).getVersion())); HttpRespBO httpRespBO = HttpUtil.httpPostJson(url + uri, JSON.toJSONString(configIdMap), readTimeout); List<ConfigVO> configList = httpResp2ConfigVOList(httpRespBO); if (configList.isEmpty()) { continue; } configList.forEach(configVO -> { Map<String, Object> result = new HashMap<>(); DataTransUtil.buildFlattenedMap(result, configVO.getConfigData(), ""); ConfigBO configBO = this.configMap.get(configVO.getId()); configBO.setVersion(configVO.getVersion()); List<ConfigDataBO> configDataList = configBO.getConfigDataList(); Map<String, ConfigDataBO> configDataMap = configDataList.stream() .collect(Collectors.toMap(ConfigDataBO::getKey, Function.identity())); result.forEach((key, value) -> { ConfigDataBO configDataBO = configDataMap.get(key); if (configDataBO == null) { configDataList.add(new ConfigDataBO(key, value.toString())); } else { configDataBO.setValue(value.toString()); List<RefreshFieldBO> refreshFieldList = configDataBO.getRefreshFieldList(); if (refreshFieldList == null) { refreshFieldList = new ArrayList<>(); configDataBO.setRefreshFieldList(refreshFieldList); } refreshFieldList.forEach(refreshFieldBO -> { try { Field field = refreshFieldBO.getField(); field.setAccessible(true); field.set(refreshFieldBO.getBean(), value.toString()); } catch (Exception e) { log.error("startShortPolling set Field error", e); } }); } }); }); } catch (Exception e) { log.error("startShortPolling error", e); } } }); thread.setName("startShortPolling"); thread.setDaemon(true); thread.start(); } public void startSpringBootLongPolling(ConfigurableEnvironment environment, ConfigurableBeanFactory beanFactory) { if (configMap.isEmpty() || refreshFieldValueList.isEmpty()) { log.info("configMap.size:{} refreshFieldValueList.size:{}", configMap.size(), refreshFieldValueList.size()); return; } MutablePropertySources propertySources = environment.getPropertySources(); MapPropertySource configCenter = (MapPropertySource) propertySources.get(PROPERTY_SOURCE_NAME); if (configCenter