Как асинхронно отсылать сообщения WebSocket?

Question

[Bastax13]

У меня есть список вебсокет-клиентов.
Сейчас для отправки сообщения всем клиентам я использую перебор этого списка:

for client in ws_clients:
    client.write_message("hello")

Но даже при не очень большом количестве клиентов появляется существенная задержка между получением сообщения первым и последним клиентом в списке.

Можно как-нибудь отправлять сообщения асинхронно (одновременно каждому клиенту)?

Comments

[Сергей Горностаев]

send_message? Это что за метод такой? Он асинхронный?

[Bastax13]

Сергей Горностаев, ошибся. Это write_message.

Answer 1

[Сергей Горностаев]

Попробуйте так:

async def on_message(self, message):
    for client in ws_clients:
        asyncio.ensure_future(client.write_message(message))

Comments

[Roman K]

Лучше-таки это делать не в цикле, иначе весь смысл await'ов теряется. В eslint есть даже правило на await в цикле. Жаль, в питоне нет нормального подобного инструмента.

async def on_message(self, message):
    futures = [client.write_message(message) client in ws_clients]
    await asyncio.gather(*futures)

Answer 2

[Дмитрий]

from multiprocessing.dummy import Pool


def sm(client):
    client.send_message('hello')


p = Pool(100)
_ = p.map(sm, ws_clients)
p.close()
p.join()

Comments

[Сергей Горностаев]

Во-первых, как это относится к Tornado? Во-вторых, p.map(sm, ws_clients) ничем принципиально не отличается от for client in ws_clients: sm(cleint) В-третьих, p.join() заблокирует цикл событий и поставит весь сервер раком.

[Roman K]

В-третьих, пора бы уже научиться отличать треды от процессов и не плодить процессы там, где они нафиг не нужны
В-четвёртых, пора уже научиться запускать процессы/треды через asyncio (loop.run_in_executor), с релиза 3.5 прошло слишком много времени