如果我从非异步函数调用协程并且事件循环已经在运行,是否可以从协程获取结果?

kri*_*han 5 python-asyncio python-3.6

我在将 asyncio 集成到较旧的代码库中时遇到了麻烦。其中大部分是可以管理的,但我遇到了非异步函数需要调用协程的麻烦。这似乎可以通过在相关协程上运行loop.run_until_complete()来最容易地完成。当这种情况发生在调用堆栈的顶部时(即,当我们可以保证循环尚未运行时),它工作得很好 - 协程可以调用任何其他协程。我遇到麻烦的是我们无法保证循环尚未运行。下面的(诚然有些人为的)代码说明了这一点:

import asyncio

import aioredis
from asyncio_extras import (
        contextmanager as async_contextmanager)

async def is_flag_set(redis_pool, key):
    async with acquire_redis_connection(redis_pool) as redis_connection:
        return await redis_connection.get(key)
###

async def helper_1():
    pool = await create_redis_pool()
    return await is_flag_set(pool, 'my_key')

def test_1():
    loop = asyncio.get_event_loop()
    return loop.run_until_complete(
            helper_1())

###

def helper_2(pool=None):
    loop = asyncio.get_event_loop()
    if pool is None:
        pool = loop.run_until_complete(create_redis_pool(
            db_number=0))
    return loop.run_until_complete(is_flag_set(pool, 'my_key'))

def test_2():
    return helper_2()

###

async def helper_3(db_number):
    pool = await create_redis_pool(
            db_number=db_number)
    return helper_2()


def test_3():
    loop = asyncio.get_event_loop()
    return loop.run_until_complete(
            helper_3(db_number=1))

###

# Not relevant to the question, just included
# for completeness:

@async_contextmanager.async_contextmanager
async def acquire_redis_connection(pool):
    connection = await pool.acquire()
    try:
        yield connection
    finally:
        pool.release(connection)

async def create_redis_pool(db_number=0):
    global redis_pool
    redis_pool = await aioredis.create_pool(
            address=('localhost', 6379),
            db=db_number,
            encoding='utf-8',
            minsize=5,
            maxsize=15)
    return redis_pool

if __name__ == '__main__':
    print(test_3())
Run Code Online (Sandbox Code Playgroud)

我的问题的重点是功能helper_2。当在 test_2 中调用它时,循环未运行,因此它可以安全地发出其协程。然而,当在 test_3 中调用它时,循环已经启动,我们得到了这个异常:

  File "/usr/local/lib/python3.6/asyncio/base_events.py", line 408, in run_forever
    raise RuntimeError('This event loop is already running')
RuntimeError: This event loop is already running
Run Code Online (Sandbox Code Playgroud)

我理解为什么我们会遇到异常,但我想知道是否有好的策略来管理这个问题。我在这个集成中并不经常遇到它,但我遇到过几次——主要是在尝试编写测试用例时。据我从文档和 StackOverflow 上的类似问题来看,如果您知道自己处于非异步函数中并且事件循环已经在运行,则无法从协程中“获得答案”。

Mik*_*mov 2

如果您知道自己处于非异步函数中并且事件循环已经在运行,则无法从协程“获取答案”。

您可以在不同的事件循环中运行协程。可以这样做:

global_loop = asyncio.get_event_loop()
# Change current event loop:
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
try:
    return loop.run_until_complete(coro())
finally:
    # return old state to not affect outer code:
    asyncio.set_event_loop(global_loop)
Run Code Online (Sandbox Code Playgroud)

但上面的代码是阻塞的:它不能与外部协程同时运行。这意味着除了在coro. 只有当多个协程在单个事件循环内同时运行时,才能实现任何异步优势。你应该了解清楚。

请注意,在外部事件循环运行时以阻塞方式调用协程这一事实意味着您有冻结此外部事件循环的风险。在这里阅读更多相关信息。