Работа с websockets в aiohttp, как организовать двунаправленный обмен?

Question

[Алексей]

Добрый день. Никак не соображу как мне организовать двунаправленный обмен данными через websocket соединение в aiohttp. Например вот есть такой обработчик:

import asyncio

q_transmit = asyncio.Queue()
q_receive = asyncio.Queue()

@asyncio.coroutine
def websocket_handler(request):
    ws = web.WebSocketResponse()
    ws.start(request)
    while True:
        # input_msg = yield from ws.receive()
        # yield from q_receive.put(input_msg)
        output_msg = yield from q_transmit.get()
        ws.send_str(output_msg)
    return ws

Есть очередь q_transmit через которую поступают сообщения для конкретного соединения и я их отправляю клиенту, но куда мне воткнуть закоментированные строки для приёма сообщений от клиента для меня пока загадка. Ведь в данном случае пока один из операторов yield from не завершится то второй не начнётся и получается что я могу либо принимать сообщения либо отправлять. Как сделать так что бы можно было и принимать сообщения и отправлять и делать это незавсимо?

Answer 1

[Andrey K]

import asyncio

q_transmit = asyncio.Queue()
q_receive = asyncio.Queue()


@asyncio.coroutine
def check_queue(ws):
    while True:
        output_msg = yield from q_transmit.get()
        ws.send_str(output_msg)


@asyncio.coroutine
def websocket_handler(request):
    ws = web.WebSocketResponse()
    ws.start(request)

    loop.call_soon(asyncio.async, check_queue(ws))

    while True:
        msg = yield from ws.receive()

    return ws

Comments

[Алексей]

Спасибо, кое что начало проясняться. Подскажите ещё, в докумнетации под примером есть фраза:

You must use the only websocket task for both reading (e.g yield from ws.receive()) and writing but may have multiple writer tasks which can only send data asynchronously (by yield from ws.send_str('data') for example).

Я правильно понимаю что после создания websocket соединения читать через него я могу только из той же корутины а писать могу и в других?

[Andrey K]

Да.