「WebSockets」Python实现服务器端与客户端通信

Table of Contents

前言

在阅读一个老外的项目的时候，发现使用了WebSockets协议来做命令转发、数据转发。下文将会贴一段代码，实现的功能是：

创建一个本地服务器，接收客户端的信息，将客户端的信息发送回去。
创建一个客户端，客户端发送数据到服务器。

服务器 Server

import asyncio
import websockets

# 接收客户端消息并处理，这里只是简单把客户端发来的返回回去
async def recv_msg(websocket):
    while True:
        recv_text = await websocket.recv()
        response_text = f"your submit context: {recv_text}"
        await websocket.send(response_text)

# 服务器端主逻辑
# websocket和path是该函数被回调时自动传过来的，不需要自己传
async def main_logic(websocket, path):
    await recv_msg(websocket)

# 把ip换成自己本地的ip
start_server = websockets.serve(main_logic, '192.168.16.25', 5678)
asyncio.get_event_loop().run_until_complete(start_server)
asyncio.get_event_loop().run_forever()

客户端 Client

import asyncio
import websockets

# 向服务器端发送认证后的消息
async def send_msg(websocket):
    while True:
        _text = input("please enter your context: ")
        if _text == "exit":
            print(f'you have enter "exit", goodbye')
            await websocket.close(reason="user exit")
            return False
        await websocket.send(_text)
        recv_text = await websocket.recv()
        print(f"{recv_text}")

# 客户端主逻辑
async def main_logic():
    async with websockets.connect('ws://192.168.16.25:5678') as websocket:

        await send_msg(websocket)

asyncio.get_event_loop().run_until_complete(main_logic())

发表评论取消回复