com.github.yizzuide.milkomeda.universe.parser.yml.YmlParser Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2021 yizzuide All rights Reserved.
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

package com.github.yizzuide.milkomeda.universe.parser.yml;

import com.github.yizzuide.milkomeda.universe.extend.env.CollectionsPropertySource;
import org.apache.commons.lang3.StringUtils;

import java.util.HashMap;
import java.util.Map;

/**
 * YmlParser
 * yml解析器
 *
 * @author yizzuide
 * @since 3.0.0
 * @version 3.13.0
 * 

 * Create at 2020/03/29 19:24
 */
public class YmlParser {
    /**
     * 解析别名
     * @param nodeMap  拥有别名的节点Map，无别名键时添加本身
     * @return  别名Map
     */
    @SuppressWarnings("unchecked")
    public static Map parseAliasMap(Map nodeMap) {
        Map aliasNodeMap = new HashMap<>();
        for (Map.Entry node : nodeMap.entrySet()) {
            String key = node.getKey();
            Object value = node.getValue();
            // 别名替换
            if (value instanceof Map) {
                Map valueMap = (Map) value;
                key = String.valueOf(valueMap.keySet().toArray()[0]);
                value = valueMap.get(key);
            }
            YmlAliasNode ymlAliasNode = new YmlAliasNode();
            ymlAliasNode.setKey(key);
            ymlAliasNode.setValue(value);
            aliasNodeMap.put(node.getKey(), ymlAliasNode);
        }
        return aliasNodeMap;
    }

    /**
     * 多别名键解析
     * @param nodeMap           拥有别名的节点Map
     * @param ownerAliasKeys    拥有别名的key列表
     * @return  解析后的节点Map
     */
    public static Map parseAliasNodePath(Map nodeMap, String... ownerAliasKeys) {
        return parseAliasNodePath(nodeMap, null, ownerAliasKeys);
    }

    /**
     * 多别名键解析
     * @param nodeMap           拥有别名的节点Map
     * @param replaceData       如果值存在，使用替换的源强制替换配置的字段
     * @param ownerAliasKeys    拥有别名的key列表
     * @return  解析后的节点Map
     */
    public static Map parseAliasNodePath(Map nodeMap, Object replaceData, String... ownerAliasKeys) {
        Map aliasNodeMap = new HashMap<>();
        for (String ownerAliasKey : ownerAliasKeys) {
            parseAliasNodePath(nodeMap, aliasNodeMap, ownerAliasKey, null, replaceData);
        }
        return aliasNodeMap;
    }

    /**
     * 解析某个别名字段
     * @param nodeMap       拥有别名的节点Map
     * @param result        添加替换别名的目标map
     * @param ownerAliasKey 拥有别名的key
     * @param defaultValue  默认的值
     * @param replaceData   如果值存在，使用替换的源强制替换配置的字段
     */
    public static void parseAliasMapPath(Map nodeMap, Map result, String ownerAliasKey, Object defaultValue, Object replaceData) {
        Map aliasNodeMap = new HashMap<>(2);
        parseAliasNodePath(nodeMap, aliasNodeMap, ownerAliasKey, defaultValue, replaceData);
        if (aliasNodeMap.isEmpty()) {
            return;
        }
        YmlAliasNode ymlAliasNode = aliasNodeMap.get(ownerAliasKey);
        result.put(ymlAliasNode.getKey(), ymlAliasNode.getValue());
    }

    /**
     * 解析某个别名字段
     * @param nodeMap       拥有别名的节点Map
     * @param aliasNodeMap  添加替换别名的目标map（包住源key）
     * @param ownerAliasKey 拥有别名的key
     * @param defaultValue  默认的值
     * @param replaceData   如果值存在，使用替换的源强制替换配置的字段
     */
    @SuppressWarnings("all")
    public static void parseAliasNodePath(Map nodeMap, Map aliasNodeMap, String ownerAliasKey, Object defaultValue, Object replaceData) {
        String key = ownerAliasKey;
        Object value = nodeMap.get(ownerAliasKey);
        // 判定别名节点flag
        boolean hasAliasNode = false;
        // 未指定的配置字段，如果有默认值
        if ((value == null || StringUtils.isBlank(value.toString())) && defaultValue != null) {
            value = defaultValue;
        } else if (value instanceof Map) { // 别名替换
            Map valueMap = (Map) value;
            if (valueMap.size() > 0) {
                key = String.valueOf(valueMap.keySet().toArray()[0]);
                value = valueMap.get(key);
                hasAliasNode = true;
            }
        }

        // 配置中未添加该返回的字段，不计入返回结果
        if (value == null) {
            return;
        }

        // 创建别名节点
        YmlAliasNode ymlAliasNode = new YmlAliasNode();
        ymlAliasNode.setKey(key);
        ymlAliasNode.setValue(value);

        // 替换数据来源
        if (replaceData instanceof Map) {
            Map replaceMap = (Map) replaceData;
            Object initSetValue = ymlAliasNode.getValue();
            ymlAliasNode.setValue(hasAliasNode ? replaceMap.get(ownerAliasKey) : replaceMap.get(key));
            // 如果替换的Map没有值，使用yml里配置的值
            if (ymlAliasNode.getValue() == null) {
                ymlAliasNode.setValue(initSetValue);
            }
            // Replace empty token
            ymlAliasNode.setValue(CollectionsPropertySource.of(ymlAliasNode.getValue()));
            aliasNodeMap.put(ownerAliasKey, ymlAliasNode);
            return;
        }
        // 替换是指定的值
        if (StringUtils.isBlank(value.toString()) && replaceData != null) {
            ymlAliasNode.setValue(replaceData);
        }
        // Replace empty token
        ymlAliasNode.setValue(CollectionsPropertySource.of(ymlAliasNode.getValue()));
        aliasNodeMap.put(ownerAliasKey,  ymlAliasNode);
    }
}