在 Windows 中使用信号中断 python multiprocessing.Process

Question

我有一个 python 脚本，它使用 multiprocessing.Process 类生成一个新进程。这个过程应该永远运行以监控东西。在 Unix 上，我现在可以使用 os.kill() 向该特定进程发送信号，并在该进程中使用 signal.signal(...) 来实现我的特定中断处理程序。在 Windows 上无法正常工作。

我阅读了使用popen 的方法。我也可以为 Process 类指定 CREATE_NEW_PROCESS_GROUP 标志吗？以及如何？

这是我的示例代码：

import multiprocessing as mp
import time
import signal
import os
import platform

def my_h(signal, frame):
    print("recieved signal", signal)
    raise InterruptedError

def a_task():
    signal.signal(signal.SIGINT, my_h)
    print("this is 'a_task'", os.getpid())
    try:
        while True:
            print(time.time())
            time.sleep(1)
    except Exception as e:
        print(type(e), e)
    print("'a_task' is at end")


if __name__ == '__main__':
    p = mp.Process(target=a_task)
    p.start()
    time.sleep(1)

    if platform.system() == 'Windows':
        print("send CTRL_C_EVENT")
        os.kill(p.pid, signal.CTRL_C_EVENT)
    elif platform.system() == 'Linux':
        print("send SIGINT")
        os.kill(p.pid, signal.SIGINT)

    time.sleep(3)
    try:
        os.kill(p.pid, signal.SIGTERM)
    except:
        pass

Answer 1

我找到了一种解决方法，使用multiprocessing.Event 类实现信号。

然后线索是找到在主线程中引发 KeybordInterrupt 的 interrupt_main() 方法（在 thread (Python2) 或 _thread (Python3) 中），这是我想要中断的进程。

import multiprocessing as mp
import time
import signal
import os
import threading
import _thread

def interrupt_handler(interrupt_event):
    print("before wait")
    interrupt_event.wait()
    print("after wait")
    _thread.interrupt_main()

def a_task(interrupt_event, *args):
    task = threading.Thread(target=interrupt_handler, args=(interrupt_event,))
    task.start()

    print("this is 'a_task'", os.getpid())
    try:
        while True:
            print(time.time())
            time.sleep(1)
    except KeyboardInterrupt:
        print("got KeyboardInterrupt")
    print("'a_task' is at end")


if __name__ == '__main__':
    interrupt_event = mp.Event()
    p = mp.Process(target=a_task, args = (interrupt_event, tuple()))
    p.start()
    time.sleep(2)
    print("set interrupt_event")
    interrupt_event.set()

    time.sleep(3)
    try:
        os.kill(p.pid, signal.SIGTERM)
    except:
        pass