4

我有以下场景：我有一个python服务器，在收到请求后，需要解析一些信息，尽快将结果返回给用户，然后自行清理。我尝试使用以下逻辑来设计它：

Consumer: *==*   (wait for result)   *====(continue running)=====...
              \                     / return
Producer:      *======(prase)====*=*
                                  \
Cleanup:                           *==========*

我一直在尝试使用异步任务和协程来使这种情况无效。我尝试的一切最终要么是生产者在返回之前等待清理完成，要么是返回杀死清理。理论上我可以让消费者在向用户显示结果后调用清理，但我拒绝相信 Python 不知道如何“即发即弃”并返回。

例如，这段代码：

import asyncio

async def Slowpoke():
    print("I see you shiver with antici...")
    await asyncio.sleep(3)
    print("...pation!")

async def main():
    task = asyncio.create_task(Slowpoke())
    return "Hi!"

if __name__ == "__main__":
    print(asyncio.run(main()))
    while True:
        pass

返回：

I see you shiver with antici...
Hi!

并且永远不会到达...pation。

我错过了什么？

4

4 回答 4

2

我设法使用线程而不是 asyncio 让它工作：

import threading
import time

def Slowpoke():
    print("I see you shiver with antici...")
    time.sleep(3)
    print("...pation")

def Rocky():
    t = threading.Thread(name="thread", target=Slowpoke)
    t.setDaemon(True)
    t.start()
    time.sleep(1)
    return "HI!"

if __name__ == "__main__":
    print(Rocky())
    while True:
        time.sleep(1)
于 2019-08-08T08:00:35.487 回答
2

asyncio似乎不是特别适合这个问题。您可能想要简单的线程：

这样做的原因是您的任务在父级完成时被杀死。通过在那里抛出一个daemon线程，您的任务将继续运行，直到它完成，或者直到程序退出。

import threading
import time

def Slowpoke():
    try:
        print("I see you shiver with antici...")
        time.sleep(3)
        print("...pation!")
    except:
        print("Yup")
        raise Exception()

def main():
    task = threading.Thread(target=Slowpoke)
    task.daemon = True
    task.start()
    return "Hi!"

if __name__ == "__main__":
    print(main())
    while True:
        pass
于 2019-08-08T08:03:35.173 回答
1

asyncio.run...

[...] 创建一个新的事件循环并在最后关闭它。[...]

你的 coro，包裹在task执行期间没有机会完成main.
如果您返回Task对象并打印它，您会看到它处于取消状态：

async def main():
    task = asyncio.create_task(Slowpoke())
    # return "Hi!"
    return task

if __name__ == "__main__":
    print(asyncio.run(main()))

# I see you shiver with antici...
# <Task cancelled coro=<Slowpoke() done, defined at [...]>>

在main创建和调度任务（并打印“嗨！”）后结束时，事件循环关闭，这将导致其中所有正在运行的任务被取消。

您需要保持事件循环运行，直到任务完成，例如通过await将其输入main：

async def main():
    task = asyncio.create_task(Slowpoke())
    await task
    return task

if __name__ == "__main__":
    print(asyncio.run(main()))

# I see you shiver with antici...
# ...pation!
# <Task finished coro=<Slowpoke() done, defined at [..]> result=None>
于 2019-08-08T07:50:13.407 回答
0

（我希望我确实正确理解了你的问题。ASCII 图像和文本描述在我的脑海中并不完全对应。"Hi!"是结果，"Antici..pation"是清理，对吧？我也喜欢那个音乐剧，顺便说一句）

一种可能的基于 asyncio 的解决方案是尽快返回结果。返回会终止任务，这就是为什么有必要对清理进行一劳永逸。它必须伴随着等待所有清理完成的关闭代码。

import asyncio

async def Slowpoke():
    print("I see you shiver with antici...")
    await asyncio.sleep(3)
    print("...pation!")

async def main():
    result = "Hi!"
    asyncio.create_task(Slowpoke())
    return result

async def start_stop():
    # you can create multiple tasks to serve multiple requests
    task = asyncio.create_task(main())
    print(await task)

    # after the last request wait for cleanups to finish
    this_task = asyncio.current_task()
    all_tasks = [ 
        task for task in asyncio.all_tasks()
        if task is not this_task]
    await asyncio.wait(all_tasks)

if __name__ == "__main__":
    asyncio.run(start_stop())

另一种解决方案是使用其他方法（不返回）将结果传递给等待任务，以便在解析后立即开始清理。Future 被认为是低级的，但无论如何这里是一个示例。

import asyncio

async def main(fut):
    fut.set_result("Hi!")
    # result delivered, continue with cleanup
    print("I see you shiver with antici...")
    await asyncio.sleep(3)
    print("...pation!")

async def start_stop():
    fut = asyncio.get_event_loop().create_future()
    task = asyncio.create_task(main(fut))
    print(await fut)

    this_task = asyncio.current_task()
    all_tasks = [ 
        task for task in asyncio.all_tasks()
        if task is not this_task]
    await asyncio.wait(all_tasks)

if __name__ == "__main__":
    asyncio.run(start_stop())
于 2019-08-08T09:46:18.880 回答