PythonWebSocket长连接教程：asyncio与websockets详解

本文深入剖析了Python中使用websockets库实现稳定WebSocket长连接的关键实践与常见陷阱：明确指出连接“一建立就断开”的根本原因在于未启动接收循环或心跳机制，强调必须在同一线程的asyncio事件循环中直接await send/recv/ping等操作，严禁混用线程池；同时详解了高并发下的CPU飙升根源、安全重连的退避策略与资源清理要点，并对比了websockets与aiohttp在错误粒度、协议控制和生命周期管理上的本质差异，为构建健壮、可维护的实时通信服务提供了清晰、可落地的技术指南。

WebSocket连接一建立就断开？检查 await websocket.recv() 是否被阻塞

asyncio 下的 WebSocket 长连接本质是靠持续轮询或等待消息维持的，不是“连上就自动活着”。常见现象是客户端连上服务端后几秒内主动断开，抓包看到 1001 Going Away 或服务端抛 ConnectionClosedOK。根本原因往往是没启动接收循环——哪怕你只发不收，也得显式调用 recv() 或 iter 启动协程监听。

• 必须在连接建立后立即启动一个 async for message in websocket: 或 while True: await websocket.recv() 循环，否则连接会在空闲超时后被底层关闭
• 如果业务逻辑不需要收消息，也要写个空循环 + await asyncio.sleep(1) 心跳占位，但更推荐发 ping：调用 await websocket.ping() 每 30 秒一次
• websockets 默认 ping 超时是 20 秒，若网络延迟高或服务端响应慢，需显式调大：ping_timeout=30 和 ping_interval=25

并发多个 WebSocket 连接时 CPU 占用飙升？别用 loop.run_in_executor 包 websocket.send()

websockets 的 send() 和 recv() 本身已是协程，内部基于 asyncio.StreamReader/Writer，全程异步无阻塞。有人为“保险”把它丢进线程池执行，结果触发大量线程切换和事件循环争抢，CPU 直接拉满。

• 所有 websocket.* 方法（send、recv、close、ping）都必须在同一个 asyncio event loop 中直接 await，不能混用线程池
• 真正需要线程池的是：JSON 序列化大对象、读写本地文件、调用同步第三方 SDK —— 这些才该用 loop.run_in_executor(None, json.dumps, huge_data)
• 发送高频小消息（如每秒 50 条）时，注意不要在循环里反复 await websocket.send(json.dumps(...))，先批量组装再 send 一次，减少协程调度开销

如何安全地重连？别在 except 里直接 await websockets.connect()

捕获 ConnectionClosedError 或 OSError 后立刻重连，容易触发密集重试风暴，把服务端打挂。更糟的是，未清理旧连接资源（比如还在运行的 recv 任务）会导致 socket 句柄泄漏。

• 重连前务必确保旧连接已彻底关闭：调用 await websocket.close() 并等待其完成；若已异常断开，检查 websocket.closed 属性再决定是否调用
• 加退避策略：首次失败后等 1 秒，第二次 2 秒，第三次 4 秒……上限建议 30 秒，可用 asyncio.sleep(min(30, 2 ** attempt))
• 用 asyncio.create_task() 启动重连协程，而不是在主逻辑里 await，避免阻塞其他连接的处理
• 务必设重试上限（比如 5 次），超过后应记录错误并退出，而不是无限循环

为什么用 websockets 不用 aiohttp 的 WebSocket 支持？看错误码和连接粒度

两者都能做 async WebSocket，但 aiohttp 的 WS 实现绑定在 HTTP session 生命周期里，连接异常时抛出的错误类型模糊（常是泛化的 ClientError），且无法单独控制 ping/pong 行为；而 websockets 是专为 WebSocket 协议设计的库，错误分类细（ConnectionClosedError / ConnectionClosedOK / InvalidState），连接对象生命周期清晰，调试时看到 code=1006 就知道是底层连接异常，code=1008 是协议违规，不用猜。

• 如果你需要精确控制 close code、自定义 subprotocol、或要求连接对象能独立于 HTTP 请求存在（比如后台常驻连接推送），必须选 websockets
• aiohttp 更适合“HTTP 接口带一个临时 WebSocket 升级”的场景，比如单次聊天页面加载后建连，关页即断
• 注意 websockets 不内置 HTTP 服务，要提供升级接口得自己写 http:// 端点或搭配 starlette/fastapi

事情说清了就结束。最常漏掉的是：忘了在 recv 循环里包 try/except 捕获 ConnectionClosed 类异常，导致整个 task 崩溃后连接静默丢失。

好了，本文到此结束，带大家了解了《PythonWebSocket长连接教程：asyncio与websockets详解》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多文章知识！