Python WebSocket实时数据获取：四种常见实现方式对比

Python WebSocket实时数据获取方案对比

第一种方法是使用create_connection进行链接。该方法需要预先通过pip install websocket-client安装依赖库。需要注意的是，此方法并不被推荐，因为链接状态不够稳定，容易出现断连情况，且建立连接的过程较为耗时。

import time
from websocket import create_connection

url = 'wss://i.cg.net/wi/ws'
while True: # 一直链接，直到连接上就退出循环
  time.sleep(2)
  try:
    ws = create_connection(url)
    print(ws)
    break
  except Exception as e:
    print('连接异常：', e)
    continue
while True: # 连接上，退出第一个循环之后，此循环用于一直获取数据
  ws.send('{"event":"subscribe", "channel":"btc_usdt.ticker"}')
  response = ws.recv()
  print(response)

第二种方法运行效果表现良好，连接过程相对容易实现，且获取数据的速度较快，适合对实时性有较高要求的场景。

import json
from ws4py.client.threadedclient import WebSocketClient


class CG_Client(WebSocketClient):

  def opened(self):
    req = '{"event":"subscribe", "channel":"eth_usdt.deep"}'
    self.send(req)

  def closed(self, code, reason=None):
    print("Closed down:", code, reason)

  def received_message(self, resp):
    resp = json.loads(str(resp))
    data = resp['data']
    if type(data) is dict:
      ask = data['asks'][0]
      print('Ask:', ask)
      bid = data['bids'][0]
      print('Bid:', bid)


if __name__ == '__main__':
  ws = None
  try:
    ws = CG_Client('wss://i.cg.net/wi/ws')
    ws.connect()
    ws.run_forever()
  except KeyboardInterrupt:
    ws.close()

第三种方式在逻辑上与第一种方法类似，仅改变了代码的写法结构。然而，其实际运行效果并不理想，同样存在连接耗时较长以及容易断开的缺点。

import websocket

while True:
  ws = websocket.WebSocket()
  try:
    ws.connect("wss://i.cg.net/wi/ws")
    print(ws)
    break
  except Exception as e:
    print('异常：', e)
    continue
print('OK')
while True:
  req = '{"event":"subscribe", "channel":"btc_usdt.deep"}'
  ws.send(req)
  resp = ws.recv()
  print(resp)

第四种方法的运行效果尚可，能够满足基本需求。需要注意的是，在调用run_forever函数时，内部包含多个参数配置，开发者需要根据具体场景自行调整和优化这些参数。

import websocket


def on_message(ws, message): # 服务器有数据更新时，主动推送过来的数据
  print(message)


def on_error(ws, error): # 程序报错时，就会触发on_error事件
  print(error)


def on_close(ws):
  print("Connection closed ……")


def on_open(ws): # 连接到服务器之后就会触发on_open事件，这里用于send数据
  req = '{"event":"subscribe", "channel":"btc_usdt.deep"}'
  print(req)
  ws.send(req)


if __name__ == "__main__":
  websocket.enableTrace(True)
  ws = websocket.WebSocketApp("wss://i.cg.net/wi/ws",
                on_message=on_message,
                on_error=on_error,
                on_close=on_close)
  ws.on_open = on_open
  ws.run_forever(ping_timeout=30)

以上即为本文介绍的全部内容，希望这些内容能对读者的学习与实践有所帮助。同时也欢迎大家继续支持本站，获取更多技术分享。