带有异步计时器的Python异步websocket客户端

Question

我需要有一个长时间运行的websocket客户端，它接收来自websocket服务器的推送消息，并且我需要监视客户端的连接状态：如果连接中断，我需要发现。

我的方法是定期记录一个常量字符串，如果没有检测到日志消息，就触发警报。

我的想法是：1）有一个websocket客户机来响应不定期传入的消息。2）同时有一个循环，当websocket客户机抛出一个ConnectionClosed exeption时停止记录消息。

我对新的3.5异步语法很感兴趣。这个websocket实现是专门基于Asyncio的。文件里的客户和我需要的一模一样。

但是，我不知道如何添加第二个coroutine来执行日志记录语句，并在websocket连接抛出ConnectionClosed时以某种方式停止。

下面是开始对话的内容，但这不起作用，因为alive方法阻塞了事件循环。我正在寻找的是一个优雅的解决方案，以同时运行这两种方法。

#!/usr/bin/env python

import asyncio
import logging

import websockets

logger = logging.getLogger(__name__)

is_alive = True


async def alive():
    while is_alive:
        logger.info('alive')
        await asyncio.sleep(300)


async def async_processing():
    async with websockets.connect('ws://localhost:8765') as websocket:
        while True:
            try:
                message = await websocket.recv()
                print(message)

            except websockets.exceptions.ConnectionClosed:
                print('ConnectionClosed')
                is_alive = False
                break


asyncio.get_event_loop().run_until_complete(alive())
asyncio.get_event_loop().run_until_complete(async_processing())

Answer 1

实际上，run_until_complete在这里被阻塞了，因为它会一直等到alive完成。

你可以用2步来解决：

1. 调度与asyncio.ensure_future（立即运行而不等待结果）对应，每个返回任务。
2. 使用Asyncio.wait
等待任务完成

tasks = [
   asyncio.ensure_future(alive()),
   asyncio.ensure_future(async_processing())
]
asyncio.get_event_loop().run_until_complete(asyncio.wait(tasks))

asyncio.get_event_loop().run_until_complete(asyncio.wait([   
   alive(),
   async_processing()
]))