父进程退出后如何退出子进程
我们知道当子进程推出的时候,父进程会收到 SIGCHLD 信号,从而可以采取相应的操作。但是当父进程退出的时候,系统会把子进程的父进程更改为pid=0的 init 进程,而且子进程不会收到任何信号。而我们经常想在父进程退出的时候,让子进程也推出。在 Python 中可以有如下几种做法。
设置子进程为 daemon
这里的 daemon 和系统的守护进程没有任何关系,是 quitwhenparent_dies 的意思。也就是当父进程退出的时候,会自动尝试关闭 daemon=True 的子进程。
p = multiprocessing.Process(target=foo) p.daemon = True p.start()
在子进程中设置 PDEATHSIG
在 Linux 中,进程可以要求内核在父进程退出的时候给自己发信号。使用系统调用 prctl。
prctl(PR_SET_PDEATHSIG, SIGHUP);
在 Python中也有对应的包 python-prctl,可以在子进程中这样使用,这样在父进程挂掉的时候,子进程就会收到 SIGHUP 信号:
# apt-get install build-essential libcap-dev # pip install python-prctl import signal import prctl prctl.set_pdeathsig(signal.SIGHUP)
缺点:只支持 linux
父进程在终止的时候回收子进程
可以使用 atexit.register 在主进程中注册代码:
# pip install psutil import psutil import atexit import os import signal @atexit.register def kill_children(): print("quitting, press Ctrl-C to force quit") current_process = psutil.Process() children = current_process.children(recursive=True) for child in children: print("Child pid is {}".format(child.pid)) os.kill(child.pid, signal.SIGTERM)
使用 atexit 在收到 SIGTERM 的时候并不能触发,所以最好使用 signal 注册到主进程对应的信号上。
缺点是当使用 kill -9 或者使用 os._exit 的时候不会调用这些函数。