如何将已在循环中运行的阻塞函数的协程/任务添加到循环中

2022-03-25 00:00:00 python python-asyncio

问题描述

希望下面的代码解释了我希望比问题标题做得更好的是什么。

import asyncio
import time

loop = asyncio.get_event_loop()

class Foo(object):
    def __init__(self, num):
        self.num = num
    @property
    def number(self):
        # Somehow need to run get_number() in the loop.
        number = self.get_number()
        return number
    @asyncio.coroutine
    def get_number(self):
        yield from loop.run_in_executor(None, time.sleep, self.num)
        return self.num


@asyncio.coroutine
def runner(num):
    print("getting Foo({})".format(num))
    foo = Foo(num)
    yield from asyncio.sleep(1)
    print("accessing number property of Foo({})".format(num))
    number = foo.number
    print("Foo({}) number is {}".format(num, number))


tasks = [
    asyncio.async(runner(3)),
    asyncio.async(runner(1)),
    ]
go = loop.run_until_complete(asyncio.wait(tasks))

我不知道如何处理number函数中的注释。我试过各种各样的方法,但实际上我只是"往墙上扔*,希望有东西能坚持下去"。

这是this question的后续内容。我想在不执行yield from的情况下访问该属性,因为我需要从模板(例如mako)访问该属性,并且随处写入yield from并不理想(考虑到mako可能正在阻塞,这甚至是不可能的)。在一个完美的世界里,我会用reify decorator运行所有这些。

如果我要使用yield from,代码将非常简单。

class Foo(object):
    def __init__(self, num):
        self.num = num
    @property
    @asyncio.coroutine
    def number(self):
        yield from loop.run_in_executor(None, time.sleep, self.num)
        return self.num


@asyncio.coroutine
def runner(num):
    print("getting Foo({})".format(num))
    foo = Foo(num)
    yield from asyncio.sleep(1)
    print("accessing number property of Foo({})".format(num))
    number = yield from foo.number
    print("Foo({}) number is {}".format(num, number))

#getting Foo(3)
#getting Foo(1)
#accessing number property of Foo(3)
#accessing number property of Foo(1)
#Foo(1) number is 1
#Foo(3) number is 3

我在该主题上找到this answer,但我看不到添加完成回调如何与我的工作流配合使用。


解决方案

您所要求的是不可能的,因为当您在主线程中(您想要调用foo.number而不使用Year)时,您需要显式地将控制权交还给主循环。这正是yield from所做的。

否则,您需要在一个单独的线程中运行调用foo.number的函数,该线程将能够阻塞(不产生)并等待get_number的结果,而不阻塞主循环

相关文章