记录我自己写的常用的针对树形数据的处理函数：

let treeFn = {
  // 当前是是否为空值
  isNUll(val) {
    return val === null || val === void 0 || val === '' || val.toString() === 'NaN';
  },
  /**
   * 递归查询树形数据中id节点完整信息 - 通用
   * @param arr 物料树形下拉
   * @param id 树节点id
   * @param idField 树形结构id 字段
   * @param {string} children 判断树形结构子集的字段
   * @return {null|*}
   */
  findTreeItem(arr = [], id, idField = 'id', children = 'children') {
    if (this.isNUll(id)) return null;
    // let idField = this.cascaderProps.value; // 树形结构id 字段
    for (let i = 0; i < arr.length; i++) {
      let item = arr[i];
      if (item[idField] === id) {
        return item;
      }

      if (item[children] && item[children].length) {
        let res = this.findTreeItem(item[children], id, idField, children);
        if (res) return res;
      }
    }

    return null;
  },
  /**
   * cascader 级联物料下拉通用方法 - 根据最后一个节点找到其完整的节点(即：带上父节点)
   * eg：'12102002' = > ['铁矿粉', '12102002']
   * @param arr 物料树形下拉
   * @param id 树节点id
   * @param idField 树形结构id 字段
   * @param {string} children 判断树形结构子集的字段
   * @param {boolean} fuzzyMatch 判断树形数据节点与id是否相同时是否使用模糊匹配，默认false，即全等
   * @return string[]
   */
  getCascaderCodeByLastCode(arr = [], id, idField = 'id', children = 'children', fuzzyMatch = false) {
    if (this.isNUll(id)) return []
    // let idField = this.cascaderProps.value; // 树形结构id 字段

    let dgFn = (brr, parents = []) => {
      for (let i = 0; i < brr.length; i++) {
        let item = brr[i];
        let currentLevelIds = parents.concat(item[idField]) // 当前层code节点

        // 判断是否是目标子节点
        if (fuzzyMatch) { // 模糊匹配
          if (item[idField] === id || item[idField].includes(id)) { // 找到此节点
            return currentLevelIds;
          }
        } else { // 全等
          if (item[idField] === id) { // 找到此节点
            return currentLevelIds;
          }
        }

        if (item[children] && item[children].length) {
          let res = dgFn(item[children], currentLevelIds);
          if (res) return res;
        }
      }
    }

    return dgFn(arr, []) || [id]
  },
  /*
  * 树形数据扁平化
  * @param {any[]} arr 需要扁平化处理的数组
  * @param {string} children 判断树形结构子集的字段
  * @return 扁平化的一维数组，注意只保留了子节点，有子集的父节点本身不收纳
  * */
  flatTreeList(arr = [], children = 'children') {
    if (!arr || !arr.length) return []
    return arr.reduce((preArr, nextItem) => {
      // if(nextItem.children && nextItem.children.length){
      //   return preArr.concat(dg(nextItem.children))
      // }else{
      //   preArr.push(nextItem)
      // }
      // return preArr
      return preArr.concat(nextItem && nextItem[children] && nextItem[children].length ? this.flatTreeList(nextItem[children], children) : nextItem)
    }, [])
  }
}

若对你有帮助，请点个赞吧，若能打赏不胜感激，谢谢支持！
本文地址：https://www.jianshu.com/p/383509864b41?v=1732501925436，转载请注明出处，谢谢。