腾讯云开发实践：美股 WebSocket 行情自动重连机制实现

import websocket
import json
import time

class CloudUSStockWSClient:
    def __init__(self, ws_url, symbols):
        self.ws_url = ws_url
        self.symbols = symbols
        self.ws_app = None
        self.retry_count = 0
        self.max_retry = 10

    def on_message(self, ws, msg):
        data = json.loads(msg)
        print("云环境实时行情：", data)

    def on_close(self, ws, code, msg):
        print("云网络连接断开，启动重连流程")
        self.reconnect()

    def on_open(self, ws):
        print("云连接建立，恢复行情订阅")
        self.retry_count = 0
        sub_msg = {"action": "subscribe", "symbols": self.symbols}
        ws.send(json.dumps(sub_msg))

    def reconnect(self):
        if self.retry_count >= self.max_retry:
            print("重连超限，触发云监控告警")
            return
        wait = min(2 ** self.retry_count, 30)
        time.sleep(wait)
        self.retry_count += 1
        self.connect()

    def connect(self):
        self.ws_app = websocket.WebSocketApp(
            self.ws_url,
            on_open=self.on_open,
            on_message=self.on_message,
            on_close=self.on_close
        )
        self.ws_app.run_forever(ping_interval=10)

# 云环境初始化（适配AllTick API）
if __name__ == "__main__":
    WS_URL = "wss://api.alltick.co/stock/ws"
    TARGET_SYMBOLS = ["AAPL", "MSFT", "GOOGL"]
    client = CloudUSStockWSClient(WS_URL, TARGET_SYMBOLS)
    client.connect()