迭代器的关门艺术：预防抽象漏洞的贴心指南

前端

2023-11-12 14:47:19

在这个抽象至上的技术世界里，我们经常会构建复杂而庞大的系统。而当系统中的组件之间存在抽象关系时，就可能会产生一种叫做抽象漏洞 的棘手问题。

想象一下，你在使用一种迭代器对象来遍历一个集合。突然，你发现了一个 bug：由于没有正确关闭迭代器，导致系统出现了一系列不可预知的错误。这就是抽象漏洞的经典例子。

为何关闭迭代器至关重要？

迭代器是一个允许你遍历集合中的元素的对象。然而，为了释放迭代器占用的资源并防止内存泄漏，在使用完成后必须关闭它。

如果不关闭迭代器，会导致以下问题：

内存泄漏： 迭代器会继续持有对集合的引用，导致集合无法被垃圾回收器回收。
意外行为： 未关闭的迭代器可能会产生不可预料的结果，比如返回错误的元素或引发异常。

关闭迭代器的正确姿势

关闭迭代器的正确方法因编程语言而异。在 JavaScript 中，使用 done() 方法即可：

const iterable = ['a', 'b', 'c'];
const iterator = iterable[Symbol.iterator]();

while (true) {
  const { value, done } = iterator.next();
  if (done) break;
  // 处理 value
}

// 释放资源
iterator.return();

在 Python 中，使用 close() 方法：

from collections import Iterator

class MyIterator(Iterator):
    def __init__(self, iterable):
        self.iterable = iterable
        self.index = 0

    def __next__(self):
        if self.index < len(self.iterable):
            item = self.iterable[self.index]
            self.index += 1
            return item
        else:
            raise StopIteration

    def close(self):
        # 释放资源
        pass

iterable = [1, 2, 3]
iterator = MyIterator(iterable)

while True:
    try:
        item = next(iterator)
        # 处理 item
    except StopIteration:
        break

# 释放资源
iterator.close()