下面是一个使用websocket-client
库的Python WebSocket重连示例代码:
import websocket
import time
def on_message(ws, message):
print("Received: " + message)
def on_error(ws, error):
print("Error: " + str(error))
def on_close(ws):
print("Connection closed")
def on_open(ws):
print("Connection opened")
# 连接参数设置
ws_url = "wss://example.com/ws"
headers = {
'Authorization': 'Bearer <token>',
}
while True:
try:
# 创建WebSocket连接对象并添加头部信息
ws = websocket.WebSocketApp(ws_url, header=headers)
# 设置回调函数
ws.on_open = on_open
ws.on_message = on_message
ws.on_error = on_error
ws.on_close = on_close
# 开始运行WebSocket连接并保持活动状态直到出现错误或关闭连接。
ws.run_forever()
except Exception as e:
print(f'Connection error: {e}')
# 等待5秒钟后重新连接
time.sleep(5)
在这个例子中,我们创建了一个 WebSocketApp
对象,并添加了一些头部信息和回调函数。然后,我们启动了WebSocket连接,并通过 run_forever()
方法保持它的活动状态。如果出现错误或者连接被关闭,则捕获异常并等待5秒钟后尝试重新连接。
您可以根据具体情况修改这段代码来实现自己的WebSocket重连逻辑。