使用asyncio同时收听两个websockets

import asyncio
import websockets

async def handle_websockets(uri1, uri2):
    async with websockets.connect(uri1) as websocket1, websockets.connect(uri2) as websocket2:
        while True:
            # 接收和处理websocket1的消息
            message1 = await websocket1.recv()
            # 处理websocket1的消息

            # 接收和处理websocket2的消息
            message2 = await websocket2.recv()
            # 处理websocket2的消息

async def main():
    uri1 = 'ws://example.com/ws1'
    uri2 = 'ws://example.com/ws2'
    await handle_websockets(uri1, uri2)

asyncio.run(main())