_.unionBy([arrays], [iteratee=_.identity])

50

_.unionBy([arrays], [iteratee=_.identity])

_.unionBy类似union方法，区别是传递一个iteratee方法来处理每一个数组元素来生成标准

参数

[arrays] (...Array): 需要处理的多个数组

[iteratee=_.identity] (Function): 处理每一个元素的遍历器

返回值

(Array): 返回连起来的去重后的数组

例子

_.unionBy([2.1], [1.2, 2.3], Math.floor);
// => [2.1, 1.2]
 
// The `_.property` iteratee shorthand.
_.unionBy([{ 'x': 1 }], [{ 'x': 2 }, { 'x': 1 }], 'x');
// => [{ 'x': 1 }, { 'x': 2 }]

源代码：

iteratee参数也可以直接传递对象的属性key，会被baseIteratee处理，省略

import baseFlatten from './.internal/baseFlatten.js'
import baseUniq from './.internal/baseUniq.js'
import isArrayLikeObject from './isArrayLikeObject.js'
import last from './last.js'

/**
 * This method is like `union` except that it accepts `iteratee` which is
 * invoked for each element of each `arrays` to generate the criterion by
 * which uniqueness is computed. Result values are chosen from the first
 * array in which the value occurs. The iteratee is invoked with one argument:
 * (value).
 *
 * @since 4.0.0
 * @category Array
 * @param {...Array} [arrays] The arrays to inspect.
 * @param {Function} iteratee The iteratee invoked per element.
 * @returns {Array} Returns the new array of combined values.
 * @see difference, union, unionWith, without, xor, xorBy
 * @example
 *
 * unionBy([2.1], [1.2, 2.3], Math.floor)
 * // => [2.1, 1.2]
 */
//类似union方法，区别是传递一个iteratee方法来处理每一个数组元素来生成标准
function unionBy(...arrays) {
  let iteratee = last(arrays)//最后一个元素是iteratee
  if (isArrayLikeObject(iteratee)) {//如果iteratee参数是array-like对象，就赋值为undefined
    iteratee = undefined
  }
  return baseUniq(baseFlatten(arrays, 1, isArrayLikeObject, true), iteratee)
  //调用baseUniq处理
}

export default unionBy

baseUniq

import SetCache from './SetCache.js'
import arrayIncludes from './arrayIncludes.js'//判断数组是否包含给定值
import arrayIncludesWith from './arrayIncludesWith.js'//判断数组是否包含给定值，区别是它的comparator需要作为参数传入
import cacheHas from './cacheHas.js'
import createSet from './createSet.js'
import setToArray from './setToArray.js'

/** Used as the size to enable large array optimizations. */
const LARGE_ARRAY_SIZE = 200

/**
 * The base implementation of `uniqBy`.
 *
 * @private
 * @param {Array} array The array to inspect.
 * @param {Function} [iteratee] The iteratee invoked per element.
 * @param {Function} [comparator] The comparator invoked per element.
 * @returns {Array} Returns the new duplicate free array.
 */
//uniqBy的基础实现，数组去重
function baseUniq(array, iteratee, comparator) {
  let index = -1//循环索引
  let includes = arrayIncludes//类似数组的includes方法，判断数组是否包含给定值
  let isCommon = true//有没有自定义比较器或者有没有开启缓存的标记

  const { length } = array//数组长度
  const result = []//结果数组
  let seen = result//临时数组，用来存入不重复的数组元素值

  if (comparator) {//如果传递了自定义比较器，incledes方法就用arrayIncludesWith
    isCommon = false
    includes = arrayIncludesWith
  }
  else if (length >= LARGE_ARRAY_SIZE) {//如果数组长度超过200，就设置缓存
    const set = iteratee ? null : createSet(array)//如果没有iteratee和comparator，就直接使用ES6的set类型去重
    if (set) {//如果可以用set类型去重，直接新建set对象然后再转换成数组后返回结果
      return setToArray(set)
    }
    isCommon = false//否则开启数组缓存
    includes = cacheHas
    seen = new SetCache
  }
  else {//如果没有自定义比较器，数组长度也没有超过200
    seen = iteratee ? [] : result
  }
  outer:
  while (++index < length) {//遍历数组
    let value = array[index]//当前元素
    const computed = iteratee ? iteratee(value) : value//iteratee处理后的当前元素

    value = (comparator || value !== 0) ? value : 0//如果没有传递comparator，将-0变为0
    if (isCommon && computed === computed) {//如果没有传递自定义比较器，或者没有开启缓存，并且computed不是NaN
      let seenIndex = seen.length//seen数组的长度
      while (seenIndex--) {//遍历seen数组，如果有和当前值一样的，就跳出本次循环直接进行下一次
        if (seen[seenIndex] === computed) {
          continue outer
        }
      }
      if (iteratee) {//如果有iteratee参数，seen结尾插入当前不重复的元素
        seen.push(computed)
      }
      result.push(value)//结果数组结尾插入当前不重复的元素
    }
    else if (!includes(seen, computed, comparator)) {//如果有自定义比较器或者开启了缓存，并且seen中没有当前元素
      if (seen !== result) {//开启缓存的情况seen数组要插入新值
        seen.push(computed)
      }
      result.push(value)//结果数组插入新值
    }
  }
  return result
}

export default baseUniq