Python：使用 asyncio 模块等待 2 个独立任务的结果

Question

对于 Python 3.4+，我可以使用asyncio来分派独立的任务。

现在，我有两个长期运行的独立函数。 两个函数都返回结果，那么我如何将两个结果收集到一个对象中，使用asyncio将这两个函数放在一个 IO 循环中。

import asyncio 

async def long_func1():
   await ...
   ...

async def long_func2():
   await ...
   ...

Answer 1

如果我得到你想要做的...

为了同时运行这两个函数，您可以使用asyncio.create_task和asyncio.wait来获取它们的结果。

看看下面的例子：

import sys
import asyncio

async def test_1():
    return sys._getframe().f_code.co_name

async def test_2():
    return sys._getframe().f_code.co_name


async def main():
    t1 = asyncio.create_task(test_1())
    t2 = asyncio.create_task(test_2())

    done, pending = await asyncio.wait([t1, t2])
    results = [d.result() for d in done]

    # The order of the finished function is unknown
    print(results)

if __name__ == '__main__':
    loop = asyncio.get_event_loop()
    loop.run_until_complete(main())
    loop.close()

它将运行两个函数并等待它们完成。

一旦他们done它会聚集在两者的结果done功能的results列表。