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

/**
 *
 * @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
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值