json树形结构数据的常用方法

1，初始化树形结构数据

/*
 *  树形数据递归，最后一级节点children初始化设为null
 *
 *  data      树数据(数组)
 *
 */
export function InitializationTree(data = []) {
    if (!data || !Array.isArray(data)) {
        return;
    }
    // 循环遍历json数据
    for (let i = 0; i < data.length; i++) {
        if (data[i].children.length < 1) {
            // children若为空数组，则将children设为undefined
            data[i].children = null;
        } else {
            // children若不为空数组，则继续 递归调用 本方法
            InitializationTree(data[i].children);
        }
    }
    return data;
}

2，根据id找到当前节点对象

/*
 *  树形数据递归，根据id找到当前节点对象
 *
 *  data      树数据(数组)
 *  targetId  查找的id
 *
 *  返回 节点id的对象
 */
export function TraverseTree(data = [], targetId) {
    if (!data || !Array.isArray(data)) {
        return;
    }
    for (let i = 0; i < data.length; i++) {
        if (data[i].id === targetId) {
            return data[i]
        } else if (data[i].children && data[i].children.length > 0) {
            let traverseTreeData = TraverseTree(data[i].children, targetId);
            if (traverseTreeData) return traverseTreeData
        }
    }
}

3，根据子节点id找到父节点集合

/*
 *  树形数据递归，根据子节点id找到父节点集合
 *
 *  treeData      树数据(数组)
 *  targetId  	  子节点id
 *
 *  返回 节点父级id的集合数组（包含当前子节点）
 */
export function GetTreeNodesParents(treeData = [], targetId) {
    let result = [];
    let callback = (data = [], targetId) => {
        if (!data || !Array.isArray(data)) {
            return;
        }
        for (const i in data) {
            if (data[i].id === targetId) {
                result.unshift(data[i].id);
                callback(treeData, data[i].parentId);
            }
            if (data[i].children && Array.isArray(data[i].children)) {
                callback(data[i].children, targetId);
            }
        }
    };
    callback(treeData, targetId);
    return result
}

4，过滤不需要的节点，返回新的tree

/*
*	treeData: Array   树数据
*	filterKeys: Array 需要过滤掉的节点id数组 如：[1,2,3]
*/
export function GetFilterTreeNodes(treeData = [], filterKeys = []) {
    let result = [];
    let callback = (data, nodes) => {
        if (!data || !Array.isArray(data) || !Array.isArray(nodes)) {
            return;
        }
        for (let i = 0; i < data.length; i++) {
            if (filterKeys.includes(data[i].id)) continue;
            const treeNode = {...data[i], children: []};
            nodes.push(treeNode);
            if (data[i].children && Array.isArray(data[i].children)) {
                callback(data[i].children, treeNode.children);
            }
        }
    };
    callback(treeData, result);
    return result;
}