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;
    }

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

相关推荐
aiwery2 分钟前
大模型场景下的推送技术选型:轮询 vs WebSocket vs SSE
前端·agent
会豪3 分钟前
前端插件-不固定高度的DIV如何增加transition
前端
却尘3 分钟前
Server Actions 深度剖析(2):缓存管理与重新验证,如何用一行代码干掉整个客户端状态层
前端·客户端·next.js
小菜全3 分钟前
Vue 3 + TypeScript 事件触发与数据绑定方法
前端·javascript·vue.js
Hilaku7 分钟前
面试官开始问我AI了,前端的危机真的来了吗?
前端·javascript·面试
zheshiyangyang17 分钟前
TypeScript学习【一】
javascript·学习·typescript
shellvon1 小时前
前端攻防:揭秘 Chrome DevTools 与反调试的博弈
前端·逆向
β添砖java1 小时前
案例二:登高千古第一绝句
前端·javascript·css
却尘1 小时前
Server Actions 深度剖析:这就是个披着 React 外衣的 RPC
前端·rpc·next.js
TNTLWT1 小时前
单例模式(C++)
javascript·c++·单例模式