yaml 文件（字符串）的解析
package com.yin.common.util;
import org.apache.commons.lang.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.yaml.snakeyaml.Yaml;
import sun.security.action.GetPropertyAction;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.security.AccessController;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Objects;
/**
* @author yin
* @since 2022/3/20 14:45
*/
public class YmlUtil {
private static Logger logger = LoggerFactory.getLogger(YmlUtil.class);
public static void main(String[] args) {
// Map<String, Object> targetMap = parseYamlFile("D:\\code\\plugin_demo\\web\\src\\main\\resources\\");
// for (<String, Object> entry : ()) {
// (() + ":" + ());
// }
StringBuilder ymlBuilder= new StringBuilder();
try( FileInputStream fileInputStream = new FileInputStream(new File("D:\\code\\plugin_demo\\web\\src\\main\\resources\\"));
InputStreamReader inputStreamReader = new InputStreamReader(fileInputStream);
BufferedReader bufferedReader=new BufferedReader(inputStreamReader)) {
String lineStr = "";
while ((lineStr = bufferedReader.readLine()) != null) {
ymlBuilder.append(lineStr).append("\\r\\n");
}
}catch (Exception e) {
logger.error("read fail", e);
}
System.out.println(ymlBuilder.toString());
Map<String, Object> targetMap = parseYamlString(ymlBuilder.toString());
for (Map.Entry<String, Object> entry : targetMap.entrySet()) {
System.out.println(entry.getKey() + ":" + entry.getValue());
}
}
/**
* 解析yaml 文件为map
* @param yamlFile 解析yaml
* @return 生成后的map
*/
public static Map<String,Object> parseYamlFile(String yamlFile){
File file = new File(yamlFile);
if (!file.exists()) {
return Collections.EMPTY_MAP;
}
LinkedHashMap<String, Object> sourceMap = null;
try (FileInputStream fileInputStream = new FileInputStream(file);) {
sourceMap= new Yaml().loadAs(fileInputStream
, LinkedHashMap.class);
} catch (Exception e) {
logger.error("parseYamlFile fail", e);
}
if (Objects.isNull(sourceMap)) {
return Collections.EMPTY_MAP;
}
LinkedHashMap<String, Object> targetMap = new LinkedHashMap<>(sourceMap.size());
fillAllPathMap(sourceMap, targetMap, "");
return targetMap;
}
/**
* 解析yaml 字符串为map
* @param yaml 解析yaml字符串
* @return 生成后的map
*/
public static Map<String,Object> parseYamlString(String yaml){
String lineSeparator = AccessController.doPrivileged(
new GetPropertyAction(""));
yaml = yaml.replace("\\r", "");
yaml = yaml.replace("\\n", lineSeparator);
LinkedHashMap<String, Object> sourceMap = null;
try {
sourceMap = new Yaml().loadAs(yaml, LinkedHashMap.class);
} catch (Exception e) {
logger.error("parseYamlString fail", e);
}
if (Objects.isNull(sourceMap)) {
return Collections.EMPTY_MAP;
}
LinkedHashMap<String, Object> targetMap = new LinkedHashMap<>(sourceMap.size());
fillAllPathMap(sourceMap, targetMap, "");
return targetMap;
}
/**
* 将sourceMap 中的多层结构改为一层
*/
private static void fillAllPathMap(Map<String,Object> sourceMap,Map<String,Object> targetMap,String prefix){
prefix = StringUtils.isEmpty(prefix) ? prefix : prefix + ".";
for (Map.Entry<String, Object> entry : sourceMap.entrySet()) {
Object value = entry.getValue();
if(Objects.nonNull(value) && (value instanceof Map)){
fillAllPathMap((Map) value, targetMap, prefix+entry.getKey());
}else{
targetMap.put(prefix+entry.getKey(), value);
}
}
}
}