17370845950

迭代器协议:Generator函数与Symbol.iterator
答案是Symbol.iterator方法使对象可迭代,通过实现该方法并返回具有next()的迭代器对象,可让自定义结构支持for...of和扩展运算符,如数组、字符串等内置类型均遵循此协议。

在JavaScript中,迭代器协议定义了对象如何被遍历。实现这一协议的关键是 Symbol.iterator 方法和能够生成迭代器的 Generator函数。它们让自定义数据结构支持 for...of 循环和扩展运算符等语法。

Symbol.iterator 与可迭代协议

一个对象如果实现了 Symbol.iterator 方法,就被称为可迭代对象。这个方法必须返回一个迭代器对象,该对象具备 next() 方法,用于逐步获取值。

例如,数组、字符串、Map 等内置类型都实现了 Symbol.iterator:

const arr = [1, 2, 3];
const iterator = arr[Symbol.iterator]();
iterator.next(); // { value: 1, done: false }
iterator.next(); // { value: 2, done: false }

我们也可以为自定义对象添加 Symbol.iterator 方法,使其可被 for...of 遍历:


const myCollection = {
  items: ['a', 'b', 'c'],
  [Symbol.iterator]() {
    let i = 0;
    return {
      next: () => {
        return i < this.items.length
          ? { value: this.items[i++], done: false }
          : { done: true };
      }
    };
  }
};

之后就能直接使用 for...of:

for (const item of myCollection) {
  console.log(item); // 输出 a, b, c
}

Generator 函数简化迭代器创建

手动实现 next() 和状态管理比较繁琐。Generator 函数(用 function* 定义)能自动返回一个符合迭代器协议的对象,极大简化过程。

每次调用 yield,都会暂停执行并返回一个 { value, done } 对象:

function* gen() {
  yield 1;
  yield 2;
  yield 3;
}

const g = gen(); g.next(); // { value: 1, done: false } g.next(); // { value: 2, done: false }

利用这一点,我们可以更简洁地实现 Symbol.iterator:

const myCollection = {
  items: ['x', 'y', 'z'],
  *[Symbol.iterator]() {
    for (const item of this.items) {
      yield item;
    }
  }
};

现在同样可以使用 for...of 遍历,代码更清晰易读。

结合使用的优势

Generator 函数天然符合迭代器协议,因此非常适合用来实现 Symbol.iterator 方法。这种方式既保持了逻辑简洁,又支持延迟计算(惰性求值),适合处理大量或无限数据流。

比如实现一个无限计数器:

const counter = {
  *[Symbol.iterator]() {
    let i = 0;
    while (true) yield i++;
  }
};

// 取前5个值 for (const num of counter) { if (num > 4) break; console.log(num); // 0,1,2,3,4 }

基本上就这些。通过 Generator 和 Symbol.iterator 的配合,JavaScript 提供了一致且灵活的遍历机制。