package com.bw.edgeagent.adapter.util; import io.quarkus.logging.Log; import org.yaml.snakeyaml.DumperOptions; import org.yaml.snakeyaml.Yaml; import java.io.*; import java.util.HashMap; import java.util.Map; /** * @Author mxf * @Description 参考 https://blog.csdn.net/qq_44843880/article/details/107937185?spm=1001.2101.3001.6650.1&utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7ECTRLIST%7ERate-1-107937185-blog-106142404.t5_refersearch_landing&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7ECTRLIST%7ERate-1-107937185-blog-106142404.t5_refersearch_landing&utm_relevant_index=1 * @Date 2022/9/22 17:48 */ public class YmlUtil { private YmlUtil() { } private static DumperOptions dumperOptions = new DumperOptions(); static { //设置yaml读取方式为块读取 dumperOptions.setDefaultFlowStyle(DumperOptions.FlowStyle.BLOCK); dumperOptions.setDefaultScalarStyle(DumperOptions.ScalarStyle.PLAIN); dumperOptions.setPrettyFlow(false); } public static boolean updateYaml(String url, String paramKey, Object paramValue) { Yaml yaml = new Yaml(dumperOptions); //yaml文件路径 String yamlUr = url; Map map = null; try { //将yaml文件加载为map格式 map = yaml.loadAs(new FileInputStream(yamlUr), Map.class); } catch (FileNotFoundException e) { e.printStackTrace(); } //获取当前参数值并且修改 boolean flag = updateYaml(paramKey, paramValue, map, yamlUr, yaml); return flag; } /** * 修改yaml中属性的值 * * @param key key是properties的方式： aaa.bbb.ccc (key不存在不修改) * @param value 新的属性值 （新属性值和旧属性值一样，不修改） * @param yamlToMap map结构 * @param path 路径 * @return true 修改成功，false 修改失败。 */ public static boolean updateYaml(String key, Object value, Map<String, Object> yamlToMap, String path, Yaml yaml) { Object oldVal = getValue(key, yamlToMap); //未找到key 不修改 /* if (null == oldVal) { return false; }*/ System.out.println(key+" 原始旧值为："+oldVal); try { Map<String, Object> resultMap = setValue(yamlToMap, key, value); if (resultMap != null) { yaml.dump(resultMap, new FileWriter(path)); return true; } else { return false; } } catch (Exception e) { System.out.println("yaml file update failed !"); } return false; } /** * key格式：aaa.bbb.ccc * 通过properties的方式获取yaml中的属性值 * * @param key * @param yamlMap * @return */ public static Object getValue(String key, Map<String, Object> yamlMap) { String[] keys = key.split("[.]"); Object o = yamlMap.get(keys[0]); if (key.contains(".")) { if (o instanceof Map) { return getValue(key.substring(key.indexOf(".") + 1), (Map<String, Object>) o); } else { return null; } } else { return o; } } /** * 修改yaml中的属性值 */ public static Map<String, Object> setValue(Map<String, Object> map, String key, Object value) { String[] keys = key.split("\\."); int len = keys.length; Map temp = map; for (int i = 0; i < len - 1; i++) { if (temp.containsKey(keys[i])) { temp = (Map) temp.get(keys[i]); } else { return null; } if (i == len - 2) { temp.put(keys[i + 1], value); } } for (int j = 0; j < len - 1; j++) { if (j == len - 1) { map.put(keys[j], temp); } } return map; } public boolean insertYaml(String key, Object value, Map<String, Object> yamlToMap, String path) { Yaml yaml = new Yaml(dumperOptions); String[] keys = key.split("\\."); int len = keys.length; Map temp = yamlToMap; for (int i = 0; i < len; i++) { if (temp.containsKey(keys[i])) { temp = (Map) temp.get(keys[i]); } else { //该处对yaml的层级需要有判断，可能有问题 if (i + 2 == len) { Map<String, Object> newMap = new HashMap<>(); newMap.put(keys[len - 1], value); temp.put(keys[i], newMap); break; } else { temp.put(keys[i], value); break; } } } try { yaml.dump(yamlToMap, new FileWriter(path)); } catch (Exception e) { System.out.println("yaml file insert failed !"); return false; } return true; } /* public static void main(String[] args) { boolean b = YmlUtil.updateYaml("./././config/application.yml", "north.gatewayId", "value1"); System.out.println("修改结果：" + b); }*/ }