同步调用协程

Question

想象以下非常常见的情况：您编写了一个又长又复杂的函数，并意识到应该将一些代码提取到一个单独的函数中以便重用和/或可读性。通常，这个额外的函数调用不会改变程序的语义。

但是，现在假设您的函数是一个协程，并且您要提取的代码至少包含一个异步调用。现在将它提取到一个单独的函数中会突然改变程序的语义，方法是插入一个新点，协程在该点上产生，事件循环控制，并且可以在其间安排任何其他协程。

之前的例子：

async def complicated_func():
    foo()
    bar()
    await baz()

之后的示例：

async def complicated_func():
    foo()
    await extracted_func()

async def extracted_func():
    bar()
    await baz()

在前面的示例中，保证complicated_func 在调用foo() 和调用bar() 之间不会被挂起。重构之后，这个保证就失去了。

我的问题是：是否可以调用extracted_func() 使其立即执行，就好像它的代码是内联的？或者有没有其他方法可以在不改变程序语义的情况下执行这些常见的重构任务？

Answer 1

重构之后，这个保证就失去了。

其实不然。

是否可以调用extracted_func() 使其立即执行，就像它的代码是内联的一样？

已经是这样了。

await some_coroutine() 表示some_coroutine 可能会将控制权交还给事件循环，但在它真正等待未来（例如某些 I/O 操作）之前不会这样做。

考虑这个例子：

import asyncio

async def coro():
    print(1)
    await asyncio.sleep(0)
    print(3)

async def main():
    loop.call_soon(print, 2)
    await coro()

loop = asyncio.get_event_loop()
loop.run_until_complete(main())

注意2 如何按预期在1 和3 之间打印。

这也意味着可以通过编写这样的代码来冻结事件循环：

async def coro():
    return

async def main():
    while True:
        await coro()

在这种情况下，事件循环永远没有机会运行另一个任务。