websocket 代码实现

#服务端
import asyncio
import websockets

websocket_users = {}    #将websocket对象保存，方便后续转发数据


async def recv_user_msg(websocket):
    recv_text = await websocket.recv()
    print("recv_text:", websocket.pong, recv_text)
    # for r in range(10):
    #     time.sleep(2)
    #     await websocket.send("*****************************")
    text = recv_text.split(':')
    if text[0] not in websocket_users:
        websocket_users[text[0]] = websocket
        print(websocket_users)
    else:
        response_text = f"{text[0]}: {text[1]}"
        # await websocket_users[text[0]].send(response_text)
        print(websocket_users)
        for i, r in websocket_users.items():
            if i != text[0]:
                await r.send(response_text)


# 服务器端主逻辑
async def run(websocket, path):
    while True:
        try:
            # await check_user_permit(websocket)
            await recv_user_msg(websocket)
        except websockets.ConnectionClosed:
            print("ConnectionClosed...", path)  # 链接断开
            print("websocket_users old:", websocket_users)
            for k in websocket_users.copy():
                if websocket_users[k] == websocket:
                    websocket_users.pop(websocket)
            print("websocket_users new:", websocket_users)
            break
        except websockets.InvalidState:
            print("InvalidState...")  # 无效状态
            break
        except Exception as e:
            print("Exception:", e)


if __name__ == '__main__':
    print("127.0.0.1:8181 websocket...")
    asyncio.get_event_loop().run_until_complete(websockets.serve(run, "127.0.0.1", 8181))
    asyncio.get_event_loop().run_forever()

#客户端
import asyncio
import websockets


async def hello():
    uri = "ws://localhost:8181"
    async with websockets.connect(uri) as websocket:
        name = input("What's your name? ")

        await websocket.send(name)
        print(f"> {name}")

        greeting = await websocket.recv()
        print(f"< {greeting}")


asyncio.get_event_loop().run_until_complete(hello())