Asyncio和无限循环

Question

@asyncio.coroutine
    def listener():
        while True:
            message = yield from websocket.recieve_message()
            if message:
                yield from handle(message)

loop = asyncio.get_event_loop()
loop.run_until_complete(listener())

假设我正在使用带有asyncio的websockets。 这意味着我收到来自websockets消息。 当我收到消息时，我想处理消息但是我的代码丢失了所有异步的东西。 因为yield from handle(message)的yield from handle(message)绝对阻塞......我怎么能找到一种方法使其无阻塞？ 比如，在同一时间处理多条消息。 在我可以处理另一条消息之前，不必等待消息处理。

谢谢。

Answer 1

如果你不关心handle消息的返回值，你可以简单地为它创建一个新的Task，它将在你的websocket阅读器旁边的事件循环中运行。 这是一个简单的例子：

@asyncio.coroutine
def listener():
    while True:
        message = yield from websocket.recieve_message()
        if message:
            asyncio.ensure_future(handle(message))

ensure_future将创建一个任务并将其附加到默认事件循环。 由于循环已经运行，它将与您的websocket阅读器并行处理。 实际上，如果它是一个运行缓慢的I / O阻塞任务（如发送电子邮件），您可以轻松地同时运行几十个句柄（消息）任务。 它们在需要时动态创建，并在完成时销毁（比生成线​​程的开销低得多）。

如果你想要更多的控制，你可以简单地写入阅读器中的asyncio.Queue，并拥有一个可以使用队列的固定大小的任务池，这是多线程或多进程编程中的典型模式。

@asyncio.coroutine
def consumer(queue):
    while True:
        message = yield from queue.get()
        yield from handle(message)

@asyncio.coroutine
def listener(queue):
    for i in range(5):
         asyncio.ensure_future(consumer(queue))
    while True:
        message = yield from websocket.recieve_message()
        if message:
            yield from q.put(message)

q = asyncio.Queue()
loop = asyncio.get_event_loop()
loop.run_until_complete(listener(q))