17370845950

什么是迭代器_JavaScript中迭代器和生成器如何工作
迭代器对象必须有next()方法,它返回{value, done}对象,未结束时done为false,结束后始终返回{value: undefined, done: true},否则会导致for...of死循环。

迭代器对象必须有 next() 方法

JavaScript 中的迭代器是一个普通对象,只要它实现了 next() 方法,并且该方法返回形如 { value, done } 的对象,就符合迭代器协议。这个方法不接受参数(或只接受一个被忽略的参数),每次调用都返回下一个值和是否遍历结束的标志。

常见错误是手动实现时忘记 done: true 的终止条件,导致 for...of 死循环;或者在已结束状态后继续调用 next(),仍返回 { value: undefined, done: false } —— 正确做法是之后始终返回 { value: undefined, done: true }

示例:手动实现一个数组迭代器:

const arr = [10, 20, 30];
const iterator = {
  index: 0,
  next() {
    if (this.index < arr.length) {
      return { value: arr[this.index++], done: false };
    }
    return { value: undefined, done: true };
  }
};

Symbol.iterator 是让对象可被 for...of 遍历的关键

一个对象能否被 for...of、解构赋值、扩展运算符([...obj])消费,取决于它是否部署了 Symbol.iterator 方法,且该方法返回一个迭代器对象。

内置类型如 ArrayStringMapSet 都默认实现了它;普通对象(plain object)默认没有,所以不能直接 for...of 遍历。

如果你想让自定义对象可迭代,只需添加该方法:

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

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

生成器函数返回迭代器,不是直接执行函数体

生成器函数用 function* 声明,调用它不会立即运行函数体,而是返回一个迭代器对象 —— 这个对象遵循迭代器协议,且其 next() 方法会逐步执行函数体内带 yield 的代码段。

关键点:

  • yield 暂停执行并产出值,下次 next() 才恢复(从暂停处继续,不是重头开始)
  • 生成器内部状态(变量、执行位置)由返回的迭代器实例持有,多个调用互不影响
  • 首次 next() 启动执行,直到遇到第一个 yield 或函数结束
  • next(value) 传参,该值会成为上一个 yield 表达式的返回值(常用于协程通信)

示例:

function* countdown(n) {
  while (n > 0) {
    yield n;
    n--;
  }
}

const iter = countdown(3); console.log(iter.next()); // { value: 3, done: false } console.log(iter.next()); // { value: 2, done: false } console.log(iter.next()); // { value: 1, done: false } console.log(iter.next()); // { value: undefined, done: true }

生成器可以 yield* 委托另一个可迭代对象

yield* 不是语法糖,它会“展开”右侧的可迭代对象(即调用其 Symbol.iterator,逐个 yield 其产出的值),并把控制权完全交给那个迭代器,直到它完成。这比手动写循环更简洁,也支持递归或组合多个数据源。

注意:yield* 右侧必须是可迭代的(有 Symbol.iterator 方法的对象),否则会报 TypeError: xxx is not iterable

常见误用场景:

  • 对普通对象直接 yield* obj(没实现迭代协议 → 报错)
  • 期望 yield* 返回整个数组,实际它只是逐个产出值,不返回新数组
  • 在异步生成器中混用普通 yield*(应改用 yield* asyncGenerator(),且需配合 for await...of

示例:

function* numbers() {
  yield 1;
  yield* [2, 3]; // 展开数组
  yield* "ab";   // 展开字符串(每个字符)
}

[...numbers()]; // [1, 2, 3, 'a', 'b']

生成器的状态管理是隐式的,但非常实在 —— 每次 next() 调用都依赖前一次暂停的位置和局部变量快照。一旦忘了保存状态(比如在 yield 后重置计数器),行为就会出人意料。