Python多進(jìn)程池。當(dāng)工作進(jìn)程之一確定不再需要執(zhí)行更多工作時(shí),如何退出腳本?
您可以使用中的回調(diào)Pool.apply_async。
這樣的事情應(yīng)該可以為您完成工作。
from multiprocessing import Pooldef part_crack_helper(args): solution = do_job(args) if solution:return True else:return Falseclass Worker(): def __init__(self, workers, initializer, initargs):self.pool = Pool(processes=workers, initializer=initializer, initargs=initargs) def callback(self, result):if result: print 'Solution found! Yay!' self.pool.terminate() def do_job(self):for args in product(seed_str, repeat=4): self.pool.apply_async(part_crack_helper, args=args, callback=self.callback)self.pool.close()self.pool.join()print 'good bye'w = Worker(num_proc, init, [total_count])w.do_job()解決方法
mp.set_start_method(’spawn’)total_count = Counter(0)pool = mp.Pool(initializer=init,initargs=(total_count,),processes=num_proc)pool.map(part_crack_helper,product(seed_str,repeat=4))pool.close()pool.join()
因此,我有一組工作人員在做一些工作。它只需要找到一個(gè)解決方案。因此,當(dāng)一個(gè)工作進(jìn)程找到解決方案時(shí),我想停止一切。
我想到的一種方法是只調(diào)用sys.exit()。但是,由于其他進(jìn)程正在運(yùn)行,因此似乎無(wú)法正常工作。
另一種方法是檢查每個(gè)進(jìn)程調(diào)用的返回值(part_crack_helper函數(shù)的返回值),然后在該進(jìn)程上終止調(diào)用。但是,我不知道在使用該map函數(shù)時(shí)該怎么做。
我應(yīng)該如何實(shí)現(xiàn)?
