Python：geventwebsocket 库高级用法举例和应用详解

from gevent import monkey  # 引入gevent的monkey模块以进行IO多路复用
monkey.patch_all()  # 进行monkey patching，使标准库的socket可以异步使用

from gevent.pywsgi import WSGIServer  # 导入WSGI服务器用于处理HTTP请求
from geventwebsocket import WebSocketServer, WebSocketApplication, Resource  # 导入WebSocket相关模块

# 自定义WebSocket应用类
class MyWebSocketApp(WebSocketApplication):
    def on_open(self):  # 当WebSocket连接建立时调用
        print("WebSocket连接已打开")

    def on_message(self, message):  # 当收到消息时调用
        print(f"收到消息: {message}")
        self.send(f"服务器响应: {message}")  # 发送响应回客户端

    def on_close(self, reason):  # 当WebSocket连接关闭时调用
        print("WebSocket连接已关闭，原因:", reason)

# 创建WebSocket服务器
server = WebSocketServer(("0.0.0.0", 8000), Resource({"": MyWebSocketApp}))  # 启动WebSocket服务器
print("WebSocket服务器已启动，监听在：8000端口")
server.serve_forever()  # 让服务器一直运行

import websocket  # 导入WebSocket-client库
import threading  # 导入线程库以支持多线程运行

def on_message(ws, message):  # 处理接收到的消息
    print(f"来自服务器的消息: {message}")  # 打印接收到的消息

def on_open(ws):  # 当WebSocket连接打开时调用
    print("WebSocket连接已建立")
    ws.send("Hello, Server!")  # 发送消息到服务器

# 创建WebSocket客户端实例，并定义回调函数
ws = websocket.WebSocketApp("ws://localhost:8000",
                             on_message=on_message,
                             on_open=on_open)

# 启动WebSocket客户端线程
wst = threading.Thread(target=ws.run_forever)  # 在另一个线程中运行WebSocket
wst.start()  # 启动线程

from gevent import monkey
monkey.patch_all()

from gevent.pywsgi import WSGIServer
from geventwebsocket import WebSocketServer, WebSocketApplication, Resource

clients = []  # 用于存放连接的客户端

class BroadcastApp(WebSocketApplication):
    def on_open(self):
        clients.append(self)  # 将新客户添加到客户端列表
        print("新客户端已连接")

    def on_message(self, message):
        print(f"收到消息: {message}")
        for client in clients:  # 遍历所有连接的客户端
            client.send(f"广播消息: {message}")  # 向所有客户端发送消息

    def on_close(self, reason):
        clients.remove(self)  # 从客户端列表中移除已关闭的连接
        print("客户端已断开，原因:", reason)

server = WebSocketServer(("0.0.0.0", 8000), Resource({"": BroadcastApp}))
print("广播服务器已启动，监听在：8000端口")
server.serve_forever()