优雅处理 WebSocket 断开与清理方法

本文深入剖析了 WebSocket 客户端在遭遇 Ctrl+C（SIGINT）中断时无法发送关闭通知的典型痛点，揭示了 asyncio.CancelledError 被异常机制忽略、清理逻辑失效的根本原因，并给出一套切实可行的优雅断连方案：精准捕获 CancelledError 实现中断时主动通信、依托 async with 确保资源自动释放、分层处理 ConnectionClosedOK/ConnectionClosedError 等连接状态，让客户端能在被强制终止前向服务端发送关键业务消息（如“client cancelled”），真正实现双向可控、状态一致、无资源泄漏的连接生命周期管理——原来一次看似简单的 Ctrl+C，背后竟藏着异步编程中协程调度、异常传播与资源清理的精密协同。

本文详解 WebSocket 客户端在接收 SIGINT（Ctrl+C）中断信号后无法正常发送关闭消息的问题，提供基于 `asyncio.CancelledError` 捕获、`async with` 自动资源管理及异常分层处理的完整解决方案。

在使用 websockets 库构建异步 WebSocket 通信时，一个常见却易被忽视的问题是：当用户通过 Ctrl+C 中断客户端程序时，KeyboardInterrupt 会触发 asyncio 事件循环的取消操作，进而导致当前正在运行的协程（如 ws.recv()）被抛出 asyncio.CancelledError —— 而非 KeyboardInterrupt 或其他常规异常。由于该异常是协程取消机制的内部信号，它会直接跳过 except Exception: 分支，强制进入 finally 块，且若此时连接已处于半关闭状态，await ws.send(...) 将阻塞或抛出 ConnectionClosedError，致使清理逻辑失效。

要真正实现“中断即通知、退出即通信”，需遵循三个关键原则：

1. 显式捕获 asyncio.CancelledError：它是 asyncio 协程被取消时唯一保证抛出的异常，必须单独处理；
2. 使用 async with websockets.connect(...)：自动确保连接在退出时调用 ws.close()，避免资源泄漏与状态不一致；
3. 区分连接关闭类型：优先捕获 websockets.exceptions.ConnectionClosedOK（对端正常关闭）与 ConnectionClosedError（异常断连），再兜底处理其他错误。

以下是修复后的客户端完整示例：

import websockets
import asyncio

async def keep_recv():
    async with websockets.connect('ws://localhost:22235') as ws:
        try:
            while True:
                try:
                    response = await ws.recv()
                    print(f"Received from server: {response}")
                    # 模拟长任务：此处不应用 create_task + sleep，应直接 await
                    await asyncio.sleep(10)  # ✅ 正确：阻塞当前协程 10 秒
                except websockets.exceptions.ConnectionClosedOK:
                    print("Connection closed cleanly by server")
                    break
                except websockets.exceptions.ConnectionClosedError:
                    print("Connection closed unexpectedly")
                    break
                except asyncio.CancelledError:
                    print("Coroutine cancelled (e.g., Ctrl+C received)")
                    await ws.send("client cancelled")  # ✅ 在取消时主动通知服务端
                    break
                except Exception as e:
                    print(f"Unexpected error: {type(e).__name__}: {e}")
                    break
        finally:
            # async with 已隐式调用 close()，此处为冗余保护，可省略
            pass

if __name__ == "__main__":
    asyncio.run(keep_recv())

⚠️ 注意事项：

• asyncio.create_task(asyncio.sleep(10000)) 是错误写法——它创建了独立任务但未 await，导致主循环无法感知其完成，且 sleep(10000) 实际是 10 秒（单位为秒，非毫秒）；应直接 await asyncio.sleep(10)。
• 服务端无需额外修改，但建议为其 handler 添加异常防护，例如：

  async def handler(ws, path):
      try:
          async for message in ws:
              print(message)
              await ws.send("received: " + message)
      except websockets.exceptions.ConnectionClosedOK:
          print("Client disconnected normally")
      except websockets.exceptions.ConnectionClosedError:
          print("Client disconnected abnormally")

总结而言，WebSocket 的优雅中断不依赖于“等待响应”，而在于主动控制协程生命周期。通过精准捕获 CancelledError、合理使用上下文管理器、并明确区分连接状态，即可确保 Ctrl+C 触发时，客户端仍能向服务端发送最后一条业务消息（如 "client cancelled"），服务端亦能可靠接收并响应，从而实现双向可控的连接终止流程。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《优雅处理 WebSocket 断开与清理方法》文章吧，也可关注golang学习网公众号了解相关技术文章。