websocket 代码实现

最新推荐文章于 2025-04-20 14:44:55 发布

ambition_star

最新推荐文章于 2025-04-20 14:44:55 发布

阅读量418

点赞数

分类专栏：网络编程文章标签： websocket

本文链接：https://blog.youkuaiyun.com/ambition_star/article/details/119428698

版权

网络编程专栏收录该内容

4 篇文章

订阅专栏

本文档展示了如何使用Python的websockets库实现一个简单的WebSocket服务端和客户端。服务端接收并转发用户消息，同时维护一个连接用户列表。当接收到消息时，会将消息广播给所有其他在线用户。客户端则发送用户输入的名字，并接收服务器的响应。文中提到的困惑点在于WebSocket的ping/pong机制，它们用于心跳检测，确保连接的活跃性。在没有消息交换时，服务器或客户端可以主动发送ping，对方需回应pong以确认连接仍有效。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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())