tree/formatListToTree.js

/**
 * @module Tree/formatListToTree
* @description list 数据结构 转换成 树结构
* @param {Array} data 需要转换的数据
* @param {String} id 节点 id
* @param {String} pid 父级节点 id
* @param {String} child 子树为节点对象的某个属性值
* @param {Object} labels 需要新增的字段名集合 { label: 'category_name' }
* @return {Array}
*
* @example
* formatListToTree({data: [{id:1}, {id: 2}, {id: 3, pid: 1}]})
* =>
* [ { id: 1, children: [ {id: 3, pid: 1} ] }, { id: 2 } ]
*/
function formatListToTree({
  data = [],
  id = "id",
  pid = "pid",
  child = "children",
  labels = null
}) {
  if (!Array.isArray(data)) {
    return [];
  }
  const isObject = obj => Object.prototype.toString.call(obj) === "[object Object]";
  const arr = [];
  const tmpMap = {};
  const len = data.length;
  for (let i = 0; i < len; i++) {
    const val = data[i];
    if (val[id]) {
      tmpMap[val[id]] = val;
    }
    const item = isObject(labels) ? insertNewKeys(labels, data[i]) : data[i];
    const node = tmpMap[item[pid]];
    if (node) {
      node[child] = node[child] || [];
      node[child].push(item);
    } else {
      arr.push(item);
    }
  }

  return arr;
}

/**
 * 向原始对象中新增属性
 * @param {Object} labels
 * @param {Object} obj
 */
function insertNewKeys(labels, obj) {
  const _obj = {};
  Object.keys(labels).forEach(key => {
    _obj[key] = obj[labels[key]];
  });
  return Object.assign(obj, _obj);
}

export default formatListToTree;