服务端：
import asyncio
import websockets# 保存已连接的客户端列表
connected_clients = set()async def handle_websocket(websocket):# 将新的客户端添加到已连接客户端列表remote_address = websocket.remote_addressconnected_clients.add(websocket)print(f"新的客户端已连接:{remote_address}")try:# 接收客户端消息并发送回复async for message in websocket:print(f"收到客户端{remote_address}消息：{message}")# 广播发送消息给所有客户端await broadcast(message)except websockets.ConnectionClosedError:print(f"客户端{remote_address}连接关闭")except Exception as e:print(f"处理客户端时出现错误：{e}")finally:# 从已连接客户端列表中移除客户端connected_clients.remove(websocket)async def broadcast(message):# 遍历所有已连接的客户端，并发送消息for client in connected_clients:await client.send(message)async def main():# 创建WebSocket服务器并注册处理程序server = await websockets.serve(handle_websocket, "127.0.0.1", 8888)# 运行服务器await server.wait_closed()# 运行主函数
asyncio.run(main())#https://blog.csdn.net/yuanfeidexin321/article/details/131482280
客户端：
import websocketdef on_message(ws, message):print(f"收到服务端回复：{message}")def on_error(ws, error):print(f"Error: {error}")def on_close(ws):print("Connection closed")def on_open(ws):print("Connection opened")ws.send("测试一下，python版的同步客户端")websocket.enableTrace(False) # 启用调试日志ws = websocket.WebSocketApp("ws://localhost:8888",on_message=on_message,on_error=on_error,on_close=on_close,on_open=on_open)ws.run_forever()
异步客户端：
import asyncio
import websocketsasync def receive_msg(websocket):reply = await websocket.recv()print(f"收到服务端回复：{reply}")async def send_message(websocket):message = '测试消息，python版本的异步客户端'await websocket.send(message)print('发送消息：', message)await asyncio.sleep(2)async def asyncio_tasks(url):try:async with websockets.connect(url) as websocket:# 启动发送消息的任务send_task = asyncio.create_task(send_message(websocket))# 启动接收消息的任务receive_task = asyncio.create_task(receive_msg(websocket))# 等待两个任务完成await asyncio.gather(send_task, receive_task)except websockets.ConnectionClosedError as e:print(f"WebSocket连接关闭: {e}")# 示例使用
asyncio.run(asyncio_tasks("ws://127.0.0.1:8888"))
