学会javascript之迭代器

学习JavaScript之迭代器

什么是迭代器

迭代器（Iterator）是一种设计模式，它是一个对象，它基于某种集合来迭代，并返回单个元素。迭代器提供了一种方法来访问集合中的元素，而不必暴露集合的内部。在JavaScript中，迭代器通常是一个包含next()方法的对象，这个方法将返回集合中的下一个元素。

如何使用迭代器

创建迭代器

要创建一个迭代器，我们需要在集合对象上定义一个next()方法，它返回一个包含两个属性的对象：value和done。value属性是集合中的下一个元素，而done属性是一个布尔值，表示是否还有更多的元素。下面是一个简单的迭代器示例，它可以遍历一个数字数组：

let arr = [1, 2, 3];
let iterator = arr[Symbol.iterator]();

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

使用迭代器

使用迭代器遍历集合的方法有很多种，下面是两个示例：

// 使用for...of循环
let arr = [1, 2, 3];
for(let value of arr) {
  console.log(value);
}

// 手动遍历
let iterator = arr[Symbol.iterator]();
let next = iterator.next();
while(!next.done) {
  console.log(next.value);
  next = iterator.next();
}

迭代器示例

下面是一个更复杂的示例，它演示如何使用迭代器来遍历树形结构。假设我们有一个Node对象，它可以表示一个树节点，每个节点都有一个value属性和一个children数组，其中包含该节点的子节点。我们可以使用递归函数来遍历每个节点，并使用迭代器将其放入一个单独的数组中。以下是示例代码：

class Node {
  constructor(value) {
    this.value = value;
    this.children = [];
  }

  addNode(node) {
    this.children.push(node);
  }

  *[Symbol.iterator]() {
    yield this.value;
    for(let child of this.children) {
      yield* child;
    }
  }
}

let root = new Node(1);
let node2 = new Node(2);
let node3 = new Node(3);
let node4 = new Node(4);
let node5 = new Node(5);
let node6 = new Node(6);

root.addNode(node2);
root.addNode(node3);
node2.addNode(node4);
node2.addNode(node5);
node3.addNode(node6);

let values = [];
for(let value of root) {
  values.push(value);
}
console.log(values); // [ 1, 2, 4, 5, 3, 6 ]

在这个示例中，我们定义了一个Node类来表示树形结构中的节点。Node类实现了迭代器接口，它通过递归调用yield*来遍历节点的所有子节点。我们还定义了一个values数组，并使用for...of循环来遍历根节点并将所有值添加到数组中。最终输出的结果是[1, 2, 4, 5, 3, 6]，这是根节点和其所有子节点的值的顺序。