From f65f995cc54a81ec7320b022abd49cab7bdf32a8 Mon Sep 17 00:00:00 2001 From: odboy Date: Mon, 9 Dec 2024 19:43:33 +0800 Subject: [PATCH] upd: format --- .../config/context/ClientConfigLoader.java | 394 +++++++++--------- 1 file changed, 206 insertions(+), 188 deletions(-) diff --git a/kenaito-config-core/src/main/java/cn/odboy/config/context/ClientConfigLoader.java b/kenaito-config-core/src/main/java/cn/odboy/config/context/ClientConfigLoader.java index 7f98fc6..c7a39bc 100644 --- a/kenaito-config-core/src/main/java/cn/odboy/config/context/ClientConfigLoader.java +++ b/kenaito-config-core/src/main/java/cn/odboy/config/context/ClientConfigLoader.java @@ -5,8 +5,16 @@ import cn.hutool.core.thread.ThreadUtil; import cn.hutool.core.util.StrUtil; import cn.odboy.config.constant.ClientConfigConsts; import cn.odboy.config.constant.ClientConfigVars; -import cn.odboy.config.model.msgtype.ClientInfo; import cn.odboy.config.netty.ConfigClient; +import java.io.File; +import java.nio.charset.StandardCharsets; +import java.nio.file.Files; +import java.nio.file.Path; +import java.nio.file.Paths; +import java.util.HashMap; +import java.util.Map; +import java.util.Properties; +import java.util.Set; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.BeansException; @@ -18,16 +26,6 @@ import org.springframework.core.env.ConfigurableEnvironment; import org.springframework.core.env.MapPropertySource; import org.yaml.snakeyaml.Yaml; -import java.io.File; -import java.nio.charset.StandardCharsets; -import java.nio.file.Files; -import java.nio.file.Path; -import java.nio.file.Paths; -import java.util.HashMap; -import java.util.Map; -import java.util.Properties; -import java.util.Set; - /** * 配置加载器 * @@ -36,188 +34,208 @@ import java.util.Set; */ @Configuration public class ClientConfigLoader { - private static final Logger logger = LoggerFactory.getLogger(ClientConfigLoader.class); + private static final Logger logger = LoggerFactory.getLogger(ClientConfigLoader.class); - /** - * 定时将配置写盘，缓存配置信息 - */ - private final Thread fixedTimeFlushConfigFileThread = - ThreadUtil.newThread( - () -> { - while (true) { - // 原来是所有的配置写到了一个文件中 - // try { - // if (!lastConfigs.isEmpty()) { - // List fileContent = new ArrayList<>(); - // for (Map.Entry kve : lastConfigs.entrySet()) { - // fileContent.add(kve.getKey() + "=" + kve.getValue()); - // } - // FileUtil.writeLines( - // fileContent, - // FileUtil.file(clientInfo.getCacheDir(), "config"), - // StandardCharsets.UTF_8); - // } - // Thread.sleep(5 * 1000); - // } catch (Exception e) { - // // 忽略 - // } - // 改为每个配置文件写到单独的文件中 - if (!ClientConfigVars.originConfigs.isEmpty()) { - try { - // 降低频率，减小对三方应用本身的影响 - Thread.sleep(30 * 1000); - } catch (Exception e) { - // 忽略 - } - for (Map.Entry kve : ClientConfigVars.originConfigs.entrySet()) { - try { - FileUtil.writeString( - kve.getValue(), - FileUtil.file(ClientConfigConsts.clientInfo.getCacheDir(), kve.getKey()), - StandardCharsets.UTF_8); - } catch (Exception e) { - logger.error( - "应用 {},环境 {},缓存配置文件 {} 失败", - ClientConfigConsts.clientInfo.getDataId(), - ClientConfigConsts.clientInfo.getEnv(), - kve.getKey(), - e); - } - } - } - } - }, - "fixedTimeFlushConfigFileThread"); - - @Bean - public BeanFactoryPostProcessor configLoader(ConfigurableEnvironment environment) { - return new BeanFactoryPostProcessor() { - @Override - public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) - throws BeansException { - String defaultCacheDir = getDefaultCacheDir(); - initClientInfo(defaultCacheDir, environment); - logger.info("客户端属性: {}", ClientConfigConsts.clientInfo); - validateCacheDirPath(defaultCacheDir, ClientConfigConsts.clientInfo.getCacheDir()); - createCacheDir(ClientConfigConsts.clientInfo.getCacheDir()); - fixedTimeFlushConfigFileThread.start(); - ThreadUtil.execAsync( - () -> { - try { - ConfigClient.getInstance().start(ClientConfigConsts.clientInfo.getServer(), ClientConfigConsts.clientInfo.getPort()); - } catch (InterruptedException e) { - logger.error("Netty客户端启动失败", e); - throw new RuntimeException(e); - } - }); - // 这里加个同步锁，等客户端准备就绪后，拉取配置完成时 - synchronized (ClientConfigConsts.clientInfo) { - while (!ClientConfigVars.isConfigLoaded) { - try { - // 等待配置加载完成 - ClientConfigConsts.clientInfo.wait(); - } catch (InterruptedException e) { - Thread currentThread = Thread.currentThread(); - String currentThreadName = currentThread.getName(); - currentThread.interrupt(); - logger.error("中断线程: {}", currentThreadName, e); - } - } - // 判断配置中心服务是否处于离线状态 - if (ClientConfigVars.isServerOffline) { - logger.info("配置中心离线，尝试从本地缓存加载配置文件"); - String cacheDir = ClientConfigConsts.clientInfo.getCacheDir(); - FileUtil.walkFiles( - FileUtil.file(cacheDir), - (file -> { - try { - String fileName = file.getName(); - ClientConfigVars.originConfigs.clear(); - // 配置原生内容 - ClientConfigVars.originConfigs.put(fileName, FileUtil.readString(file, StandardCharsets.UTF_8)); - // 转换为应用能识别的配置项 - ClientConfigVars.lastConfigs.clear(); - String suffix = FileUtil.getSuffix(file); - boolean isYml = "yml".equals(suffix) || "yaml".equals(suffix); - if (isYml) { - Yaml yaml = new Yaml(); - ClientConfigVars.lastConfigs.put(fileName, yaml.load(ClientConfigVars.originConfigs.get(fileName))); - } else { - Properties properties = new Properties(); - properties.load(StrUtil.getReader(ClientConfigVars.originConfigs.get(fileName))); - Map tempMap = new HashMap<>(1); - for (Map.Entry kv : properties.entrySet()) { - String key = (String) kv.getKey(); - tempMap.put(key, kv.getValue()); - } - ClientConfigVars.lastConfigs.put(fileName, tempMap); - } - } catch (Exception e) { - logger.info("配置文件转map失败", e); - } - })); - } - // 合并配置项 - ClientConfigVars.cacheConfigs.clear(); - Set>> filename2ConfigMap = ClientConfigVars.lastConfigs.entrySet(); - for (Map.Entry> filename2Config : filename2ConfigMap) { - ClientConfigVars.cacheConfigs.putAll(filename2Config.getValue()); - } - MapPropertySource propertySource = - new MapPropertySource(ClientConfigConsts.PROPERTY_SOURCE_NAME, ClientConfigVars.cacheConfigs); - environment.getPropertySources().addFirst(propertySource); + /** 定时将配置写盘，缓存配置信息 */ + private final Thread fixedTimeFlushConfigFileThread = + ThreadUtil.newThread( + () -> { + while (true) { + // 原来是所有的配置写到了一个文件中 + // try { + // if (!lastConfigs.isEmpty()) { + // List fileContent = new ArrayList<>(); + // for (Map.Entry kve : lastConfigs.entrySet()) { + // fileContent.add(kve.getKey() + "=" + kve.getValue()); + // } + // FileUtil.writeLines( + // fileContent, + // FileUtil.file(clientInfo.getCacheDir(), "config"), + // StandardCharsets.UTF_8); + // } + // Thread.sleep(5 * 1000); + // } catch (Exception e) { + // // 忽略 + // } + // 改为每个配置文件写到单独的文件中 + if (!ClientConfigVars.originConfigs.isEmpty()) { + try { + // 降低频率，减小对三方应用本身的影响 + Thread.sleep(30 * 1000); + } catch (Exception e) { + // 忽略 } + for (Map.Entry kve : ClientConfigVars.originConfigs.entrySet()) { + try { + FileUtil.writeString( + kve.getValue(), + FileUtil.file(ClientConfigConsts.clientInfo.getCacheDir(), kve.getKey()), + StandardCharsets.UTF_8); + } catch (Exception e) { + logger.error( + "应用 {},环境 {},缓存配置文件 {} 失败", + ClientConfigConsts.clientInfo.getDataId(), + ClientConfigConsts.clientInfo.getEnv(), + kve.getKey(), + e); + } + } + } } - }; - } + }, + "fixedTimeFlushConfigFileThread"); - private static void initClientInfo(String defaultCacheDir, ConfigurableEnvironment environment) { - ClientConfigConsts.clientInfo.setServer( - environment.getProperty(ClientConfigConsts.DEFAULT_CONFIG_NAME_SERVER, String.class, ClientConfigConsts.DEFAULT_CONFIG_SERVER)); - ClientConfigConsts.clientInfo.setPort( - environment.getProperty(ClientConfigConsts.DEFAULT_CONFIG_NAME_PORT, Integer.class, ClientConfigConsts.DEFAULT_CONFIG_PORT)); - ClientConfigConsts. clientInfo.setEnv( - environment.getProperty(ClientConfigConsts.DEFAULT_CONFIG_NAME_ENV, String.class, ClientConfigConsts.DEFAULT_CONFIG_ENV)); - ClientConfigConsts.clientInfo.setDataId( - environment.getProperty(ClientConfigConsts.DEFAULT_CONFIG_NAME_DATA_ID, String.class, ClientConfigConsts.DEFAULT_CONFIG_DATA_ID)); - ClientConfigConsts.clientInfo.setCacheDir( - environment.getProperty(ClientConfigConsts.DEFAULT_CONFIG_NAME_CACHE_DIR, String.class, defaultCacheDir)); - } - - private static String getDefaultCacheDir() { - String defaultCacheDir; - String os = System.getProperty("os.name"); - if (os.toLowerCase().startsWith(ClientConfigConsts.OS_TYPE_WIN)) { - defaultCacheDir = ClientConfigConsts.DEFAULT_PATH_WIN; - } else if (os.toLowerCase().startsWith(ClientConfigConsts.OS_TYPE_MAC)) { - defaultCacheDir = ClientConfigConsts.DEFAULT_PATH_MAC; - } else { - defaultCacheDir = ClientConfigConsts.DEFAULT_PATH_MAC; - } - return defaultCacheDir; - } - - private static void validateCacheDirPath(String defaultCacheDir, String cacheDir) { - if (defaultCacheDir.contains(ClientConfigConsts.DEFAULT_PATH_WIN_SEP) - && !cacheDir.contains(ClientConfigConsts.DEFAULT_PATH_WIN_SEP)) { - throw new RuntimeException(ClientConfigConsts.DEFAULT_CONFIG_NAME_CACHE_DIR + " 配置的路径不正确"); - } - if (cacheDir.contains(ClientConfigConsts.DEFAULT_PATH_WIN_SEP) && !cacheDir.contains(ClientConfigConsts.DEFAULT_PATH_SEP_WIN)) { - throw new RuntimeException( - ClientConfigConsts.DEFAULT_CONFIG_NAME_CACHE_DIR + " 配置的路径不正确, 正确的路径示范, " + ClientConfigConsts.DEFAULT_PATH_WIN); - } - } - - /** - * 创建缓存文件夹 - */ - private static void createCacheDir(String cacheDir) { - Path path = Paths.get(cacheDir); - if (!Files.exists(path)) { - File mkdir = FileUtil.mkdir(cacheDir); - if (!mkdir.canWrite()) { - throw new RuntimeException("缓存文件夹创建失败, 无读写权限"); + @Bean + public BeanFactoryPostProcessor configLoader(ConfigurableEnvironment environment) { + return new BeanFactoryPostProcessor() { + @Override + public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) + throws BeansException { + String defaultCacheDir = getDefaultCacheDir(); + initClientInfo(defaultCacheDir, environment); + logger.info("客户端属性: {}", ClientConfigConsts.clientInfo); + validateCacheDirPath(defaultCacheDir, ClientConfigConsts.clientInfo.getCacheDir()); + createCacheDir(ClientConfigConsts.clientInfo.getCacheDir()); + fixedTimeFlushConfigFileThread.start(); + ThreadUtil.execAsync( + () -> { + try { + ConfigClient.getInstance() + .start( + ClientConfigConsts.clientInfo.getServer(), + ClientConfigConsts.clientInfo.getPort()); + } catch (InterruptedException e) { + logger.error("Netty客户端启动失败", e); + throw new RuntimeException(e); + } + }); + // 这里加个同步锁，等客户端准备就绪后，拉取配置完成时 + synchronized (ClientConfigConsts.clientInfo) { + while (!ClientConfigVars.isConfigLoaded) { + try { + // 等待配置加载完成 + ClientConfigConsts.clientInfo.wait(); + } catch (InterruptedException e) { + Thread currentThread = Thread.currentThread(); + String currentThreadName = currentThread.getName(); + currentThread.interrupt(); + logger.error("中断线程: {}", currentThreadName, e); } + } + // 判断配置中心服务是否处于离线状态 + if (ClientConfigVars.isServerOffline) { + logger.info("配置中心离线，尝试从本地缓存加载配置文件"); + String cacheDir = ClientConfigConsts.clientInfo.getCacheDir(); + FileUtil.walkFiles( + FileUtil.file(cacheDir), + (file -> { + try { + String fileName = file.getName(); + ClientConfigVars.originConfigs.clear(); + // 配置原生内容 + ClientConfigVars.originConfigs.put( + fileName, FileUtil.readString(file, StandardCharsets.UTF_8)); + // 转换为应用能识别的配置项 + ClientConfigVars.lastConfigs.clear(); + String suffix = FileUtil.getSuffix(file); + boolean isYml = "yml".equals(suffix) || "yaml".equals(suffix); + if (isYml) { + Yaml yaml = new Yaml(); + ClientConfigVars.lastConfigs.put( + fileName, yaml.load(ClientConfigVars.originConfigs.get(fileName))); + } else { + Properties properties = new Properties(); + properties.load( + StrUtil.getReader(ClientConfigVars.originConfigs.get(fileName))); + Map tempMap = new HashMap<>(1); + for (Map.Entry kv : properties.entrySet()) { + String key = (String) kv.getKey(); + tempMap.put(key, kv.getValue()); + } + ClientConfigVars.lastConfigs.put(fileName, tempMap); + } + } catch (Exception e) { + logger.info("配置文件转map失败", e); + } + })); + } + // 合并配置项 + ClientConfigVars.cacheConfigs.clear(); + Set>> filename2ConfigMap = + ClientConfigVars.lastConfigs.entrySet(); + for (Map.Entry> filename2Config : filename2ConfigMap) { + ClientConfigVars.cacheConfigs.putAll(filename2Config.getValue()); + } + MapPropertySource propertySource = + new MapPropertySource( + ClientConfigConsts.PROPERTY_SOURCE_NAME, ClientConfigVars.cacheConfigs); + environment.getPropertySources().addFirst(propertySource); } + } + }; + } + + private static void initClientInfo(String defaultCacheDir, ConfigurableEnvironment environment) { + ClientConfigConsts.clientInfo.setServer( + environment.getProperty( + ClientConfigConsts.DEFAULT_CONFIG_NAME_SERVER, + String.class, + ClientConfigConsts.DEFAULT_CONFIG_SERVER)); + ClientConfigConsts.clientInfo.setPort( + environment.getProperty( + ClientConfigConsts.DEFAULT_CONFIG_NAME_PORT, + Integer.class, + ClientConfigConsts.DEFAULT_CONFIG_PORT)); + ClientConfigConsts.clientInfo.setEnv( + environment.getProperty( + ClientConfigConsts.DEFAULT_CONFIG_NAME_ENV, + String.class, + ClientConfigConsts.DEFAULT_CONFIG_ENV)); + ClientConfigConsts.clientInfo.setDataId( + environment.getProperty( + ClientConfigConsts.DEFAULT_CONFIG_NAME_DATA_ID, + String.class, + ClientConfigConsts.DEFAULT_CONFIG_DATA_ID)); + ClientConfigConsts.clientInfo.setCacheDir( + environment.getProperty( + ClientConfigConsts.DEFAULT_CONFIG_NAME_CACHE_DIR, String.class, defaultCacheDir)); + } + + private static String getDefaultCacheDir() { + String defaultCacheDir; + String os = System.getProperty("os.name"); + if (os.toLowerCase().startsWith(ClientConfigConsts.OS_TYPE_WIN)) { + defaultCacheDir = ClientConfigConsts.DEFAULT_PATH_WIN; + } else if (os.toLowerCase().startsWith(ClientConfigConsts.OS_TYPE_MAC)) { + defaultCacheDir = ClientConfigConsts.DEFAULT_PATH_MAC; + } else { + defaultCacheDir = ClientConfigConsts.DEFAULT_PATH_MAC; } + return defaultCacheDir; + } + + private static void validateCacheDirPath(String defaultCacheDir, String cacheDir) { + if (defaultCacheDir.contains(ClientConfigConsts.DEFAULT_PATH_WIN_SEP) + && !cacheDir.contains(ClientConfigConsts.DEFAULT_PATH_WIN_SEP)) { + throw new RuntimeException(ClientConfigConsts.DEFAULT_CONFIG_NAME_CACHE_DIR + " 配置的路径不正确"); + } + if (cacheDir.contains(ClientConfigConsts.DEFAULT_PATH_WIN_SEP) + && !cacheDir.contains(ClientConfigConsts.DEFAULT_PATH_SEP_WIN)) { + throw new RuntimeException( + ClientConfigConsts.DEFAULT_CONFIG_NAME_CACHE_DIR + + " 配置的路径不正确, 正确的路径示范, " + + ClientConfigConsts.DEFAULT_PATH_WIN); + } + } + + /** 创建缓存文件夹 */ + private static void createCacheDir(String cacheDir) { + Path path = Paths.get(cacheDir); + if (!Files.exists(path)) { + File mkdir = FileUtil.mkdir(cacheDir); + if (!mkdir.canWrite()) { + throw new RuntimeException("缓存文件夹创建失败, 无读写权限"); + } + } + } }