贝利信息

JavaScript迭代器协议如何自定义遍历【教程】

日期:2026-01-21 00:00 / 作者:夢幻星辰
要让对象支持 for...of,需在对象上定义[Symbol.iterator]方法,该方法每次调用须返回新迭代器实例,其next()方法返回{value, done}格式对象。

JavaScript 迭代器协议本身不提供“自定义遍历逻辑”的 API,而是要求你实现 Symbol.iterator 方法并返回一个符合迭代器协议的对象——也就是说,不是“配置”遍历,而是“构造”一个能被 for...ofArray.from、展开运算符等消费的对象。

如何让一个对象支持 for...of

关键不是重写语言机制,而是让对象返回一个有 next() 方法的对象。这个 next() 必须返回形如 { value: any, done: boolean } 的对象。

例如,让一个范围对象可遍历:

const range = {
  from: 1,
  to: 3,
  [Symbol.iterator]() {
    let current = this.from;
    return {
      next: () => {
        if (current <= this.to) {
          return { value: current++, done: false };
        } else {
          return { value: undefined, done: true };
        }
      }
    };
  }
};

for (const n of range) console.log(n); // 1, 2, 3

next() 中的 donevalue 怎么配合

done: true 表示迭代结束,此时 value 仍会被读取(比如 for...of 不再使用它,但 iterator.next().value 仍可访问),但规范不要求它有意义;常见做法是设为 undefined

为什么不能直接在原型上挂 [Symbol.iterator]

可以挂,但多数情况不推荐——因为 Symbol.iterator 是非枚举属性,for...in 看不到,不影响遍历行为;真正的问题在于:如果你在 Object.prototype 或某个基类原型上添加它,会影响所有继承者,可能破坏预期行为(比如让普通 plain object 变成可遍历,而它本不该有业务语义上的“遍历顺序”)。

与生成器函数 function* 的关系

生成器函数是创建迭代器最简洁的方式:它自动返回符合协议的迭代器对象,且天然支持暂停/恢复、状态保持。

等价写法:

class Range {
  constructor(from, to) {
    this.from = from;
    this.to = to;
  }
  *[Symbol.iterator]() {
    for (let i = this.from; i <= this.to; i++) {
      yield i;
    }
  }
}

自定义迭代器的核心约束很轻,但容易忽略两点:一是每次 [Symbol.iterator] 调用必须返回新迭代器(避免共享状态),二是 next() 返回值结构不能错——哪怕只是漏写 donefor...of 就会无限循环。