ExitStack 资源怎么配置或排查

当资源数量由配置或输入决定时，硬编码多层 with 很快会变得笨重。ExitStack 的正确用法不是“设置一个资源参数”，而是在每次资源成功获取后立即登记清理动作：上下文管理器交给 enter_context()，普通释放函数交给 callback()，离开 with 时按登记的相反顺序执行。

官方文档：https://docs.python.org/3/library/contextlib.html

排查 ExitStack 时先问三个问题：资源是否真的注册成功、栈是否进入了 with 生命周期、清理动作是否被 pop_all() 移交。大多数“没有关闭”的问题都落在这三处。

先按资源类型选择登记方法

enter_context(cm) 会调用上下文管理器的 __enter__()，并把它的 __exit__() 放入栈中，返回值就是资源本身。文件、锁和自定义上下文管理器通常走这条路径。对只有 close()、release() 或删除动作的对象，则用 callback() 显式登记。

from contextlib import ExitStack
from io import StringIO

def build_resources(paths):
    # 获取一个资源就登记一次，后续获取失败时前面的资源仍会回滚
    with ExitStack() as stack:
        files = [stack.enter_context(open(path, encoding="utf-8")) for path in paths]
        buffer = StringIO()
        stack.callback(buffer.close)  # 普通函数没有 __exit__，使用 callback
        return [file.read() for file in files]
        # 离开 with 后按 buffer、文件的逆序执行清理

列表推导式中途打开失败时，已经打开的文件会被栈回收。注意，ExitStack 不会因为对象被垃圾回收就自动执行回调；脱离 with 时要显式调用 close()，否则登记动作只是留在栈里。

push、callback 和 pop_all 不要混用

callback(fn, ...) 适合固定参数的释放函数，回调不会收到异常三元组，也不能通过返回真值抑制异常。push(exit) 更接近注册 __exit__()，可以覆盖上下文管理器进入过程的一部分，也允许退出函数参与异常处理。

pop_all() 不会执行任何清理，它把当前回调栈转移给一个新的 ExitStack。常见场景是“全部打开成功后再把关闭责任交给调用方”：失败留在原 with 中自动回滚，成功则保存新栈的 close 方法。

from contextlib import ExitStack

def open_all_or_rollback(paths):
    # 只有全部资源准备好，才把清理责任交给返回值
    with ExitStack() as stack:
        handles = [stack.enter_context(open(path, "rb")) for path in paths]
        release_all = stack.pop_all().close
        return handles, release_all
        # pop_all 后原 stack 为空，调用方最终负责 release_all()

如果“资源已经分配，但后续校验失败”也必须释放，可在校验前用 push() 登记回滚；校验通过后再 pop_all()。这比用一个 cleanup_needed 布尔变量更不容易漏掉异常路径。

资源没有释放时按生命周期排查

还要看清理顺序：ExitStack 采用后进先出，后获取的连接、临时文件或锁会先释放。如果资源之间有依赖，登记顺序就必须从外层依赖到内层资源。进入阶段的异常也要单独看，enter_context() 成功返回前未登记完成的资源，不会凭空出现在栈中。

异步资源要换 AsyncExitStack

同步 ExitStack 不能替代异步清理。异步上下文管理器使用 AsyncExitStack.enter_async_context()，异步释放函数用 push_async_callback()，最后由 async with 或 await stack.aclose() 收尾。同步文件可以留在同步栈，不能把需要 await 的 aclose() 当成普通 callback。

from contextlib import AsyncExitStack

async def use_connections(connectors):
    # 每个连接由异步栈登记，任何一个连接失败都会释放已成功连接的对象
    async with AsyncExitStack() as stack:
        connections = [
            await stack.enter_async_context(connector())
            for connector in connectors
        ]
        return len(connections)
        # async with 退出时会等待异步清理动作完成

常见问题

ExitStack 实例化后会自动关闭文件吗？

不会。它只保存回调，必须通过 with、close() 或资源接收方的关闭方法触发清理。

什么时候用 push 而不是 callback？

需要让清理函数接收异常信息、参与异常抑制，或要覆盖 __enter__() 的部分过程时用 push；普通无参释放优先用 callback。

pop_all 后原来的 stack 还能清理资源吗？

不能。回调已经转移到新栈，原栈为空；应保存新栈的 close 或新栈对象，并明确由谁负责调用。

ExitStack 能管理异步连接吗？

使用 AsyncExitStack，并配合 enter_async_context、push_async_callback 和 aclose。