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()方法接收服务器的响应。