递归展开数组对象,根据主键递归查找树形数据中对应的数据

/**
 *
 * @param {Object}
 * @param {Array} arr 数组对象
 * @param {String} childrenKey 子数组的key
 * @param {Number=} expandLevel 可选 展开的层级
 * @param {String=} levelKey 可选 树形结构中元素的层级字段 层级从0开始 当expandLevel有值时 levelKey为必须
 * @param {Boolean=} noParents 返回展开数组时不包含根节点 当root为true时 level为必须 否则默认为1
 * @example expandArrayObject({ arr, childrenKey[, level, root] })
 * @description 递归展开数组对象 默认展开时包含父节点
 */
export function expandArrayObject({
  arr,
  childrenKey,
  expandLevel,
  levelKey,
  noParents
}) {
  const list = []
  let currentLevel = 1
  const _level = expandLevel || (noParents && 1)
  let parentCode = null
  const recursiveExpand = (arr, pCode) => {
    parentCode = pCode
    if (_level && currentLevel > _level) {
      return
    }
    Array.isArray(arr) &&
      arr.forEach((item) => {
        currentLevel = item[levelKey] + 1
        if (noParents && currentLevel < _level) {
          recursiveExpand(item[childrenKey])
        } else {
          list.push(item)
          recursiveExpand(item[childrenKey])
        }
      })
  }
  recursiveExpand(arr, parentCode)
  return list
}

/**
 *
 * @param {Object}
 * @param arr 查找数据的元数组
 * @param childrenKey 子数组的key
 * @param codeKey 主键key
 * @param code 需要找到的元素的主键值
 * @description 根据主键递归查找树形数据中对应的数据
 */
export function getChildrenRecursive({ arr, childrenKey, codeKey, code }) {
  let result
  arr.forEach((item) => {
    if (item[codeKey] === code) {
      result = item
    } else {
      getChildrenRecursive({
        arr: item[childrenKey],
        childrenKey,
        codeKey,
        code
      })
    }
  })
  return result
}
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容