17370845950

生成器函数如何在外部提前终止并清理资源
生成器函数可通过 throw() 或 close() 提前终止并触发清理逻辑,关键在于用 try...finally 保证 finally 中的资源释放代码必然执行,或响应 GeneratorExit 异常完成清理。

生成器函数可以通过外部调用 throw()close() 方法提前终止,从而触发清理逻辑。关键在于利用 try...finally 块或 generator.close() 触发的 GeneratorExit 异常来执行资源释放。

使用 try...finally 保证清理执行

在生成器内部用 try...finally 包裹资源操作,无论是否被外部中断,finally 块都会运行:

  • 打开文件、启动网络连接、创建子进程等耗资源操作应放在 try 前或 try
  • 关闭、释放、取消等清理动作必须写在 finally 块中
  • 即使调用 gen.close()gen.throw(GeneratorExit)finally 仍会执行

响应 close() 主动终止并清理

调用生成器对象的

close() 方法会向其抛出 GeneratorExit 异常,此时生成器必须退出(不能捕获该异常并继续 yield):

  • GeneratorExit 是一个特殊异常,不可被常规 except Exception: 捕获,需显式写 except GeneratorExit:
  • 若在 finally 外捕获了 GeneratorExit,必须在处理完后重新抛出或直接 return,否则会触发 RuntimeError
  • 推荐做法是不显式捕获 GeneratorExit,而依赖 finally 完成清理

用 throw() 发送自定义异常触发清理路径

外部可调用 gen.throw(exc_type, exc_value, traceback) 向生成器注入任意异常,适合主动中断并走特定错误处理分支:

  • 生成器内可用 except MyCustomError: 捕获并执行对应清理逻辑
  • 注意:若异常未被处理,会向上冒泡;若生成器已结束,会引发 StopIterationRuntimeError
  • 配合 try...except...finally 可实现“按需中断 + 统一清理”

实际清理示例(文件读取场景)

以下是一个安全释放文件句柄的生成器写法:

def read_lines_safely(filename):
    f = None
    try:
        f = open(filename)
        for line in f:
            yield line.strip()
    finally:
        if f is not None and not f.closed:
            f.close()  # 无论正常结束、close() 还是 throw() 都会执行

外部可随时调用 gen.close() 终止迭代并确保文件关闭。