设计模式系列：迭代器模式

2023-11-08 01:45:24

迭代器模式的原理

迭代器模式是一种设计模式，它可以让你以一种统一的方式来处理各种不同的数据结构。使用迭代器模式，你可以轻松地遍历数据结构中的元素，而无需考虑数据结构的具体实现细节。

迭代器模式的核心思想是将数据结构的遍历操作封装成一个单独的对象，即迭代器对象。迭代器对象提供了两个基本方法：

next()：返回数据结构中的下一个元素。
hasNext()：检查数据结构中是否还有下一个元素。

你可以使用这两个方法来遍历数据结构中的所有元素。例如，以下代码演示了如何使用迭代器模式来遍历一个数组：

const array = [1, 2, 3, 4, 5];
const iterator = array[Symbol.iterator]();

while (iterator.hasNext()) {
  const element = iterator.next();
  console.log(element);
}

输出：

迭代器模式的实现

迭代器模式可以通过多种方式实现。最简单的方法是使用内置的迭代器对象，如数组的 Symbol.iterator() 方法。

你也可以自己实现迭代器对象。例如，以下代码演示了如何实现一个链表的迭代器对象：

class LinkedList {
  constructor() {
    this.head = null;
    this.tail = null;
  }

  add(value) {
    const node = new Node(value);

    if (this.head === null) {
      this.head = node;
      this.tail = node;
    } else {
      this.tail.next = node;
      this.tail = node;
    }
  }

  [Symbol.iterator]() {
    let current = this.head;

    return {
      next() {
        if (current === null) {
          return { done: true };
        }

        const value = current.value;
        current = current.next;

        return { value, done: false };
      },
    };
  }
}

class Node {
  constructor(value) {
    this.value = value;
    this.next = null;
  }
}

const linkedList = new LinkedList();
linkedList.add(1);
linkedList.add(2);
linkedList.add(3);
linkedList.add(4);
linkedList.add(5);

for (const element of linkedList) {
  console.log(element);
}