使用循环 run_in_executor 时如何使用 CTRL-C 优雅地结束异步程序

Question

以下代码需要按 3 次 CTRL-C 才能结束，我怎样才能让它只以一次结束？ （所以它在 Docker 中运行良好）

import asyncio
import time


def sleep_blocking():
    print("Sleep blocking")
    time.sleep(1000)


async def main():
    loop = asyncio.get_event_loop()
    await loop.run_in_executor(None, sleep_blocking)


try:
    asyncio.run(main())
except KeyboardInterrupt:
    print("Nicely shutting down ...")

我已经阅读了许多与 asyncio 相关的问题和答案，但还没有弄清楚这一点。第一个 CTRL-C 什么都不做，第二个打印“很好地关闭...”然后挂起。第三个 CTRL-C 打印一个丑陋的错误。

我使用的是 Python 3.9.10 和 Linux。

（编辑：每条评论更新代码@mkrieger1）

Answer 1

立即无条件退出 Python 程序的方法是调用 os._exit()。如果您的后台线程正在做一些重要的事情，这可能是不明智的。但是，以下程序会按照您的要求进行操作（python 3.10，Windows10）：

import asyncio
import time
import os


def sleep_blocking():
    print("Sleep blocking")
    time.sleep(1000)


async def main():
    loop = asyncio.get_event_loop()
    loop.run_until_complete(loop.run_in_executor(None, sleep_blocking))


try:
    asyncio.run(main())
except KeyboardInterrupt:
    print("Nicely shutting down ...")
    os._exit(42)

Answer 2

从here 我们知道实际上不可能杀死在线程执行器中运行的任务。如果我用ProcessPoolExecutor 替换默认线程执行程序，我会得到您正在寻找的行为。代码如下：

import concurrent.futures
import asyncio
import time


def sleep_blocking():
    print("Sleep blocking")
    time.sleep(1000)


async def main():
    loop = asyncio.get_event_loop()
    x = concurrent.futures.ProcessPoolExecutor()
    await loop.run_in_executor(x, sleep_blocking)


try:
    asyncio.run(main())
except KeyboardInterrupt:
    print("Nicely shutting down ...")

结果是：

$ python asynctest.py
Sleep blocking
^CNicely shutting down ...