本文介绍在结构化并发场景下，如何通过取消作用域（cancelscope）和 nursery 机制主动中断正在运行的 `async with` 块，实现连接异常时的上下文强制退出与异常传播。
在异步编程中，async with 提供了优雅的资源生命周期管理，但其默认行为是“被动等待”——即只有当内部协程自然结束或抛出异常时，__aexit__ 才会被调用。然而，在真实网络服务（如 WebSocket、长连接 RPC）中，连接可能在后台被对端静默关闭，此时我们需要主动通知并强制终止当前上下文，让外层 except 能及时捕获 ConnectionLostException 等业务异常。
直接在回调函数中“抛出异常到外部协程”是不可能的（Python 协程不支持跨栈异常注入），正确解法是利用结构化并发原语：通过取消正在运行的协程任务，触发其自然退出，并在 __aexit__ 或 finally 中统一收口、补发异常。
✅ 推荐方案：使用 trio.open_nursery() + 后台健康检查
trio 的 nursery 是目前最健壮的解决方案——它天然支持多任务协同取消，并能自动聚合并发异常为 ExceptionGroup（Python 3.11+）或 MultiError（Trio）。以下是生产就绪的实现：
import trio
from contextlib import asynccontextmanager
class SomeServiceConnection:
def __init__(self, service):
self.service = service
async def connect(self):
# 模拟连接逻辑（不依赖回调）
await self.service.connect()
async def disconnect(self):
self.service.disconnect()
async def wait_disconnected(self):
# 阻塞直到连接断开（由底层驱动实现，如监听 socket EOF）
await self.service.wait_for_disconnect()
@asynccontextmanager
async def get_service_connection(service):
connection = SomeServiceConnection(service)
await conn
ection.connect()
try:
async with trio.open_nursery() as nursery:
# 启动后台任务：监听连接状态，断开时主动 raise 异常
nursery.start_soon(connection.wait_disconnected)
yield connection
# 正常退出时，显式取消 nursery（可选，yield 后自动 cancel）
nursery.cancel_scope.cancel()
finally:
await connection.disconnect()
? 关键点解析：nursery.start_soon(connection.wait_disconnected) 启动一个独立任务，该任务会在连接断开时完成（或抛出异常）；一旦该任务因异常结束（如 RuntimeError("connection lost")），整个 nursery 会立即取消所有子任务（包括 yield 后仍在运行的 send_stuff 循环）；nursery 自动将该异常包装进 ExceptionGroup 并重新抛出至 async with 外层，使 except ConnectionLostException: 可精准捕获；finally 中的 disconnect() 仍会执行，确保资源清理不被跳过。
⚠️ 注意事项与替代方案
不要手动调用 cancel() 在回调里：像 connection_state_callback(state) 这类外部回调无法访问当前 nursery 或 CancelScope，强行调用会导致取消目标错误或静默失败。
-
若不用 Trio？可用 asyncio + asyncio.Event 模拟（兼容性更强）：
@asynccontextmanager async def get_service_connection_asyncio(service): connection = SomeServiceConnection(service) disconnect_event = asyncio.Event() await connection.connect(lambda: disconnect_event.set()) # 注册回调设事件 try: async with asyncio.TaskGroup() as tg: tg.create_task(_monitor_disconnect(disconnect_event)) yield connection finally: await connection.disconnect() async def _monitor_disconnect(event: asyncio.Event): await event.wait() raise ConnectionLostException("Connection lost unexpectedly") 异常类型建议自定义：继承 Exception 并命名清晰（如 ConnectionLostException），便于外层区分业务异常与系统错误。
✅ 总结
强制退出异步上下文管理器的本质，不是“向协程注入异常”，而是通过结构化并发工具（nursery / TaskGroup / CancelScope）协调任务生命周期，在连接失效时触发受控取消，再由上下文管理器统一收口并抛出语义明确的异常。这既符合结构化并发原则，也保证了资源安全与错误可追溯性。推荐优先采用 trio 的 nursery 方案；若项目受限于 asyncio，则配合 Event + TaskGroup（Python 3.11+）或 create_task + cancel() 实现等效逻辑。
