1. package com.hq.schoolcj.util;
    3. import lombok.extern.slf4j.Slf4j;
    4. import org.yaml.snakeyaml.Yaml;
    6. import java.util.HashMap;
    7. import java.util.LinkedHashMap;
    8. import java.util.Map;
    9. import java.util.regex.Matcher;
    10. import java.util.regex.Pattern;
    12. /**
    13.  * 获取yml参数配置帮助类
    14.  */
    15. @Slf4j
    16. public class YamlUtil {
    17.     private  Pattern p1 = Pattern.compile("\\$\\{.*?\\}");
    19.     /**
    20.      * key:文件索引名
    21.      * value：配置文件内容
    22.      */
    23.     private static Map<String , LinkedHashMap> ymls = new HashMap<>();
    24.     /**
    25.      * String:当前线程需要查询的文件名
    26.      */
    27.     private static ThreadLocal<String> nowFileName = new InheritableThreadLocal<>();
    29.     /**
    30.      * 加载配置文件
    31.      * @param fileName
    32.      */
    33.     public static void loadYml(String fileName){
    34.         nowFileName.set(fileName);
    35.         if (!ymls.containsKey(fileName)){
    36.             ymls.put(fileName , new Yaml().loadAs(YamlUtil.class.getResourceAsStream("/" + fileName),LinkedHashMap.class));
    37.         }
    38.     }
    40.     /**
    41.      * 读取yml文件中的某个value。
    42.      * 支持解析 yml文件中的 ${} 占位符
    43.      * @param key
    44.      * @return Object
    45.      */
    46.     public  Object getValue(String key){
    47.         String[] keys = key.split("[.]");
    48.         Map ymlInfo = (Map) ymls.get(nowFileName.get()).clone();
    49.         for (int i = 0; i < keys.length; i++) {
    50.             Object value = ymlInfo.get(keys[i]);
    51.             if (i < keys.length - 1){
    52.                 ymlInfo = (Map) value;
    53.             }else if (value == null){
    54.                 throw new RuntimeException("key不存在");
    55.             }else {
    56.                 String g;
    57.                 String keyChild;
    58.                 String v1 = (String)value;
    59.                 for(Matcher m = p1.matcher(v1); m.find(); value = v1.replace(g, (String)getValue(keyChild))) {
    60.                     g = m.group();
    61.                     keyChild = g.replaceAll("\\$\\{", "").replaceAll("\\}", "");
    62.                 }
    63.                 return value;
    64.             }
    65.         }
    66.         return "";
    67.     }
    69.     /**
    70.      * 读取yml文件中的某个value
    71.      * @param fileName  yml名称
    72.      * @param key
    73.      * @return Object
    74.      */
    75.     public Object getValue(String fileName , String key){
    76.         loadYml(fileName);
    77.         return getValue(key);
    78.     }
    80.     /**
    81.      * 框架私有方法，非通用。
    82.      * 获取 server.servlet.context-path的值
    83.      * @return
    84.      */
    85.     public String getContextPath(){
    86.         return (String) getValue("application.yml", "server.servlet.context-path");
    87.     }
    89.     /**
    90.      * 读取yml文件中的某个value，返回String
    91.      * @param fileName
    92.      * @param key
    93.      * @return String
    94.      */
    95.     public String getValueToString(String fileName , String key) {
    96.         return (String) getValue(fileName, key);
    97.     }
    98. }