使用 asyncio.wait() 优化 WebSocket 广播性能

import websockets
import cv2
import asyncio
import time

def predict(image):
    # 模拟耗时的数据处理过程
    time.sleep(0.1)
    return "test"

async def echo(websocket, path):
    global vidCap, i
    while True:
        ret, image = vidCap.read()
        if ret:
            start = time.time()
            result = predict(image)
            # 使用 asyncio.wait() 进行异步广播
            await asyncio.wait([ws.send(result) for ws in clients])
            end = time.time()
            print("exec time：%f s" % (end - start))
        else:
            # 视频读取完毕，退出循环
            break

async def handler(websocket, path):
    clients.add(websocket)
    try:
        await echo(websocket, path)
    finally:
        clients.remove(websocket)

async def serve():
    start_server = await websockets.serve(handler, "localhost", 8765)
    await start_server.wait_closed()

if __name__ == '__main__':
    vidCap = cv2.VideoCapture(0) # 使用摄像头
    clients = set()
    asyncio.run(serve())

import websockets
import asyncio
import time

async def get_result(uri):
    async with websockets.connect(uri) as websocket:
        while True:
            try:
                start = time.time()
                recv_text = await websocket.recv()
                print(recv_text)
                end = time.time()
                print("exec：%f s" % (end - start))
            except websockets.exceptions.ConnectionClosed:
                print("Connection closed")
                break
            except Exception as e:
                print(f"Error: {e}")
                break

if __name__ == '__main__':
    asyncio.run(get_result("ws://127.0.0.1:8765/ws"))