JavaScript迭代器模式_自定义遍历逻辑

JavaScript中通过实现Symbol.iterator方法可自定义对象遍历行为,使其支持for...of和扩展运算符;该方法需返回一个包含next()的迭代器对象,next()返回{value, done},如ReverseArray类实现反向遍历;生成器函数function*能简化迭代器编写,自动满足迭代协议,适用于无限序列如fibonacci;还可为自定义集合(如仅含偶数的EvenCollection)设计遍历逻辑,灵活控制访问顺序,适用于流数据或树结构等场景。

JavaScript中的迭代器模式能让我们自定义对象的遍历行为,让数据结构更灵活地支持for...of循环和扩展运算符。核心在于实现一个符合迭代协议的对象:拥有一个返回迭代器的[Symbol.iterator]()方法,而该迭代器需具备next()方法,返回包含valuedone属性的结果对象。

实现可迭代对象

要让一个对象可被for...of遍历,必须在原型或实例上定义Symbol.iterator方法。

例如,创建一个反向遍历数组的可迭代对象:

class ReverseArray {
  constructor(arr) {
    this.items = [...arr];
  }

[Symbol.iterator]() { let index = this.items.length - 1; return { next: () => { if (index >= 0) { return { value: this.items[index--], done: false }; } else { return { done: true }; } } }; } }

const reverseArr = new ReverseArray([1, 2, 3]); for (const item of reverseArr) { console.log(item); // 输出:3, 2, 1 }

生成器简化迭代器编写

使用生成器函数(function*)可以大幅简化迭代器的实现,因为生成器自动返回符合迭代协议的对象。

比如,构建一个无限斐波那契数列的可迭代对象:

function* fibonacci() {
  let a = 0, b = 1;
  while (true) {
    yield a;
    [a, b] = [b, a + b];
  }
}

const fib = fibonacci(); console.log(fib.next().value); // 0 console.log(fib.next().value); // 1 console.log(fib.next().value); // 1 console.log(fib.next().value); // 2

// 或者用 for...of 取前几项 for (const n of fibonacci()) { if (n > 100) break; console.log(n); }

自定义集合类的遍历逻辑

在封装数据结构时,可以通过迭代器暴露特定的访问顺序。比如一个只遍历偶数的集合:

class EvenCollection {
  constructor(numbers) {
    this.numbers = numbers.filter(n => n % 2 === 0);
  }

[Symbol.iterator]() { let idx = 0; const values = this.numbers; return { next() { if (idx < values.length) { return { value: values[idx++], done: false }; } return { done: true }; } }; } }

const evens = new EvenCollection([1, 2, 3, 4, 5, 6]); for (const n of evens) { console.log(n); // 2, 4, 6 }

基本上就这些。通过实现Symbol.iterator,你可以控制任何对象的遍历方式,结合生成器还能写出更清晰、可复用的逻辑。这种模式在处理流式数据、树结构遍历或领域特定集合时特别有用。不复杂但容易忽略细节,比如done的判断和value的返回时机。