WebSocket协议是在2008年由Web应用程序设计师和开发人员创建的，目的是为了在Web浏览器和服务器之间提供更高效、更低延迟的双向通信。它允许客户端和服务器在任何时候发送消息，无需重新建立TCP连接。WebSocket可以在Web浏览器和服务器之间传输文本和二进制数据，使得构建实时Web应用程序变得更加简单。

在Python中使用WebSocket

Python中有多个库可以帮助我们使用WebSocket，如：websockets、aiohttp等。在本文中，我们将使用websockets库来演示WebSocket编程。要安装websockets库，你可以使用pip：

pip install websockets

创建WebSocket服务器

使用websockets库，我们可以轻松地创建一个WebSocket服务器。以下是一个简单的示例：

import asyncio
import websockets

async def echo(websocket, path):
    async for message in websocket:
        print(f"Received message: {message}")
        await websocket.send(f"Echo: {message}")

start_server = websockets.serve(echo, "localhost", 8765)

asyncio.get_event_loop().run_until_complete(start_server)
asyncio.get_event_loop().run_forever()

在这个示例中，我们定义了一个名为echo的协程函数，它接收两个参数：websocket和path。该函数使用async for循环读取客户端发送的消息，并将消息发送回客户端。

然后，我们使用websockets.serve()函数创建一个WebSocket服务器，监听本地主机的8765端口。最后，我们使用asyncio的事件循环启动服务器。

连接上服务器直接推消息

我这里有一个需求，当一个客户端连接上服务器后，立马向客户端源源不断的发送消息，就可以这样改造：

import asyncio
import time
import websockets


async def echo(websocket, path):
    print(f"websocket echo: {path} , {websocket}")
    # await websocket.send(f"Echo: 11111")
    message = 1
    while True:
        await websocket.send(f"Echo: {message}")
        print(f"Echo: {message}")
        message += 1
        time.sleep(1)


start_server = websockets.serve(echo, "localhost", 8765)
print(f"websocket listen：ws://localhost:8765")
asyncio.get_event_loop().run_until_complete(start_server)
asyncio.get_event_loop().run_forever()

使用Postman测试

可以先使用postman创建一个websocket连接测试一下：

 

创建WebSocket客户端

要创建一个WebSocket客户端，我们同样可以使用websockets库。以下是一个简单的客户端示例：

import asyncio
import websockets

async def main():
    async with websockets.connect("ws://localhost:8765") as websocket:
        message = "Hello, server!"
        await websocket.send(message)
        print(f"Sent: {message}")

        response = await websocket.recv()
        print(f"Received: {response}")

asyncio.run(main())

在这个示例中，我们使用websockets.connect()函数建立与WebSocket服务器的连接。然后，我们使用send()方法向服务器发送消息，并使用recv()方法接收服务器的响应。