Python WebSocket 长连接到底怎么写才稳？

最近在做实时行情相关的项目，刚开始觉得很简单：连上、收消息、处理数据就完事了。结果线上跑几天后发现各种问题：

网络抖动导致连接断开
服务端主动踢连接
长时间没消息被中间设备断开
重连太频繁把自己服务器打挂

后来重新整理了一套比较稳的方案，核心有 3 个：

心跳
重连
退避机制

1.心跳不能省

WebSocket 建立连接不一定就一直在线。运营商网络、NAT 设备、负载均衡器都可能清理长时间无流量的连接。所以一定要做心跳。如果服务端支持 ping/pong ，直接使用协议层心跳即可。如果不支持，可以定时发送业务心跳消息。例如：

async def heartbeat(ws):

    while True:

        try:

            await ws.ping()

            await asyncio.sleep(30)

        except Exception:

            break

30 秒左右发一次通常就够用了。

2.重连是标配

WebSocket 断开是常态，不是异常。所以不要把代码写成：

await websocket.recv()

然后指望它永远不掉线。我觉得应该是：

while True:

    try:

        await connect()

    except Exception as e:

        print(e)

    await asyncio.sleep(5)

连接断开以后自动重新建立连接。这样即使服务端重启或者网络闪断，也能自动恢复。

3.别直接疯狂重连

一个常见错误可能是⬇️

断线：

while True:

    try:

        connect()

    except:

        pass

结果：

CPU 飙高
日志刷爆
服务端压力暴增

这时候就需要退避机制，也就是每次失败后逐渐增加等待时间。例如：

delay = 1

while True:

    try:

        await connect()

        delay = 1

    except Exception:

        await asyncio.sleep(delay)

        delay = min(delay \* 2, 60)

等待时间变成 1s 、2s 、4s 、8s 、16s 、32s 、60s...

连接恢复后再重置。这样会稳定很多。

4.WebSocket 行情订阅

连接建立后订阅需要的品种即可，以 BTC 为例：

import asyncio

import json

import websockets

WS\_URL = "wss://quote.alltick.io/quote-b-ws-api"

API\_TOKEN = "YOUR\_API\_TOKEN"

async def subscribe():

    async with websockets.connect(WS\_URL) as ws:

        sub\_msg = {

            "cmd\_id": 22002,

            "seq\_id": 1,

            "trace": "test",

            "data": {

                "symbol\_list": \[

                    {

                        "code": "BTCUSDT"

                    }

                \]

            },

            "token": API\_TOKEN

        }

        await ws.send(json.dumps(sub\_msg))

        while True:

            msg = await ws.recv()

            print(msg)

asyncio.run(subscribe())

如果是生产环境，我一般会把：

连接管理
心跳任务
重连逻辑
退避机制

全部封装到一个 ConnectionManager 里面。业务层只负责消费消息，不关心连接状态。这样后面维护会轻松很多。