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

本文转载自查看原文 2020-09-02 18:20 2934

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

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

1.需要引入jar包：

<dependency>
    <groupId>org.yaml</groupId>
    <artifactId>snakeyaml</artifactId>
</dependency>

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

private static DumperOptions dumperOptions = new DumperOptions();
static{
    //设置yaml读取方式为块读取
    dumperOptions.setDefaultFlowStyle(DumperOptions.FlowStyle.BLOCK);
    dumperOptions.setDefaultScalarStyle(DumperOptions.ScalarStyle.PLAIN);
    dumperOptions.setPrettyFlow(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 = 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 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) {
            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 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 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;
}

对yaml文件进行新增方法：

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;
}

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