专题一已经给大家介绍了数组扁平化,本篇将给大家介绍
lodash
中的flatten
是如何实现的。
一、lodash源码
1.基础函数
isFlattenable.js
// isFlattenable.js
import isArguments from '../isArguments.js' // 检查 value 是否是一个类 arguments 对象,在本篇不予讲解。
// ES6中内置属性,可用于判断数组是否可展开. 具体可见 MDN
const spreadableSymbol = Symbol.isConcatSpreadable;
/*
value[Symbol.isConcatSpreadable] === true时可展开,
可手动设置为false value[Symbol.isConcatSpreadable] = false
*/
/**
* 检查值是否可展开.
*
* @private
* @param {*} value 要检查的值.
* @returns {boolean} 返回布尔值,可展开 -> true, 不可展开 -> false.
*/
function isFlattenable(value) {
return Array.isArray(value) || isArguments(value) ||
!!(spreadableSymbol && value && value[spreadableSymbol]);
}
export default isFlattenable;
baseFlatten.js
// baseFlatten.js
import isFlattenable from './isFlattenable.js' // 是否可以扁平化
/**
* _.flatten 的基本实现与支持限制扁平化。
*
* @private
* @param {Array} array 需要扁平化的数组.
* @param {number} depth 扁平化的深度.
* @param {boolean} [predicate=isFlattenable] 该函数每次迭代判断是否可展开操作.
* @param {boolean} [isStrict] 是否可通过 predicate 检查.
* @param {Array} [result=[]] 初始结果数组.
* @returns {Array} 返回新的扁平数组.
*/
function baseFlatten(array, depth, predicate, isStrict, result) {
predicate || (predicate = isFlattenable);
result || (result = []);
if (array == null) {
return result; // 如果数组为空,则直接返回初始化结果
}
for (const value of array) {
if (depth > 0 && predicate(value)) {