python - Python 多处理如何优雅地退出？

Question

import multiprocessing
import time

class testM(multiprocessing.Process):

    def __init__(self):
        multiprocessing.Process.__init__(self)
        self.exit = False

    def run(self):
        while not self.exit:
            pass
        print "You exited!"
        return

    def shutdown(self):
        self.exit = True
        print "SHUTDOWN initiated"

    def dostuff(self):
        print "haha", self.exit


a = testM()
a.start()
time.sleep(3)
a.shutdown()
time.sleep(3)
print a.is_alive()
a.dostuff()
exit()

我只是想知道上面的代码为什么没有真正打印“你退出”。我究竟做错了什么？如果是这样，有人可以指出正确退出的正确方法吗？（我不是指 process.terminate 或 kill）

Answer 1

您没有看到这种情况发生的原因是您没有与子进程进行通信。您正在尝试使用局部变量（父进程的本地变量）向子进程发出它应该关闭的信号。

查看有关同步原语的信息。您需要设置可以在两个进程中引用的某种信号。一旦你有了这个，你应该能够轻弹父进程中的开关并等待孩子死亡。

试试下面的代码：

import multiprocessing
import time

class MyProcess(multiprocessing.Process):

    def __init__(self, ):
        multiprocessing.Process.__init__(self)
        self.exit = multiprocessing.Event()

    def run(self):
        while not self.exit.is_set():
            pass
        print "You exited!"

    def shutdown(self):
        print "Shutdown initiated"
        self.exit.set()


if __name__ == "__main__":
    process = MyProcess()
    process.start()
    print "Waiting for a while"
    time.sleep(3)
    process.shutdown()
    time.sleep(3)
    print "Child process state: %d" % process.is_alive()