python - 在等待信号量时收到 SIGTERM 后，python 进程如何优雅地退出？

Question

我有一个 Python 进程，它使用多处理模块生成 5 个其他 Python 进程。我们将父进程称为 P0，将其他进程称为 P1-P5。要求是，如果我们向 P0 发送 SIGTERM，它应该先关闭 P1 到 P5，然后自行退出。

问题是 P1 和 P5 正在等待信号量。因此，当我向这些进程发送 SIGTERM 时，它们会调用信号处理程序并退出。但是因为他们在等待信号量，所以他们抛出异常。有没有办法在退出前捕捉到那个异常，让P0到P5可以优雅退出？

回溯:

Traceback (most recent call last):
  File "/usr/lib64/python2.7/multiprocessing/process.py", line 258, in _bootstrap
Traceback (most recent call last):
Process Process-2:
  File "/usr/lib64/python2.7/multiprocessing/process.py", line 258, in _bootstrap
Traceback (most recent call last):
self.run()
File "/usr/lib64/python2.7/multiprocessing/process.py", line 114, in run
self._target(*self._args, **self._kwargs)
Process Process-5:
Traceback (most recent call last):
File "/usr/lib64/python2.7/multiprocessing/process.py", line 258, in _bootstrap
  self.run()
File "/usr/lib64/python2.7/multiprocessing/process.py", line 114, in run
  self._target(*self._args, **self._kwargs)
File "/opt/fireeye/scripts/mip/StaticAnalysisRunner.py", line 45, in run
  qsem.acquire()

Answer 1

您可以安装一个信号处理程序，它会抛出一个异常，然后在子进程中捕获该异常以优雅地处理退出。

这是一个脚本示例，它在子进程的信号量中等待，并在发送 SIGTERM 时正常终止。

#!/usr/bin/env python

import signal
import time
import multiprocessing

class GracefulExit(Exception):
    pass


def signal_handler(signum, frame):
    raise GracefulExit()


def subprocess_function():
    try:
        sem = multiprocessing.Semaphore()
        print "Acquiring semaphore"
        sem.acquire()
        print "Semaphore acquired"

        print "Blocking on semaphore - waiting for SIGTERM"
        sem.acquire()
    except GracefulExit:
        print "Subprocess exiting gracefully"


if __name__ == "__main__":

    # Use signal handler to throw exception which can be caught to allow
    # graceful exit.
    signal.signal(signal.SIGTERM, signal_handler)

    # Start a subprocess and wait for it to terminate.
    p = multiprocessing.Process(target=subprocess_function)
    p.start()

    print "Subprocess pid: %d" % p.pid

    p.join()

此脚本的运行示例如下:

$ ./test.py 
Subprocess pid: 7546
Acquiring semaphore
Semaphore acquired
Blocking on semaphore - waiting for SIGTERM
----> Use another shell to kill -TERM 7546
Subprocess exiting gracefully

没有来自子进程的回溯，流程显示子进程以优雅的方式退出。这是因为 SIGTERM 被子进程信号处理程序捕获，该信号处理程序抛出可以在进程内部处理的正常 Python 异常。