js把json数据转化成树形数据

/**
 * 将id、parentId这种JSON数组的数据格式转换为树节点格式
 * @param {Array} arr
 * @param {String} id
 * @param {String} pid
 * @return {Array}
 */
 function arrayToTree(arr, id, pid) {
  let data = JSON.parse(JSON.stringify(arr));
  if (!data || !data.length) return [];
  let targetData = [];                    //存储数据的容器(返回)
  let records = {};
  let itemLength = data.length;           //数据集合的个数
  for (let i = 0; i < itemLength; i++){
    let o = data[i];
    records[o[id]] = o;
  }
  for (let i = 0; i < itemLength; i++) {
    let currentData = data[i];
    let parentData = records[currentData[pid]];
    if (!parentData) {
      targetData.push(currentData);
      continue;
    }
    parentData.children = parentData.children || [];
    parentData.children.push(currentData);
  }
  return targetData;
}

/*转化函数*/ function(data, attributes) {     let resData = data;     let tree = [];     for(let i = 0; i < resData.length; i++) {         if(resData[i][attributes.parentId] === attributes.rootId) {             let obj = {                 id: resData[i][attributes.id],                 title: resData[i][attributes.name],                 children: []             };             tree.push(obj);             resData.splice(i, 1);             i--;         }     }     run(tree);       function run(chiArr) {         if(resData.length !== 0) {             for(let i = 0; i < chiArr.length; i++) {                 for(let j = 0; j < resData.length; j++) {                     if(chiArr[i].id == resData[j][attributes.parentId]) {                         let obj = {                             id: resData[j][attributes.id],                             title: resData[j][attributes.name],                             children: []                         };                         chiArr[i].children.push(obj);                         resData.splice(j, 1);                         j--;                     }                 }                 run(chiArr[i].children);             }         }     }       return tree;   }

1	var data=[{id:1,parentId:0,name:"测试1"},

1	{id:2,parentId:1,name:"测试2"}]

1 2 3 4 5

<em id="__mceDel"><br>let attributes = {    //定义数据属性名称 id: 'id', parentId: 'parentId', name: 'groupName',<br>rootId: 0 }<br>/*调用*/<br>formatTreeData(data,attributes);<br></em>

 

 

/**
 * 将数组转换为树结构数据
 *
 * @export
 * @param {Array} array
 * @param {Object} parent 父节点
 * @param {Array} tree
 * @return {Array}
 */
function array2tree(array, parent = { id: 0 }, tree = []) {
  let treeData = tree
  const children = _.filter(array, function(child) {
    return child.parentId === parent.id
  })
  if (!_.isEmpty(children)) {
    if (parent.id === 0) {
      treeData = children
    } else {
      parent['children'] = children
    }
    _.each(children, function(child) {
      array2tree(array, child)
    })
  }

  return treeData
}