通过JAVA代码对yaml文件进行增加、修改的操作

近期有个需求，需要通过java代码对yaml文件进行修改或新增，对文件进行添加依赖或者删除依赖。

首先介绍对yaml文件的操作：

1.需要引入jar包：

<dependency>
    <groupId></groupId>
    <artifactId>snakeyaml</artifactId>
</dependency>

2.在class文件中，需要添加一下属性：

private static DumperOptions dumperOptions = new DumperOptions();
static{
    //设置yaml读取方式为块读取
    ();
    ();
    (false);
}

具体作用是为了生成出的yaml格式（详细作用可百度），注意：该方法相当于重新生成yaml文件，原yaml文件中的注释会消失，暂未找到解决方法。

update方法：

public boolean updateYaml(String url, String paramKey, Object paramValue) {
Yaml yaml = new Yaml(dumperOptions);
//yaml文件路径
String yamlUr = url + yamlUrl;
Map map = null;
try {
    //将yaml文件加载为map格式
    map = (new FileInputStream(yamlUr), );
} catch (FileNotFoundException e) {
    ();
}
//获取当前参数值并且修改
boolean flag = updateYaml(paramKey, paramValue, map, yamlUr, yaml);
return flag;
}

/**
 * 修改yaml中属性的值
 *
 * @param key      key是properties的方式：  (key不存在不修改)
 * @param value    新的属性值 （新属性值和旧属性值一样，不修改）
 * @param yamlToMap map结构
 * @param path 路径
 * @return true 修改成功，false 修改失败。
 */
public 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;
    }

    try {
        Map<String, Object> resultMap = setValue(yamlToMap, key, value);
        if (resultMap != null) {
            (resultMap, new FileWriter(path));
            return true;
        } else {
            return false;
        }
    } catch (Exception e) {
        ("yaml file update failed !");
    }
    return false;
}

/**
 * key格式：
 * 通过properties的方式获取yaml中的属性值
 *
 * @param key
 * @param yamlMap
 * @return
 */
public Object getValue(String key, Map<String, Object> yamlMap) {
    String[] keys = ("[.]");
    Object o = (keys[0]);
    if ((".")) {
        if (o instanceof Map) {
            return getValue(((".") + 1), (Map<String, Object>) o);
        } else {
            return null;
        }
    } else {
        return o;
    }
}

/**
 * 修改yaml中的属性值
 */
public Map<String, Object> setValue(Map<String, Object> map, String key, Object value) {
    String[] keys = ("\\.");

    int len = ;
    Map temp = map;
    for (int i = 0; i < len - 1; i++) {
        if ((keys[i])) {
            temp = (Map) (keys[i]);
        } else {
            return null;
        }
        if (i == len - 2) {
            (keys[i + 1], value);
        }
    }
    for (int j = 0; j < len - 1; j++) {
        if (j == len - 1) {
            (keys[j], temp);
        }
    }
    return map;
}

对yaml文件进行新增方法：

public boolean insertYaml(String key, Object value, Map<String, Object> yamlToMap, String path) {

    Yaml yaml = new Yaml(dumperOptions);
    String[] keys = ("\\.");

    int len = ;
    Map temp = yamlToMap;

    for (int i = 0; i < len; i++) {
        if ((keys[i])) {
            temp = (Map) (keys[i]);
        } else {
            //该处对yaml的层级需要有判断，可能有问题
            if(i + 2 == len){
                Map<String, Object> newMap = new HashMap<>();
                (keys[len-1],value);
                (keys[i],newMap);
                break;
            }else {
                (keys[i], value);
                break;
            }
        }
    }
    try {
        (yamlToMap, new FileWriter(path));
    } catch (Exception e) {
        ("yaml file insert failed !");
        return false;
    }
    return true;
}

对yaml文件中的字段进行删除同样可以做到，具体功能没有实现。

对xml文件的操作在另一篇文章：/qq_44843880/article/details/107937431