Lodash源码阅读-indexOf

Lodash 源码阅读-indexOf

一、介绍

indexOf 是 Lodash 中用于在数组中查找元素的方法,它返回找到的第一个匹配元素的索引,如果没找到则返回 -1。与 JavaScript 原生的 Array.prototype.indexOf 类似,但 Lodash 版本提供了更好的兼容性和对 NaN 值的特殊处理。

前置学习

在深入理解 indexOf 函数之前,建议先了解以下相关概念:

  • baseIndexOf:Lodash 的内部查找函数

  • toInteger:Lodash 的整数转换函数

二、源码实现

js 复制代码
function indexOf(array, value, fromIndex) {
  var length = array == null ? 0 : array.length;
  if (!length) {
    return -1;
  }
  var index = fromIndex == null ? 0 : toInteger(fromIndex);
  if (index < 0) {
    index = nativeMax(length + index, 0);
  }
  return baseIndexOf(array, value, index);
}

三、实现原理

1. 整体思路

indexOf 函数的实现思路可以概括为:

  1. 处理边界情况(空数组、null/undefined)
  2. 处理起始位置参数(默认值、负数索引)
  3. 调用 baseIndexOf 执行实际的查找逻辑
  4. 返回找到的索引或 -1

2. 参数说明

  • array:要搜索的数组
  • value:要查找的值
  • fromIndex:开始搜索的位置(索引),可选

3. 逐行解析

js 复制代码
var length = array == null ? 0 : array.length;
if (!length) {
  return -1;
}
  • 获取数组长度,如果数组为 nullundefined,则设为 0
  • 如果数组长度为 0,直接返回 -1(优化处理)
js 复制代码
var index = fromIndex == null ? 0 : toInteger(fromIndex);
if (index < 0) {
  index = nativeMax(length + index, 0);
}
  • 处理 fromIndex 参数
  • 如果未提供,默认为 0
  • 使用 toInteger 函数确保 fromIndex 是整数
  • 如果是负数,则从数组末尾开始计算位置,且不小于 0
js 复制代码
return baseIndexOf(array, value, index);
  • 调用 baseIndexOf 执行实际的查找逻辑

四、使用示例

js 复制代码
// 基本使用
_.indexOf([1, 2, 3, 4], 2); // => 1
_.indexOf([1, 2, 3, 4], 5); // => -1

// 指定起始位置
_.indexOf([1, 2, 3, 1, 2, 3], 2, 2); // => 4

// 处理 NaN 值(这是与原生 indexOf 的主要区别)
_.indexOf([1, NaN, 3], NaN); // => 1
[1, NaN, 3].indexOf(NaN); // => -1 (原生方法找不到 NaN)

// 负数索引
_.indexOf([1, 2, 3, 4], 2, -2); // => -1 (从倒数第二个元素开始查找)
_.indexOf([1, 2, 3, 4], 3, -2); // => 2

// 处理 null/undefined
_.indexOf(null, 1); // => -1
_.indexOf(undefined, 1); // => -1

五、与原生 indexOf 的区别

  1. NaN 处理 :Lodash 的 indexOf 可以找到数组中的 NaN 值,而原生 Array.prototype.indexOf 不行

  2. 兼容性:Lodash 的实现考虑了更多的边界情况和浏览器兼容性问题

  3. fromIndex 处理:对负数索引的处理更加健壮

六、总结

  1. 安全的空值处理

    js 复制代码
    var length = array == null ? 0 : array.length;

    使用宽松相等检查 null 和 undefined,避免报错

  2. 模块化设计

    js 复制代码
    return baseIndexOf(array, value, index);

    将核心查找逻辑抽离到专门的函数中

  3. 负索引处理

    js 复制代码
    index = nativeMax(length + index, 0);

    确保负索引转换为有效的数组索引,且不小于 0

  4. 早期返回优化

    js 复制代码
    if (!length) {
      return -1;
    }

    对空数组进行快速处理,避免不必要的计算

相关推荐
a1117764 小时前
Shirone 二次元博客主题 开源
前端·开源
IT_陈寒4 小时前
Python的切片赋值把我坑惨了,这不是bug是特性
前端·人工智能·后端
掘金酱4 小时前
【社区公告】致每一位掘友:关于这次调整, 想再说几句
前端·人工智能
CoderLiu4 小时前
程序化工具调用(PTC)与动态工作流引擎:深入大模型工具调用的架构演进与实践
前端·人工智能·后端
码农胖大海4 小时前
我的第一个产品,只有一段提示词
前端·ai编程·产品
邋遢道4 小时前
# 企业级 Agent 从 0 到 1(二):技术选型与最小骨架
前端·chrome
计算机魔术师5 小时前
我看了 OpenAI 首席科学家的最新长文,把原来的认知推翻了一遍
前端
乱码三千5 小时前
使用ComfyUI+MinMax-H3音视频模型生成视频
前端·后端·github
Coder_Ke5 小时前
代码我都定位了,Codex 还在考古:于是我写了个 VS Code 插件
前端
LiaCode6 小时前
别让 AI 把仓库写成“代码平行宇宙”:从 Deslop 看生成前查重
前端·后端·github