封装JavaScript工具函数：返回数组中某值的所有索引

图 0：findAllIndices 工具函数核心概念图

在 JavaScript 日常开发中，当我们需要查找目标值在数组中的位置时，首先想到的往往是原生提供的 Array.prototype.indexOf()。然而该方法只能返回目标值第一次出现的下标；如果数组包含重复项，或者需要定位某个元素在数据流中的所有分布节点，原生 API 便无法直接胜任。本文将剖析原生方法的执行机制，从基础遍历开始，逐步封装一个兼顾高效单趟扫描、兼容特殊值（如 NaN）并支持自定义条件断言的实用工具函数。

一、原生方法的局限与核心问题

JavaScript 标准库在 Array.prototype 上挂载了 indexOf 与 lastIndexOf。它们的检索机制遵循“首个匹配原则”：一旦在遍历过程中检测到匹配项，即刻跳出循环并返回当前的索引数值。对于后续出现的所有相同元素，原生方法均会直接忽略。

开发者有时会通过在 while 循环中连续调用 indexOf(target, fromIndex) 来累加索引。这种写法虽然可用，但在底层实际上每次都在重复初始化内置扫描指针，代码逻辑相对晦涩，且当面对复杂数据类型或非严格全等匹配时依然束手无策。

图 1：原生 indexOf 单次返回机制与全量索引检索的原理对比

如上图所示，当数组中存在多个 'A' 时，原生 indexOf 在索引 0 处即刻返回终止；而我们的封装目标是持续完成单趟扫描，完整收集 [0, 2, 3] 并以数组格式返回。

二、基础实现：基于单趟循环与 reduce

要获取所有索引，最直接、性能最可控的方式是维护一个结果数组，顺序遍历源数组，当当前元素与目标值相匹配时，将当前下标推进结果集合中。

2.1 经典 for 循环实现

经典 for 循环在 V8 等现代 JavaScript 引擎中具备极高的优化上限，没有多余的闭包栈开销，适合绝大部分对性能有要求的场景：

/**
 * 基础版：返回数组中与目标值严格相等的全部索引
 * @param {Array} arr - 目标数组
 * @param {*} target - 待查找的目标值
 * @returns {number[]} 包含所有匹配索引的数组
 */
function findAllIndicesBasic(arr, target) {
  if (!Array.isArray(arr)) {
    return [];
  }

  const indices = [];
  const len = arr.length;

  for (let i = 0; i < len; i++) {
    if (arr[i] === target) {
      indices.push(i);
    }
  }

  return indices;
}

2.2 函数式写法：利用 reduce

如果你倾向于函数式编程风格，可以使用 Array.prototype.reduce 来完成累加收集：

function findAllIndicesReduce(arr, target) {
  if (!Array.isArray(arr)) return [];
  return arr.reduce((acc, curr, index) => {
    if (curr === target) acc.push(index);
    return acc;
  }, []);
}

图 2：工具函数五步执行流程图（参数校验至产出下标）

三、进阶封装：解决特殊值与复杂对象匹配

在工程实践中，单纯使用 === 会遇到两个经典的边界问题：

• NaN 的匹配失效：在 JavaScript 中，NaN === NaN 的评估结果为 false。如果数组中包含 NaN，基础版函数将无法找到任何匹配项。
• 引用类型的浅比较：如果数组存放的是对象（如 [{ id: 1 }, { id: 2 }]），即使内容相同，不同的引用对象使用 === 依然判定为不匹配。

为了解决这两个问题，我们可以借助 ES6 引入的 Object.is 规范化零值与 NaN 的判断，同时允许第二个参数既可以接收具体的比对值，也可以传入一个自定义的断言谓词函数（Predicate）：

图 3：匹配判定决策分支（谓词函数 vs Object.is vs 常规比较）

3.1 完整工业级封装代码

/**
 * 健壮通用版：返回数组中满足目标值或断言规则的所有索引
 * @param {Array} arr - 待检索的源数组
 * @param {*|Function} targetOrPredicate - 目标值，或者用于判断的谓词函数 (item, index, array) => boolean
 * @returns {number[]} 匹配下标数组
 */
function findAllIndices(arr, targetOrPredicate) {
  // 1. 防御性检查：非数组或空数组直接返回空集合
  if (!Array.isArray(arr) || arr.length === 0) {
    return [];
  }

  const indices = [];
  const len = arr.length;

  // 2. 判断是否提供了谓词回调函数
  const isPredicate = typeof targetOrPredicate === 'function';

  for (let i = 0; i < len; i++) {
    let matched = false;

    if (isPredicate) {
      // 执行自定义谓词函数
      matched = Boolean(targetOrPredicate(arr[i], i, arr));
    } else {
      // 使用 Object.is 规避 NaN === NaN 为 false 及 +0/-0 的差异问题
      matched = Object.is(arr[i], targetOrPredicate);
    }

    if (matched) {
      indices.push(i);
    }
  }

  return indices;
}

3.2 覆盖典型边界的运行示例

四、边界情况与性能注意事项

在将该函数应用到不同宿主环境或大型数据集之前，应注意以下细节：

4.1 稀疏数组（Sparse Arrays）中的空位处理

如果数组中存在未赋值的空洞（即稀疏数组，如 new Array(3) 或 [1, , 3]），使用下标 arr[i] 访问空位会返回 undefined。如果你的检索目标刚好是 undefined，直接判断可能会把稀疏数组中的空位也视作匹配。如果需要严格区分“未初始化的空位”与“值为 undefined 的显式元素”，可以在循环中结合 i in arr 检查：

// 检查当前索引是否是显式存在的属性，而非空洞
if (i in arr && Object.is(arr[i], target)) {
  indices.push(i);
}

4.2 惰性求值：使用 Generator 应对海量数据流

当数组规模达到上百万条记录，且调用方通常只需要前 N 个匹配项（或者只需要边检索边处理）时，直接返回一个巨大的索引数组会带来不必要的内存开销。此时可以提供一个生成器版本：

/**
 * 惰性生成器版：逐个产出匹配的索引
 */
function* generateAllIndices(arr, targetOrPredicate) {
  if (!Array.isArray(arr)) return;
  const isPredicate = typeof targetOrPredicate === 'function';
  const len = arr.length;

  for (let i = 0; i < len; i++) {
    const matched = isPredicate
      ? Boolean(targetOrPredicate(arr[i], i, arr))
      : Object.is(arr[i], targetOrPredicate);

    if (matched) {
      yield i;
    }
  }
}

五、实现方案横向对比与选型总结

针对不同的业务场景，我们可以选择最契合的实现范式。下表对比了各种方式的特性：

图 4：工具函数选型决策与多维度横向评估总结图

封装“返回数组中某值的所有索引”工具函数的本质，是在单趟遍历的基础上统一相等性断言契约。通过默认使用 Object.is 替代常规全等比较，并向后兼容谓词回调函数，不仅解决了原生 indexOf 只能单次命中的痛点，还能直接胜任 NaN 查找、对象属性检索与区间过滤等多类复杂场景。