Python实现WebSocket通讯与心跳控制详解

最新推荐文章于 2025-06-10 13:58:32 发布

原创

最新推荐文章于 2025-06-10 13:58:32 发布 · 1.9k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言 #功能测试 #postman #测试工具 #自动化

为了使用Python实现WebSocket通讯和心跳控制，我们通常需要一个WebSocket客户端库和一个服务器端库。这里，我们将使用websockets库作为服务器和客户端的示例。
安装必要的库

首先，你需要安装websockets库。可以使用pip进行安装：

pip install websockets

1.服务器端代码

服务器端将处理WebSocket连接，发送和接收消息，并管理心跳检测。

import asyncio  
import websockets  
import time  
  
HEARTBEAT_INTERVAL = 10  # 心跳间隔（秒）  
HEARTBEAT_TIMEOUT = 30  # 心跳超时（秒）  
  
# 心跳检测字典，存储客户端和它们最后活动时间  
clients = {}  
  
async def heartbeat_task(ws, client_id):  
    while True:  
        try:  
            # 如果客户端在HEARTBEAT_TIMEOUT内没有发送消息，则关闭连接  
            if time.time() - clients[client_id]['last_active