线程化、非阻塞的 websocket 客户端

Question

我想要运行 Python 中的一个程序，它每秒通过网络套接字向 Tornado 服务器发送一条消息。我一直在使用 websocket-client 上的示例；

这个例子不起作用，因为ws.run_forever()会停止while循环的执行。

有人可以给我一个示例，说明如何将其正确实现为线程 class，我既可以调用其发送方法，也可以接收消息吗？

import websocket
import thread
import time

def on_message(ws, message):
    print message

def on_error(ws, error):
    print error

def on_close(ws):
    print "### closed ###"

def on_open(ws):
    pass

if __name__ == "__main__":
    websocket.enableTrace(True)
    ws = websocket.WebSocketApp("ws://echo.websocket.org/", on_message = on_message, on_error = on_error, on_close = on_close)
    ws.on_open = on_open
    ws.run_forever()

    while True:
        #do other actions here... collect data etc.
        for i in range(100):
            time.sleep(1)
            ws.send("Hello %d" % i)
        time.sleep(1)

Answer 1

他们的 github page 中有一个例子就是这样做的。看起来你是从那个例子开始的，把每秒发送消息的代码从 on_open 中取出并粘贴到 [=30= 之后] 调用，BTW 运行直到套接字断开连接。

也许您对这里的基本概念有疑问。总会有一个线程专用于侦听套接字（在这种情况下，主线程进入 run_forever 等待消息的循环）。如果您想进行其他事情，则需要另一个线程。

下面是他们的示例代码的不同版本，其中没有使用主线程作为 "socket listener"，而是创建了另一个线程和 run_forever 跑到那里。我认为它有点复杂，因为您必须编写代码来确保套接字已连接，同时您可以使用 on_open 回调，但也许它会帮助您理解。

import websocket
import threading
from time import sleep

def on_message(ws, message):
    print message

def on_close(ws):
    print "### closed ###"

if __name__ == "__main__":
    websocket.enableTrace(True)
    ws = websocket.WebSocketApp("ws://echo.websocket.org/", on_message = on_message, on_close = on_close)
    wst = threading.Thread(target=ws.run_forever)
    wst.daemon = True
    wst.start()

    conn_timeout = 5
    while not ws.sock.connected and conn_timeout:
        sleep(1)
        conn_timeout -= 1

    msg_counter = 0
    while ws.sock.connected:
        ws.send('Hello world %d'%msg_counter)
        sleep(1)
        msg_counter += 1

线程化、非阻塞的 websocket 客户端

Threaded, non-blocking websocket client

python

multithreading

websocket