lodash.chunk 源码解析

解析

function chunk(array, size = 1) {
  // 保证 size 的值 ≥ 0
  size = Math.max(toInteger(size), 0)
  // undefined == null 且 null == null，保证不会取空值的 length 属性而报错
  const length = array == null ? 0 : array.length
  // 做兜底处理
  if (!length || size < 1) {
    return []
  }
  let index = 0
  let resIndex = 0
  // 计算可以分多少个数组，并创建该数组
  const result = new Array(Math.ceil(length / size))

  while (index < length) {
    result[resIndex++] = slice(array, index, (index += size))
  }
  return result
}

该函数比较简单，lodash 对传入的参数做兜底处理，首先使用 toInteger 确保 size 为整数，如果用户传入 1.3 的话将当做 2 处理，除此之外还会判断 array 的长度，当 !length || size<1 成立时，直接返回空数组，不进行下面的计算。

如果传入的采纳数符合上述条件，将会计算该 array 可以按照 size 分为多少个数组，此处采取 Math.ceil 方法进行向上取整。举个例子，如果传入的 array 长度为 5，size 为 2，那么就可以分为 3 个数组。计算完成后，接下来调用内部 slice 方法（类似于 Array.prototype.slice，但是 slice 方法可以保证正确的返回）在 while 循环内对 array 进行分割，存储分割结果到 result 中，最后将 result 返回。

文档

_.chunk(array, [size=1])

将数组（array）拆分成多个 size 长度的区块，并将这些区块组成一个新数组。如果array 无法被分割成全部等长的区块，那么最后剩余的元素将组成一个区块。

引入版本

3.0.0

参数

array (Array): 需要处理的数组 [size=1] (number): 每个数组区块的长度

(Array): 返回一个包含拆分区块的新数组（注：相当于一个二维数组）。

例子

_.chunk(['a', 'b', 'c', 'd'], 2);
// => [['a', 'b'], ['c', 'd']]

_.chunk(['a', 'b', 'c', 'd'], 3);
// => [['a', 'b', 'c'], ['d']]