Question

我有一个python池进程，如果在我希望退出池执行的任何一个进程中发生异常

我加入了池中的所有进程，因此join等待每个进程完成。如果我在目标函数中引发sys.exit（1），系统会继续无限等待，因为连接仍在等待进程完成。

如何在代码

中使用join时退出执行

from multiprocessing import Pool
import time
import sys

def printer(ip):
    try:
        for _ in xrange(5):
            print ip+str(_)
            time.sleep(1.0)
    except Exception as e:
        print e
        sys.exit(2)

def test():
    pool = Pool(processes=2)
    for i in ["hello",5]:
        result = pool.apply_async(printer,(i,))
    pool.close()    
    pool.join()
    print "good bye"

test()

Answer 1

只需返回父进程的操作状态，并使用回调来对失败做出反应。

import time
from multiprocessing import Pool


def printer(ip):
    try:
        for _ in xrange(5):
            print ip+str(_)
            time.sleep(1.0)
        return True
    except Exception as e:
        print e
        return False


class Worker():
    def __init__(self):
        self.pool = Pool(processes=2)

    def callback(self, result):
        if not result:
            print "Error raised in child process. Terminating.."
            self.pool.terminate()

    def do_job(self):
        for i in ["hello", 5]:
            self.pool.apply_async(printer, (i,), callback=self.callback)
        self.pool.close()
        self.pool.join()
        print "good bye"


def test():
    w = Worker()
    w.do_job()

如果任何人有异常，如何退出所有已连接的进程 - Python

1 个答案: