当任何一个进程完成时,如何终止池?

时间:2018-10-12 14:52:55

标签: python multiprocessing pool

我已按照建议使用以下代码,但在第一个工作人员找到匹配项后,池仍未终止。理想的结果是让20个进程运行函数f,并且当它们中的任何一个找到匹配项时,池都应终止,而实际上,我必须等到所有工作人员都找到匹配项为止。

from multiprocessing import Pool
import numpy as np


def f(z):..Output omitted


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

    def callback(self, result):
        if result:
            self.pool.terminate()

    def do_job(self):

        self.pool.map_async(f,range(1,20), callback=self.callback)
        self.pool.close()
        self.pool.join()


if __name__=="__main__":
    w = Worker(20)
    w.do_job()

0 个答案:

没有答案