Python WebSocket长链接心跳与短链接

python websocket

安装

pip install websocket-client

 

 

先来看一下，长链接调用方式：

    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()

 

 长链接，参数介绍：

（1）url: websocket的地址。

（2）header: 客户发送websocket握手请求的请求头，{'head1:value1','head2:value2'}。

（3）on_open：在创建Websocket握手时调用的可调用对象，这个方法只有一个参数，就是该类自己。

（4）on_message：这个对象在接收到服务器返回的消息时调用。有两个参数，一个是该类自己，一个是咱们从服务器获取的字符串（utf-8格式）。

（5）on_error：这个对象在遇到错误时调用，有两个参数，第一个是该类自己，第二个是异常对象。

（6）on_close：在遇到链接关闭的状况时调用，参数只有一个，就是该类自己。

（7）on_cont_message：这个对象在接收到连续帧数据时被调用，有三个参数，分别是：类自己，从服务器接受的字符串（utf-8），连续标志。

（8）on_data：当从服务器接收到消息时被调用，有四个参数，分别是：该类自己，接收到的字符串（utf-8），数据类型，连续标志。

（9）keep_running：一个二进制的标志位，若是为True，这个app的主循环将持续运行，默认值为True。

（10）get_mask_key：用于产生一个掩码。

（11）subprotocols：一组可用的子协议，默认为空。

 

长链接关键方法：ws.run_forever(ping_interval=60,ping_timeout=5)

 若是不断开关闭websocket链接，会一直阻塞下去。另外这个函数带两个参数，若是传的话，启动心跳包发送。

 

ping_interval:自动发送“ping”命令，每一个指定的时间(秒),若是设置为0，则不会自动发送。

ping_timeout:若是没有收到pong消息，则为超时(秒)。

ws.run_forever(ping_interval=60,ping_timeout=5)

#ping_interval心跳发送间隔时间

#ping_timeout 设置，发送ping到收到pong的超时时间

 

咱们看源代码，会发现这样一断代码：

ping的超时时间，要大于ping间隔时间

 

        if not ping_timeout or ping_timeout <= 0:
            ping_timeout = None
        if ping_timeout and ping_interval and ping_interval <= ping_timeout:
            raise WebSocketException("Ensure ping_interval > ping_timeout")

 

 

 

 

长链接：

示例1：

 

import websocket try: import thread except ImportError: import _thread as 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): ws.send("hello1") time.sleep(1) ws.close() thread.start_new_thread(run,()) 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(ping_interval=60,ping_timeout=5)

 

示例2：

import websocket from threading import Thread import time import sys class MyApp(websocket.WebSocketApp): def on_message(self, message): print(message) def on_error(self, error): print(error) def on_close(self): print("### closed ###") def on_open(self): def run(*args): for i in range(3): # send the message, then wait
                # so thread doesn't exit and socket
                # isn't closed
                self.send("Hello %d" % i) time.sleep(1) time.sleep(1) self.close() print("Thread terminating...") Thread(target=run).start() if __name__ == "__main__": websocket.enableTrace(True) if len(sys.argv) < 2: host = "ws://echo.websocket.org/"
    else: host = sys.argv[1] ws = MyApp(host) ws.run_forever()

 

 

短链接：

from websocket import create_connection ws = create_connection("ws://echo.websocket.org/") print("Sending 'Hello, World'...") ws.send("Hello, World") print("Sent") print("Receiving...") result = ws.recv() print("Received '%s'" % result) ws.close()

 

——