python结束当前进程ctrl_多处理 - 捕获Ctrl + C / SIGINT并在python中正常退出多进程

先前接受的解决方案具有竞争条件，并且不适用于pthread_sigmask和apply_async功能。

使用apply_async处理Ctrl + C / pthread_sigmask的正确方法是：

在创建进程apply_async之前，使进程忽略pthread_sigmask。 这种方式创建子进程继承map处理程序。

创建apply_async后，在父进程中还原原始的pthread_sigmask处理程序。

使用pthread_sigmask和apply_async而不是阻止map和apply。

等待结果超时，因为默认阻塞等待忽略所有信号。 这是Python bug [https://bugs.python.org/issue8296。]

把它放在一起：

#!/bin/env python

from __future__ import print_function

import multiprocessing

import os

import signal

import time

def run_worker(delay):

print("In a worker process", os.getpid())

time.sleep(delay)

def main():

print("Initializng 2 workers")

original_sigint_handler = signal.signal(signal.SIGINT, signal.SIG_IGN)

pool = multiprocessing.Pool(2)

signal.signal(signal.SIGINT, original_sigint_handler)

try:

print("Starting 2 jobs of 5 seconds each")

res = pool.map_async(run_worker, [5, 5])

print("Waiting for results")

res.get(60) # Without the timeout this blocking call ignores all signals.

except KeyboardInterrupt:

print("Caught KeyboardInterrupt, terminating workers")

pool.terminate()

else:

print("Normal termination")

pool.close()

pool.join()

if __name__ == "__main__":

main()

正如@YakovShklarov所指出的那样，在忽略信号和在父进程中取消它之间存在一段时间窗口，在此期间信号可能会丢失。 使用pthread_sigmask代替暂时阻止父进程中的信号传递可以防止信号丢失，但是，它在Python-2中不可用。