Lodash源码阅读-flattenDepth

Lodash 源码阅读-flattenDepth

概述

flattenDepth 函数可以控制数组"压扁"的层数。比如 [1, [2, [3, [4]], 5]],如果深度是 1,就变成 [1, 2, [3, [4]], 5];如果深度是 2,就变成 [1, 2, 3, [4], 5]。这个函数比 flatten 更灵活,可以精确控制要压扁几层。

前置学习

依赖函数

  • baseFlatten:实际干活的函数,处理数组扁平化的核心逻辑
  • toInteger:把值转成整数,用来处理深度参数

技术知识

  • 数组操作:数组的基本操作和嵌套数组的概念
  • 递归:函数自己调用自己的基本概念
  • 参数默认值:函数参数默认值的处理

源码实现

javascript 复制代码
/**
 * Recursively flatten `array` up to `depth` times.
 *
 * @static
 * @memberOf _
 * @since 4.4.0
 * @category Array
 * @param {Array} array The array to flatten.
 * @param {number} [depth=1] The maximum recursion depth.
 * @returns {Array} Returns the new flattened array.
 * @example
 *
 * var array = [1, [2, [3, [4]], 5]];
 *
 * _.flattenDepth(array, 1);
 * // => [1, 2, [3, [4]], 5]
 *
 * _.flattenDepth(array, 2);
 * // => [1, 2, 3, [4], 5]
 */
function flattenDepth(array, depth) {
  var length = array == null ? 0 : array.length;
  if (!length) {
    return [];
  }
  depth = depth === undefined ? 1 : toInteger(depth);
  return baseFlatten(array, depth);
}

实现思路

flattenDepth 的实现很简单:

  1. 先检查数组是不是空的或 null,如果是就返回空数组
  2. 处理深度参数:
    • 如果没传深度参数,默认用 1
    • 如果传了,就用 toInteger 转成整数
  3. 调用 baseFlatten 函数,告诉它要压扁几层
  4. baseFlatten 会遍历数组的每个元素:
    • 如果元素是数组且深度大于 0,就递归处理这个元素,深度减 1
    • 如果元素不是数组或深度为 0,就直接放到结果里
  5. 返回压扁后的新数组

源码解析

空值检查

javascript 复制代码
var length = array == null ? 0 : array.length;
if (!length) {
  return [];
}

这行代码做了两件事:

  • 检查数组是不是 nullundefined,如果是就把长度设为 0
  • 如果长度是 0(空数组或 null),直接返回空数组

深度参数处理

javascript 复制代码
depth = depth === undefined ? 1 : toInteger(depth);

这行代码处理深度参数:

  • 如果没传深度参数,默认用 1
  • 如果传了,就用 toInteger 转成整数,这样即使传的是字符串或小数也能正确处理

baseFlatten 的调用

javascript 复制代码
return baseFlatten(array, depth);

这里传了两个参数:

  • 第一个是要处理的数组
  • 第二个是深度参数,告诉 baseFlatten 要压扁几层

baseFlatten 函数会遍历数组的每个元素,根据深度参数决定是否继续递归处理嵌套数组。

总结

flattenDepth 函数虽然简单,但有几个重要的特点:

  1. 控制压扁层数 - 可以精确控制要压扁几层,比 flatten 更灵活

  2. 不修改原数组 - 返回新数组,保持原数组不变

  3. 处理边界情况 - 能正确处理空数组、null 等特殊情况

  4. 支持多种输入 - 通过 toInteger 可以处理各种类型的深度参数

这个函数对处理嵌套数据结构非常有用,它既能满足一定程度的扁平化需求,又不会像 flattenDeep 那样过度扁平化,让开发者可以精确控制处理的层级。

相关推荐
Ygrowly10 分钟前
搭建 Mock 服务,实现前端自调
前端
火车叼位13 分钟前
threejs api速查表, 你的下一个鼠标垫图案素材
前端·api·three.js
铃铃六16 分钟前
typescript中tpye和interface的区别
前端·typescript
页面魔术18 分钟前
🔥来听听尤雨溪是怎么亲述无虚拟dom的吧
前端·vue.js·前端框架
小old弟18 分钟前
⭐ NestJS入门指南:从零开始构建高效Node.js服务端应用
前端
不爱说话郭德纲22 分钟前
为了满足产品和设计,我撸了一个校准器ElkCalibrate📏
前端·vue.js·uni-app
IDOOY28 分钟前
OSS-服务端签名Web端直传+STS获取临时凭证+POST签名v4版本开发过程中的细节
前端·oss·服务端验签直传·sts方式·java+oss-sts验签
Hilaku41 分钟前
“全栈”正在淘汰“前端”吗?一个前端专家的焦虑与思考
前端·面试·程序员
前端大卫1 小时前
页面加载太慢?一文搞懂 JS 和 CSS 阻塞机制!
前端·javascript
十步杀一人_千里不留行1 小时前
I Built an Offline-Capable App by Myself: React Native Frontend, C# Backend
前端·react native·typescript