Python实现websocket服务端类的代码

以下是一个使用 Python 实现的简单 WebSocket 服务端类的示例代码：

import asyncio
import websockets

class WebSocketServer:
    def __init__(self, host, port):
        self.host = host
        self.port = port

    async def handle_client(self, websocket, path):
        # 处理连接建立后的逻辑
        try:
            while True:
                message = await websocket.recv()
                # 处理接收到的消息
                await websocket.send("Server received: " + message)
        except websockets.exceptions.ConnectionClosedOK:
            # 连接关闭时的逻辑
            print("Client disconnected")

    async def start(self):
        # 创建 WebSocket 服务器并绑定处理函数
        server = await websockets.serve(self.handle_client, self.host, self.port)

        print(f"Server started at ws://{self.host}:{self.port}")

        # 保持服务器运行状态
        await server.wait_closed()

# 创建 WebSocketServer 实例并启动服务器
ws_server = WebSocketServer('localhost', 8000)
asyncio.run(ws_server.start())

这个示例代码使用了 `websockets` 库来实现 WebSocket 服务端。在 `handle_client` 方法中，你可以编写处理连接建立后的逻辑，通过 `await websocket.recv()` 接收客户端发送的消息，并通过 `await websocket.send()` 发送消息给客户端。在连接关闭时，可以在 except 块中编写关闭连接时的逻辑。

你可以根据自己的需求修改 `handle_client` 方法中的逻辑，例如处理不同的消息类型、广播消息等。启动服务器时，创建一个 WebSocketServer 实例并调用 `start` 方法即可。