如何实现Trio服务器中单个任务失败时不取消所有客户端连接任务？

绝对有可行方案！核心思路就是给每个客户端处理任务套上异常捕获逻辑，把未处理的异常在任务内部消化掉，不让它冒泡到nursery层面触发全局取消。这样单个客户端的错误只会终止自身任务，完全不会牵连其他连接。

具体实现步骤 #

1. 给客户端处理函数添加全局异常捕获 #

在你处理单个客户端连接的异步函数里，用try/except块包裹所有业务逻辑，捕获所有可能的未处理异常，记录日志后正常退出任务。这样nursery不会检测到未处理异常，也就不会触发全局取消。

2. 完整代码示例 #

下面是一个可运行的Trio服务器示例，展示了如何实现异常隔离：

import trio
import logging

# 配置日志，方便定位错误详情
logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
)
logger = logging.getLogger(__name__)

async def handle_client(stream: trio.Stream) -> None:
    client_addr = None
    try:
        # 获取客户端地址，方便日志定位问题
        client_addr = stream.socket.getpeername()
        logger.info(f"Accepted connection from {client_addr}")

        # 这里替换成你的实际客户端处理逻辑
        async with stream:
            # 示例：接收客户端数据
            data = await stream.receive_some(1024)
            if not data:
                logger.info(f"Connection closed by {client_addr}")
                return
            
            # 模拟一个业务异常，测试隔离效果
            raise RuntimeError("Mock error: Failed to process client request")

    except Exception as e:
        # 保留键盘中断，保证可以正常关闭服务器
        if isinstance(e, KeyboardInterrupt):
            raise
        # 记录详细错误信息，包括堆栈追踪
        logger.error(
            f"Error handling client {client_addr or 'unknown'}: {str(e)}",
            exc_info=True
        )
    finally:
        if client_addr:
            logger.info(f"Connection with {client_addr} closed")

async def main():
    logger.info("Starting server on 0.0.0.0:8888")
    # 使用serve_tcp自动管理客户端连接的nursery
    await trio.serve_tcp(handle_client, port=8888, host="0.0.0.0")

if __name__ == "__main__":
    trio.run(main)

3. 关键细节说明 #

• 异常范围控制：捕获Exception可以覆盖绝大多数业务异常，但特意保留了KeyboardInterrupt，这样你仍然可以用Ctrl+C正常关闭服务器。
• 日志完整性：记录客户端地址和异常堆栈（exc_info=True），方便后续排查问题根源。
• 子任务异常处理：如果你的handle_client内部还启动了子任务（比如用嵌套nursery），记得给子任务也加上异常捕获，避免异常冒泡到上层任务。

为什么这个方案有效？ #

Trio的nursery取消机制只在任务抛出未被捕获的异常时才会触发。只要我们在每个客户端任务内部把异常处理掉，nursery就会认为任务是正常结束的，不会影响其他正在运行的任务。

内容的提问来源于stack exchange，提问作者Nikratio