Python客户端保持websocket打开并对收到的消息做出反应

时间:2014-12-03 12:59:44

标签: python multithreading websocket

我正在尝试在python中实现一个REST客户端,该客户端通过打开的websocket与相关服务器接收从服务器收到的消息。 这是场景:

  • 客户端使用服务器打开websocket
  • 服务器不时向客户端发送消息
  • 当客户端收到消息时,它会从中获取一些信息 服务器

我当前的客户端能够打开websocket并从服务器接收消息。但是,一旦收到消息,它就会从服务器获取信息然后终止,同时我希望让它继续监听其他消息,使其从服务器获取新内容。

以下是我的代码:

def openWs(serverIp, serverPort):
    ##ws url setting
    wsUrl = "ws://"+serverIp+":"+serverPort+"/websocket"

    ##open ws
    ws = create_connection(wsUrl)

    ##send user id
    print "Sending User ID..."
    ws.send("user_1")
    print "Sent"

    ##receiving data on ws
    print "Receiving..."
    result =  ws.recv()

    ##getting new content
    getUrl = "http://"+serverIp+":"+serverPort+"/"+result+"/entries"
    getRest(getUrl)

我不知道使用线程是否合适,我不是那么专家。 如果有人可以提供帮助,那就太好了。

提前致谢。

1 个答案:

答案 0 :(得分:3)

我完成了这段代码,做了我期待的事情。从here

获取它
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):
    def run(*args):
        for i in range(3):
            time.sleep(1)
            ws.send("Hello %d" % i)
        time.sleep(1)
        ws.close()
        print "thread terminating..."
    thread.start_new_thread(run, ())


if __name__ == "__main__":
    websocket.enableTrace(True)
    ws = websocket.WebSocketApp("ws://localhost:5000/chat",
                                on_message = on_message,
                                on_error = on_error,
                                on_close = on_close)
    ws.on_open = on_open

    ws.run_forever()