init

2022-06-19 05:56:05 +08:00
commit f285fffe30
6 changed files with 557 additions and 0 deletions
--- a/src/main/java/com/yuandian/common/Config.java
+++ b/src/main/java/com/yuandian/common/Config.java
@@ -0,0 +1,263 @@
+/**
+ * description
+ *
+ * @author eson
+ *2022年6月15日-17:38:16
+ */
+package com.yuandian.common;
+
+import java.io.IOException;
+import java.util.HashMap;
+import java.util.LinkedHashMap;
+import java.util.Map;
+import java.util.Properties;
+import java.util.concurrent.Executor;
+import java.util.concurrent.locks.Lock;
+import java.util.concurrent.locks.ReentrantLock;
+import java.util.function.Function;
+import org.yaml.snakeyaml.Yaml;
+import com.alibaba.nacos.api.NacosFactory;
+import com.alibaba.nacos.api.PropertyKeyConst;
+import com.alibaba.nacos.api.config.ConfigService;
+import com.alibaba.nacos.api.config.listener.Listener;
+import com.alibaba.nacos.api.exception.NacosException;
+import lombok.Getter;
+import lombok.Setter;
+import lombok.extern.slf4j.Slf4j;
+/**
+ * description nacos配置.
+ *
+ * @author eson
+ *2022年6月13日-17:08:46
+ */
+@Slf4j
+@Getter
+@Setter
+public class Config {
+	public static String DEFAULT_CONFIT_FILE = "application.properties";
+	public static String DEFAULT_CONFIG_ADDR = "yuandian.dataflow.config.nacos.server.addr";
+	// public static String DEFAULT_CONFIG_DATAID = "yuandian.dataflow.config.nacos.dataid";
+	// public static String DEFAULT_CONFIG_GROUP = "yuandian.dataflow.config.nacos.group";+
+	// 默认
+	public static String DEFAULT_GROUP_DATAID = "yuandian.dataflow";
+	// 所有生成的nacos客户端
+	private static HashMap<String,Config> configDict = new HashMap<>();
+	
+	// 配置的所有值主类
+	public Map<String,Object> data; 
+	//  nacos地址
+	public String serverAddr ;
+	// nacos dataId
+	public String dataId   ;
+	// nacos group
+	public String group   ;
+	// 线程安全配置锁
+	private Lock datalock;
+	// nacos 客户端类
+	private ConfigService configService;  
+    private Config(String GroupAndDataId) throws Exception {
+		String[] gad = GroupAndDataId.split("\\.");
+		if(gad.length != 2) {
+			throw new Exception("Group 或者 DataId 不能存在 '.' 的命令");
+		}
+		this.group = gad[0] + ENV_TEST;
+		this.dataId = gad[1] + ENV_TEST;
+		connect();
+    }
+	/**
+	 * 连接nacos
+	 * @throws IOException
+	 * @throws NacosException
+	 */
+	private void connect() throws IOException, NacosException {
+			if(configService != null) {
+				configService.shutDown();
+				configService = null;
+			}
+			// 获取 app
+			Properties prop = new Properties();
+			prop.load(Config.class.getClassLoader().getResourceAsStream(DEFAULT_CONFIT_FILE));
+		 
+			serverAddr = prop.getProperty(DEFAULT_CONFIG_ADDR + ENV_TEST).trim();
+			// dataId =  prop.getProperty(DEFAULT_CONFIG_DATAID).trim(); 
+			// group = prop.getProperty(DEFAULT_CONFIG_GROUP).trim();
+	
+			Properties properties = new Properties();
+			properties.put(PropertyKeyConst.SERVER_ADDR, serverAddr);
+			configService = NacosFactory.createConfigService(properties);
+			String content = configService.getConfig(dataId, group, 5000);
+			Yaml yaml = new Yaml();
+			data = yaml.load(content);
+			log.info(content);
+			
+			datalock = new ReentrantLock();
+			// 监听 配置更新事件
+			configService.addListener(dataId, group, new Listener() {
+				@Override
+				public void receiveConfigInfo(String configInfo) {
+					log.debug("recieve:" + configInfo);
+					try {
+						datalock.lock();
+						data = (Map<String, Object>)new Yaml().load(configInfo);
+						log.debug("{}",data);
+					} finally {
+						datalock.unlock();
+					}
+				}
+	
+				@Override
+				public Executor getExecutor() {
+					return null;
+				}
+			});
+	}
+	/**
+	 * 根据多个key获取yaml的值 keys 路径
+	 * @param keys 获取的key值
+	 * @return
+	 */
+	public Object get(String ...keys) {
+		var cur = data;
+		for(var i = 0; i < keys.length - 1;i++ ) { 
+			var key =  keys[i];
+			cur = (Map<String, Object>) cur.get(key);
+		}
+		return cur.get(keys[keys.length - 1]);
+	}
+	/**
+	 * 
+	 * 用于定位keys的路径后的操作. 创建keys后赋值. 如果存在keys, 可以直接赋值, 不存在则直接报错
+	 *
+	 * @author eson
+	 *2022年6月15日-下午12:06:37
+	 */
+	public class Operator {
+		Config config;
+		String[] keys;
+		Operator(Config config, String[] keys) {
+			this.config = config;
+			this.keys = keys;
+		}
+		/**
+		 * 创建seek的key
+		 * @return
+		 */
+		public Operator createKeys() { 
+			var cur = config.data;
+			for(var i = 0; i < keys.length;i++ ) { 
+				var key =  keys[i];
+				var vobj = cur.get(key);
+				if (vobj == null) {
+					vobj = new LinkedHashMap<>();
+					cur.put(key, vobj);
+				}
+				cur = (Map<String, Object>) vobj;
+			}
+			return this;
+		}
+		/**
+		 * 定位后赋值
+		 * @param value
+		 */
+		public void set(Object value) {
+			var cur = config.data;
+			for(var i = 0; i < keys.length - 1;i++ ) { 
+				var key =  keys[i];
+				cur = (Map<String, Object>) cur.get(key);
+			}
+			cur.put(keys[keys.length - 1], value);
+		}
+	}
+	/**
+	 * 定位. eg. seek("key1", "key2")
+	 * @param keys
+	 * @return
+	 */
+	public Operator seek(String ...keys) {
+		return new Operator(this, keys);
+	}
+	/**
+	 * 删除 key的值. 类型map操作. keys是一个路径 remove("a","b") --> {"a": {"b": 1}} 删除b
+	 * @param keys
+	 * @return
+	 */
+	public Object remove(String ...keys) {
+		var cur = data;
+		for(var i = 0; i < keys.length - 1;i++ ) { 
+			var key =  keys[i];
+			cur = (Map<String, Object>) cur.get(key);
+		}
+		return cur.remove(keys[keys.length - 1]);
+	}
+	/**
+	 * 更新配置
+	 * @return 返回是否发布成功.
+	 * @throws NacosException
+	 */
+	public Boolean update() throws NacosException {
+		return configService.publishConfig(dataId, group,new Yaml().dumpAsMap(data));
+	}
+	public static String ENV_TEST = "";
+	/**
+	 * 统一使用配置的入口函数 线程安全
+	 * @param GroupAndDataID 使用配置的标签 eg."group.dataId"
+	 * @param execute 匿名函数
+	 * @return 函数返回的值, 如果不需要直接返回null
+	 * @throws Exception
+	 */
+	public static Object UseConfig(String GroupAndDataID, Function<Config, Object> execute) throws Exception { 
+		log.info(GroupAndDataID);
+		Config cnf;
+		synchronized(configDict) {
+			cnf = configDict.get(GroupAndDataID);
+			if(cnf == null) {
+				cnf = new Config(GroupAndDataID);
+				configDict.put(GroupAndDataID, cnf);
+			}
+		}
+		try {
+			cnf.datalock.lock();
+			var res = execute.apply(cnf);
+			return res;			
+		} catch (Exception e) {
+			throw e;
+		} finally {
+			cnf.datalock.unlock();
+		}
+	}
+	 /**
+	  * 统一使用配置的入口函数 线程安全
+	  * @param execute 统一使用配置的入口匿名方法
+	  * @return 函数返回的值, 如果不需要直接返回null
+	 * @throws Exception
+	  */
+	public static Object UseConfig(Function<Config, Object> execute) throws Exception {
+		return UseConfig(DEFAULT_GROUP_DATAID, execute);
+	}
+	private static String Trim(String o, String mychars){
+		if(o == null) {
+			return null;
+		} else if (o.length() == 0) {
+			return o;
+		}
+		var buf = o.toCharArray();
+		var chars = mychars.toCharArray();
+		var start = 0;
+		for( ; start < buf.length; start++) {
+			for(var c: chars){
+				if(buf[start] != c) {
+					break;
+				}				
+			}
+		}
+		var end = buf.length;
+		for( ; end >= 0; end--) {
+			for(var c: chars){
+				if(buf[start] != c) {
+					break;
+				}				
+			}
+		}
+		return null;
+	} 
+}
--- a/src/main/resources/application.properties
+++ b/src/main/resources/application.properties
@@ -0,0 +1,6 @@
+server.port=3440
+
+yuandian.dataflow.config.nacos.server.addr="192.168.1.113:8848"
+yuandian.dataflow.config.nacos.server.addr-test="192.168.1.113:8848"
+
+ 
--- a/src/test/java/com/yuandian/common/ConfigTest.java
+++ b/src/test/java/com/yuandian/common/ConfigTest.java
@@ -0,0 +1,99 @@
+package com.yuandian.common;
+
+import java.time.Instant;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.MethodOrderer.OrderAnnotation;
+import org.junit.jupiter.api.Order;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.TestMethodOrder;
+import com.alibaba.nacos.api.exception.NacosException;
+import com.yuandian.common.Config;
+
+import lombok.extern.slf4j.Slf4j;
+@Slf4j
+@TestMethodOrder(OrderAnnotation.class)
+public class ConfigTest {
+    /**
+     * 测试配置基础用法
+     * @throws Exception
+     */
+    @Test
+    @Order(1)
+    void testUseConfig() throws Exception {
+        Config.ENV_TEST = "-test";
+        Config.UseConfig((cnf) -> {
+			log.info("{}",cnf.data);
+            
+            Assertions.assertEquals(cnf.get("key1", "key2"), "key_path");
+            Instant now = Instant.now();
+			cnf.data.put("use_config", now.toString());
+			try {
+				log.info("{}",cnf.update());
+			} catch (NacosException e) {
+				e.printStackTrace();
+			}
+			return null;
+		});
+    }
+    @Test
+    @Order(2)
+    void testRemove() throws Exception {
+        Config.ENV_TEST = "-test";
+        Config.UseConfig((cnf) -> {
+            cnf.remove("create1", "create2");
+			try {
+				log.info("{}",cnf.update()); // 所有增加删除操作要最后同步到nacos. 都需要update
+			} catch (NacosException e) {
+				e.printStackTrace();
+			}
+            cnf.remove("create1");
+			try {
+				log.info("{}",cnf.update()); // 所有增加删除操作要最后同步到nacos. 都需要update
+			} catch (NacosException e) {
+				e.printStackTrace();
+			}
+            return null;
+        });
+    }
+    @Test
+    @Order(3)
+    void testCreateKeys() throws Exception {
+        Config.ENV_TEST = "-test";
+ 
+        Config.UseConfig((cnf) -> {
+			log.info("{}",cnf.data); 
+            cnf.seek("create1", "create2", "create3").createKeys().set(Instant.now().toString());;
+			try {
+				log.info("{}",cnf.update()); // 所有增加删除操作要最后同步到nacos. 都需要update
+			} catch (NacosException e) {
+				e.printStackTrace();
+			}
+			return null;
+		});
+    }
+  
+    @Test
+    @Order(1)
+    void testUpdate() throws Exception {
+        Config.ENV_TEST = "-test";
+        Config.UseConfig((cnf) -> {
+            Instant now = Instant.now();
+			cnf.data.put("use_config", now.toString());
+			try {
+				log.info("{}",cnf.update());
+			} catch (NacosException e) {
+				e.printStackTrace();
+			}
+			return null;
+		});
+    }
+    @Test
+    @Order(1)
+    void testLabelConfig() throws Exception {
+        Config.UseConfig("org.fortest", (cnf)->{
+            log.info("{}", cnf.get("test")); 
+            Assertions.assertEquals(cnf.get("test"),  "groupAndDataId"); 
+            return null;
+        });
+    }
+}