从已取消的任务中获取结果

Question

我有一个程序，大致像下面的例子。 一个任务正在收集一些值并将它们返回给调用者。 有时任务可能会被取消。 在这些情况下，我仍然希望获得迄今为止任务收集的结果。 因此我捕获了CancelledError 异常，清理并返回完成的结果。

async def f():
    results = []
    for i in range(100):
        try:
            res = await slow_call()
            results.append(res)
        except asyncio.CancelledError:
            results.append('Undecided')
    return results

def on_done(task):
    if task.cancelled():
        print('Incomplete result', task.result()
    else:
        print(task.result())

async def run():
    task = asyncio.create_task(f())
    task.add_done_callback(on_done)

问题是任务取消后返回的值在任务中似乎不可用。 调用 task.result() 只会重新抛出 CancelledError。打电话给task._result 就是None。

有没有办法获取已取消任务的返回值，假设它有一个？

编辑：我现在意识到捕获CancelledError 会导致任务根本没有被取消。 这给我留下了另一个难题：我如何向任务所有者发出信号，表明这个结果只是“一半”的结果，而任务真的被取消了。 我想我可以添加一个额外的返回值来表明这一点，但这似乎与任务取消系统的整个想法背道而驰。

这里有什么好的方法建议吗？

Answer 1

我离理解用例还有很长的路要走，但以下内容对我来说是明智的：

import asyncio

async def fn(results):
    for i in range(10):
        # your slow_call
        await asyncio.sleep(0.1)
        results.append(i)

def on_done(task, results):
    if task.cancelled():
        print('incomplete', results)
    else:
        print('complete', results)

async def run():
    results = []
    task = asyncio.create_task(fn(results))
    task.add_done_callback(lambda t: on_done(t, results))
    # give fn some time to finish, reducing this will cause the task to be cancelled
    # you'll see the incomplete message if this is < 1.1
    await asyncio.sleep(1.1)

asyncio.run(run())

在run 中使用add_done_callback 和sleep 感觉很尴尬，让我觉得我不明白你在做什么。也许将包含更多调用代码的内容发布到https://codereview.stackexchange.com 将有助于获得更好的构建事物的方法的想法。请注意，与 asyncio 内置库（IMO 过早标准化）相比，还有其他库（如 trio）为 Python 协程提供更好的接口

Answer 2

我不认为这是可能的，因为在我看来，与取消任务的含义相冲突。 您可以在 slow_call 中实现类似的行为，方法是触发 CancelledError，在函数中捕获它，然后返回您想要的任何内容。