怎么实现websocket里on_message和on_open 一一对应

在 WebSocket 中，每当客户端发送一条消息到服务器，服务器收到消息后就会触发 `on_message` 回调函数。而 `on_open` 回调函数则是在客户端连接到服务器时触发。因此，`on_message` 和 `on_open` 并不是一一对应的函数。

如果你想在 `on_message` 中访问 `on_open` 中的一些变量或状态，可以考虑将这些变量或状态定义在 WebSocket 对象的属性中，这样在 `on_message` 中就可以通过 `self` 参数访问到这些属性。具体实现可以参考下面的代码示例：

import asyncio
import websockets

class MyWebSocketServer:
    def __init__(self):
        self.connected = False

    async def on_open(self, websocket, path):
        self.connected = True
        print('WebSocket connection established')

    async def on_message(self, websocket, message):
        if not self.connected:
            return

        # Do something with the message
        print(f'Received message: {message}')

    async def start(self):
        async with websockets.serve(self.on_open, 'localhost', 8765):
            await asyncio.Future()  # Keep running the server indefinitely

server = MyWebSocketServer()
asyncio.get_event_loop().run_until_complete(server.start())

在上面的示例中，`connected` 属性用于标记 WebSocket 是否已连接，它会在 `on_open` 中设置为 `True`，在 `on_message` 中进行检查。通过这种方式，我们可以保证只有已连接的 WebSocket 才能接收到并处理消息。