蟒ASYNCIO add_done_callback與異步DEF

Question

我有2個功能：第一個，def_a，是一個異步功能，第二個是def_b這是一個普通的函數和調用的def_a結果作爲與add_done_callback功能的回調。

我的代碼如下所示：

import asyncio 

def def_b(result): 
    next_number = result.result() 
    # some work on the next_number 
    print(next_number + 1) 

async def def_a(number): 
    await some_async_work(number) 
    return number + 1 

loop = asyncio.get_event_loop() 
task = asyncio.ensure_future(def_a(1)) 
task.add_done_callback(def_b) 
response = loop.run_until_complete(task) 
loop.close() 

而且它的工作完美。

當第二個函數def_b變得異步時，問題就開始了。現在，它看起來像這樣：

async def def_b(result): 
    next_number = result.result() 
    # some asynchronous work on the next_number 
    print(next_number + 1) 

但現在我不能把它提供給add_done_callback功能，因爲它不是一個普通的功能。

我的問題是 - 如果def_b是異步的，是否有可能以及如何向add_done_callback函數提供def_b函數？

Answer 1

add_done_callback被認爲是「低級」接口。當與協同程序時，您可以chain them在許多方面，例如：

import asyncio 


async def my_callback(result): 
    print("my_callback got:", result) 
    return "My return value is ignored" 


async def coro(number): 
    await asyncio.sleep(number) 
    return number + 1 


async def add_success_callback(fut, callback): 
    result = await fut 
    await callback(result) 
    return result 


loop = asyncio.get_event_loop() 
task = asyncio.ensure_future(coro(1)) 
task = add_success_callback(task, my_callback) 
response = loop.run_until_complete(task) 
print("response:", response) 
loop.close() 

記住add_done_callback仍然會調用回調，如果你的未來會引發異常（但調用result.result()要再）。