Pythonで複数のWebSocketのクライアントを立ち上げる方法

PythonNagaoShun

Pythonで複数のWebSocketクライアントを作成するには、asyncioモジュールとwebsocketsライブラリを使用することができます。

以下は、複数のWebSocketクライアントを同時に作成し、各クライアントが接続したWebSocketサーバーからメッセージを受信する方法の例です。

import asyncio
import websockets

async def receive_messages(uri):
    async with websockets.connect(uri) as websocket:
        while True:
            message = await websocket.recv()
            print(f"Received message: {message}")

async def main():
    uris = ["wss://example.com/socket1", "wss://example.com/socket2", "wss://example.com/socket3"]
    tasks = []
    for uri in uris:
        task = asyncio.create_task(receive_messages(uri))
        tasks.append(task)
    await asyncio.gather(*tasks)

asyncio.run(main())

この例では、receive_messages関数がWebSocketサーバーに接続し、メッセージを受信するためのタスクを作成します。
main関数では、複数のWebSocketクライアントを作成するために、urisリストからURIを取得し、それぞれのURIに対してreceive_messages関数を呼び出して、タスクを作成します。最後に、asyncio.gather関数を使用して、すべてのタスクを同時に実行します。

タグ: Python, Websocket