peer-protocol

基于 aiohttp 的 WebSocket 双向通信库，支持服务端广播、客户端断线重连。

特性

• 服务端：WebSocket 服务、多客户端广播、自动清理断线连接
• 客户端：断线自动重连、TCP 探活、统一生命周期回调
• 类型提示：connect/disconnect 泛型，便于静态检查

安装

pip install peer-protocol

快速开始

服务端

import asyncio
from peer_protocol import Server

async def main():
    server = Server(host="0.0.0.0", port=8080)

    @server.on_receive
    async def _(payload):
        # 收到任意客户端消息时广播给所有客户端
        await server.broadcast(payload)

    await server.run()

if __name__ == "__main__":
    asyncio.run(main())

客户端

import asyncio
from peer_protocol import Client

async def main():
    # 需包含 /ws 路径
    client = Client(url="http://127.0.0.1:8080/ws")

    @client.on_connect
    async def _(ws):
        await client.send({"type": "hello", "message": "world"})

    @client.on_receive
    async def _(payload):
        print("收到:", payload)
        await client.stop()  # 收到消息后退出

    await client.run()

if __name__ == "__main__":
    asyncio.run(main())

API 概览

Server

Client

运行示例

# 终端 1：启动服务端
python -m examples.server

# 终端 2：启动客户端
python -m examples.client