使用Python的werkzeug.wsgiClosingIterator()函数来停止迭代器

werkzeug.wsgiClosingIterator() 是 Werkzeug 库中的一个函数，用于创建一个迭代器包装器，该迭代器可以在完成迭代时执行清理操作。这在一些特定情况下非常有用，比如在迭代大量数据时自动关闭资源。

该函数的签名如下：

werkzeug.wsgiClosingIterator(app_iter, callback=None)

参数解释：

- app_iter: 要包装的迭代器。

- callback（可选）: 迭代器完成时要执行的回调函数。

下面是使用werkzeug.wsgiClosingIterator()函数的一个简单示例：

from werkzeug.wsgi import ClosingIterator

class MyIterator:
    def __init__(self, n):
        self.n = n
        self.count = 0
    
    def __iter__(self):
        return self
    
    def __next__(self):
        if self.count >= self.n:
            raise StopIteration
        self.count += 1
        return self.count
    
    def close(self):
        print("Closing the iterator")

def my_app(environ, start_response):
    my_iter = MyIterator(5)
    start_response('200 OK', [('Content-type', 'text/plain')])
    return ClosingIterator(my_iter, callback=my_iter.close)

if __name__ == "__main__":
    from werkzeug.serving import run_simple
    run_simple('localhost', 8000, my_app)

上面的示例中，定义了一个自定义迭代器MyIterator，它返回从 1 到指定数量n的数字。当迭代完毕时，将会触发迭代器的close()方法，即打印一条关闭迭代器的消息。my_app函数使用ClosingIterator将自定义的迭代器包装起来，并将其作为返回值。这意味着在响应返回给客户端后，my_iter.close()方法将会被自动调用。

这里使用 Werkzeug 提供的run_simple()方法来启动一个简单的 Web 服务器，监听本地 8000 端口。当访问这个服务器时，将会返回从 1 到 5 的数字。在服务器终止时，即按下“Ctrl + C”停止运行时，my_iter.close()方法会被调用。

通过上述示例可以看出，werkzeug.wsgiClosingIterator()函数简化了在迭代完成后执行清理操作的过程，有助于提高代码的可读性和可维护性。