JS 数组转树形结构

/**
 * 包含parentId的数组转成树形结构数组
 * @param data 原始数据数组
 * @param idName ID属性名称
 * @param parentIdName 父ID属性名称
 * @returns {[]}
 */
const listToTree = function (data, idName, parentIdName) {
    if (!idName) {
        idName = 'id';
    }
    if (!parentIdName) {
        parentIdName = 'parentId';
    }

    // * 先生成parent建立父子关系
    const obj = {};
    data.forEach((item) => {
        obj[item[idName]] = item;
    });
    const parentList = [];
    data.forEach((item) => {
        // item.hasChildren = false;
        const parent = obj[item[parentIdName]];
        if (parent) {
            // * 当前项有父节点
            parent.children = parent.children || [];
            parent.children.push(item);
            // parent.hasChildren = true;
        } else {
            // * 当前项没有父节点 -> 顶层
            parentList.push(item);
        }
    });
    return parentList;
}

 

搜索

复制