使用asyncio和aiohttp的多个非阻塞任务

时间:2017-05-12 14:09:20

标签: python async-await aiohttp

我正在尝试使用asyncio和aiohttp执行几个非阻塞任务,我不认为我这样做是有效的。我认为最好使用await而不是yield。有人可以帮忙吗?

def_init__(self):
    self.event_loop = asyncio.get_event_loop()

def run(self):
    tasks = [
        asyncio.ensure_future(self.subscribe()),
        asyncio.ensure_future(self.getServer()),]
    self.event_loop.run_until_complete(asyncio.gather(*tasks))
    try:
       self.event_loop.run_forever()

@asyncio.coroutine
def getServer(self):
    server = yield from self.event_loop.create_server(handler, ip, port)
    return server

@asyncio.coroutine
def sunbscribe(self):
    while True:
        yield from asyncio.sleep(10)
        self.sendNotification(self.sub.recieve())

def sendNotification(msg):
    # send message as a client

我必须听一个服务器并订阅收听广播,并根据广播的消息POST到另一台服务器。

1 个答案:

答案 0 :(得分:2)

根据PEP 492

  

await,类似于yield,暂停执行read_data   coroutine直到db.fetch等待完成并返回结果   数据

     

它使用实现中的收益,并进行额外的验证步骤   它的论点。等待只接受一个等待的,可以是以下之一:

所以我没有在代码中看到效率问题,因为它们使用相同的实现。

但是,我确实想知道为什么要返回server但从不使用它。

我在您的代码中看到的主要设计错误是您同时使用两者:

self.event_loop.run_until_complete(asyncio.gather(*tasks))
try:
   self.event_loop.run_forever()

从我所看到的你只需要run_forever()

一些额外提示:

在我使用asyncio的实现中,我通常会确保在出现错误时关闭循环,否则这会导致大量泄漏,具体取决于您的应用类型。

try:
    loop.run_until_complete(asyncio.gather(*tasks))
finally:  # close the loop no matter what or you leak FDs
    loop.close()

我也使用Uvloop而不是内置的,根据基准测试它更有效。

import uvloop
...
    loop = uvloop.new_event_loop()
    asyncio.set_event_loop(loop)