回答问题

asyncio.gather](https://docs.python.org/3/library/asyncio-task.html#asyncio.gather)的[文档说

如果 return_exceptions 是False(默认),第一个引发的异常会立即传播到在gather()上等待的任务。 aws 序列中的其他等待对象不会被取消,并将继续运行。

但是,从一个简单的测试看来,如果当 return_exceptions 为False时其中一项任务引发异常,则所有其他可等待对象都将被取消(或者更准确地说,如果我不清楚术语,其他可等待对象会没有完成他们的工作):

import asyncio

async def factorial(name, number, raise_exception=False):
    # If raise_exception is True, will raise an exception when
    # the loop counter > 3
    f = 1
    for i in range(2, number + 1):
        print(f'  Task {name}: Compute factorial({i})...')

        if raise_exception and i > 3:
            print(f'  Task {name}: raising Exception')
            raise Exception(f'Bad Task {name}')

        await asyncio.sleep(1)
        f *= i
    print(f'==>> Task {name} DONE: factorial({number}) = {f}')
    return f

async def main():
    tasks = [factorial('A', 5),  # this will not be finished
             factorial('B', 10, raise_exception=True),
             factorial('C', 2)]

    try:
        results = await asyncio.gather(*tasks)
        print('Results:', results)
    except Exception as e:
        print('Got an exception:', e)

asyncio.run(main())

这段代码在做什么,只是为了更简单,它定义了 3 个任务并在它们上调用asyncio.gather()。其中一项任务在其他任务之一完成之前引发异常,而另一项任务尚未完成。

实际上,我什至无法理解文档所说的内容 - 如果引发异常并被gather上等待的任务捕获,我什至无法获得返回的结果(即使其他任务会以某种方式完成)。

我遗漏了什么,还是文档有问题?

这是用 Python 3.7.2 测试的。

Answers

正如文档中所预期的那样,我已经运行了您的代码并获得了以下输出。

  Task C: Compute factorial(2)...
  Task A: Compute factorial(2)...
  Task B: Compute factorial(2)...
==>> Task C DONE: factorial(2) = 2
  Task A: Compute factorial(3)...
  Task B: Compute factorial(3)...
  Task A: Compute factorial(4)...
  Task B: Compute factorial(4)...
  Task B: raising Exception
Got an exception: Bad Task B
  Task A: Compute factorial(5)...
==>> Task A DONE: factorial(5) = 120

怎么回事

  1. 任务A、B、C提交到队列;

  2. 所有任务都在运行,而 C 最早完成。

  3. 任务 B 引发和异常。

4、await asyncio.gater()立即返回,print('Got an exception:', e)回到屏幕。

  1. 任务A继续运行并打印“u003du003d>> Task A DONE ...”

你的测试有什么问题

正如@deceze 评论的那样,您的程序在捕获异常并返回main()后立即退出。因此,任务 A 和 C 被终止是因为整个进程死亡,而不是因为取消。

要修复它,请将await asyncio.sleep(20)添加到main()函数的末尾。

Logo

学AI,认准AI Studio!GPU算力,限时免费领,邀请好友解锁更多惊喜福利 >>>

更多推荐