Python如何在内核终止时防止多处理僵尸

Question

最近我在 Python 中使用多处理时遇到了僵尸进程的问题。

我有一个脚本，它创建了多个进程（工作人员）以及另一个进程（打印机）。工作人员会将消息生成到队列中，打印机需要打印队列中的所有消息。

问题是工人通常需要工作几个小时。但有时在启动脚本后，我想更改一些代码并重新启动脚本。如果我等到工作完成，因为我使用了 join()，所以不会有僵尸。但是如果我直接关闭内核（因为 ctrl+C 对我的 spyder 不起作用），所有进程（僵尸）仍然在工作。

我尝试使用os.ppid，但 ppid 似乎没有改变。 我试过p.daemon = True，但也没有用。

所以我想问一下有没有办法确保如果我关闭内核，所有进程都会自行终止并且系统中没有僵尸。

谢谢，示例代码是：

from multiprocessing import Process, Queue
import time

def f(num, q):
    for i in range(100):
        q.put('f: num=%d\n'% (num,))
        time.sleep(10)
def g(q, filename):
    while True:
        with open(filename, 'a') as f:
            if q.qsize() > 0:
                item = q.get()
                if item == None:
                    break
                else:
                    f.write(item)
def main():
    filename = './log.txt'
    q = Queue()
    workers = [Process(target=f, args=(i,q)) for i in range(10)]
    printer = Process(target=g, args=(q,filename))
    for p in workers:
        p.daemon = True
        p.start()
    printer.daemon = True
    printer.start()
    for p in workers:
        p.join()
    q.put(None)
    printer.join()
if __name__ == '__main__':
    main()

Answer 1

子进程没有被释放，因为父进程突然终止而没有清理的可能性。因此，daemon 标志是没有用的。

我不会终止带有 IDE 的子进程，因为它可能没有考虑此类用例。我宁愿依赖操作系统工具。

在 Windows 上，您可以将 Taskkill 命令与 /T 参数一起使用。请参阅taskkill documentation。

在 Linux 上，您可以使用带有 -P 参数的 pkill 命令。 Pkill manpage.

最后一种方法是拦截 SIGTERM 信号并将其重新路由为 SIGINT。但我不建议这样做。

import os
from signal import signal, SIGINT, SIGTERM

def handler(*_):
    os.kill(os.pid, SIGINT)

signal.signal(SIGTERM, handler)

这是一个 hack，我什至不确定它是否可以在 Windows 上运行。