每日一篇——lodash——array——chunk

維和土豆發表於2019-03-16
_.chunk(array, [size=1])
複製程式碼

將陣列（array）拆分成多個 size 長度的區塊，並將這些區塊組成一個新陣列。 如果array 無法被分割成全部等長的區塊，那麼最後剩餘的元素將組成一個區塊。

使用方法

 chunk(['a', 'b', 'c', 'd'], 2)
 // => [['a', 'b'], ['c', 'd']]
 chunk(['a', 'b', 'c', 'd'], 3)
 // => [['a', 'b', 'c'], ['d']]
複製程式碼

值得注意的是，這是一個純函式，不會對傳入的data有任何影響。

原始碼分析

var baseSlice = require('./_baseSlice'),
    isIterateeCall = require('./_isIterateeCall'),
    toInteger = require('./toInteger');


/* Built-in method references for those with the same name as other `lodash` methods. */
var nativeCeil = Math.ceil,
    nativeMax = Math.max;

/**
 * Creates an array of elements split into groups the length of `size`.
 * If `array` can't be split evenly, the final chunk will be the remaining
 * elements.
 *
 * @static
 * @memberOf _
 * @since 3.0.0
 * @category Array
 * @param {Array} array The array to process.
 * @param {number} [size=1] The length of each chunk
 * @param- {Object} [guard] Enables use as an iteratee for methods like `_.map`.
 * @returns {Array} Returns the new array of chunks.
 * @example
 *
 * _.chunk(['a', 'b', 'c', 'd'], 2);
 * // => [['a', 'b'], ['c', 'd']]
 *
 * _.chunk(['a', 'b', 'c', 'd'], 3);
 * // => [['a', 'b', 'c'], ['d']]
 */
function chunk(array, size, guard) {
  if ((guard ? isIterateeCall(array, size, guard) : size === undefined)) {
    size = 1;
  } else {
    size = nativeMax(toInteger(size), 0);
  }
  var length = array == null ? 0 : array.length;
  if (!length || size < 1) {
    return [];
  }
  var index = 0,
      resIndex = 0,
      result = Array(nativeCeil(length / size));

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

module.exports = chunk;
複製程式碼

這個方法整體實現沒有很特別的地方，核心在於用Math.ceil對原陣列分割槽塊，然後迴圈切割原陣列。while迴圈寫得比較精簡，將運算和賦值寫在一起，對於不熟悉的人來說可能有點懵逼。這個方法有一個比較令人困惑的地方在於第三個引數guard以及isIterateeCall方法，這個引數在官網上沒有提及，一般我們呼叫chunk方法都只會傳前兩個引數，只有當我們這整個方法作為map之類方法的引數時，才會有第三個引數，那麼此時array就是當前元素，size就是index，而gard就變成了原陣列，這種情況下會對原陣列進行size=1的分割。當然不是隻要有三個引數就可以認為這是Iteratee的情況，簡而言之就是判斷gard[size]是否等於array，具體的isIterateeCall實現可以看如下程式碼：

var eq = require('./eq'),
    isArrayLike = require('./isArrayLike'),
    isIndex = require('./_isIndex'),
    isObject = require('./isObject');


/**
 * Checks if the given arguments are from an iteratee call.
 *
 * @private
 * @param {*} value The potential iteratee value argument.
 * @param {*} index The potential iteratee index or key argument.
 * @param {*} object The potential iteratee object argument.
 * @returns {boolean} Returns `true` if the arguments are from an iteratee call,
 *  else `false`.
 */
function isIterateeCall(value, index, object) {
  if (!isObject(object)) {
    return false;
  }
  var type = typeof index;
  if (type == 'number'
        ? (isArrayLike(object) && isIndex(index, object.length))
        : (type == 'string' && index in object)
      ) {
    return eq(object[index], value);
  }
  return false;
}

module.exports = isIterateeCall;
複製程式碼

相關文章