PHP前端开发

如何利用进程组信号优雅地终止父进程及其所有子进程?

百变鹏仔 5天前 #Python
文章标签 进程

使用进程组信号解决程序中断问题

在多进程处理中,有时需要通过发送信号来终止父进程。然而,如果子进程仍在运行,常规的信号发送方式可能无法将其一起终止。要解决此问题,需要使用进程组信号。

进程组

进程组是指父进程及其所有子进程的集合。向进程组发送信号时,信号会传递给组中的所有进程。要获取进程组id,可以使用 os.getpgid 函数。

修改代码

修改后的 a.py 文件:

import multiprocessingimport osimport signalimport timedef process1():    while true:        print("子进程运行中")        time.sleep(1)if __name__ == "__main__":    a = multiprocessing.process(target=process1)    a.daemon = true    a.start()    child_pid = a.pid    parent_pid = os.getpid()    # 获取并存储进程组id    parent_pid_group = os.getpgid(parent_pid)    # 在创建子进程后注册信号处理函数    signal.signal(signal.sigterm, handle_signal)    with open("crawler.pid", "w") as f:        f.write(str(parent_pid_group))    a.join()    print("父进程pid:", parent_pid)    print("子进程pid:", child_pid)

修改后的 b.py 文件:

import osimport signalwith open("crawler.pid", "r") as f:    try:        pid = int(f.read())        os.killpg(pid, signal.SIGTERM)        print("Signal sent successfully to process", pid)    except Exception as e:        print("Error sending signal:", e)

在修改后的代码中,我们首先获取进程组id并将其存储在变量 parent_pid_group 中。然后,我们将在创建子进程后注册信号处理函数。最后,我们在 b.py 文件中使用 os.killpg 发送信号给进程组,而不是单独的子进程。

这些修改确保当向父进程发送信号时,信号也会传递给子进程,从而实现同时终止父进程和子进程的目的。