JavaScript迭代器是统一遍历机制,要求对象实现Symbol.iterator方法返回含next()的迭代器对象,next()返回{value, done};手写Range类和BinaryTree树结构示例展示其应用,生成器函数是最简洁实现方式。

JavaScript 迭代器是一种统一的遍历机制,它让任意数据结构只要满足特定协议(即实现 [Symbol.iterator]() 方法),就能被 for...of、扩展运算符(...)、解构赋值等语法自然消费。
迭代器的核心:Iterator 接口
一个对象要成为可迭代的,必须在其原型链上提供一个名为 Symbol.iterator 的方法,该方法返回一个**迭代器对象**。这个迭代器对象必须有 next() 方法,每次调用返回形如 { value: any, done: boolean } 的对象。
- value:当前迭代出的值(可以是任意类型)
-
done:布尔值,
true表示遍历结束,false表示还有值可取
手写一个简单迭代器
比如想让一个自定义的区间类 Range 支持 for...of:
class Range {
constructor(start, end) {
this.start = start;
this.end = end;
}
[Symbol.iterator]() {
let current = this.start;
return {
next: () => {
if (current <= this.end) {
return { value: current++, done: false };
} else {
return { value: undefined, done: true };
}
}
};
}
}
// 使用
for (const n of new Range(2, 5)) {
console.log(n); // 2, 3, 4, 5
}
为更复杂的数据结构添加迭代能力
例如树结构,常需支持多种遍历顺序(前序、中序、后序、层序)。可将不同策略封装为独立的迭代器方法:
立即学习“Java免费学习笔记(深入)”;
class TreeNode {
constructor(val, left = null, right = null) {
this.val = val;
this.left = left;
this.right = right;
}
}
class BinaryTree {
constructor(root) {
this.root = root;
}
[Symbol.iterator]() {
// 默认提供中序遍历(生成器写法更简洁)
function inorder(node) {
if (!node) return;
yield inorder(node.left);
yield node.val;
yield inorder(node.right);
}
yield* inorder(this.root);
}
// 额外提供前序遍历方法
preorder() {
function traverse(node) {
if (!node) return;
yield node.val;
yield traverse(node.left);
yield* traverse(node.right);
}
return traverse(this.root);
}
}
// 使用
const tree = new BinaryTree(
new TreeNode(1,
new TreeNode(2),
new TreeNode(3)
)
);
for (const val of tree) {
console.log(val); // 2, 1, 3 (中序)
}
for (const val of tree.preorder()) {
console.log(val); // 1, 2, 3 (前序)
}
注意:function* 生成器函数会自动返回符合迭代器协议的对象,是实现迭代器最常用、最简洁的方式。
常见误区与注意事项
- 迭代器是一次性的:多次调用
[Symbol.iterator]()应返回**新**的迭代器对象,否则for...of多次执行会“失效” -
done: true后再调用next(),仍应返回{ value: undefined, done: true } - 数组、Map、Set 等内置类型已原生支持迭代器;类数组对象(如
arguments)需手动部署才能被for...of正确识别 - 不要在迭代过程中修改被遍历结构本身(如边遍历边删除节点),行为未定义且易出错











