python - 进程超时内的多处理池

Question

当我使用以下代码时，池结果总是返回超时，我在做什么逻辑上不正确？

from multiprocessing import Pool, Process, cpu_count

def add(num):
  return num+1

def add_wrap(num):
  new_num = ppool.apply_async(add, [num])
  print new_num.get(timeout=3)

ppool = Pool(processes=cpu_count() )

test = Process(target=add_wrap, args=(5,)).start()

我知道这个错误，并且会认为它会在 python 2.6.4 中修复？

Answer 1

您不能在进程之间传递 Pool 对象。

如果您尝试此代码，Python 将引发异常：“NotImplementedError：池对象无法在进程之间传递或腌制”。

from multiprocessing import Queue, Pool

q = Queue()
ppool = Pool(processes=2)                                                       
q.put([ppool])
ppool = q.get()

因此，如果您希望您的代码正常工作，只需在 add_wrap 方法中创建您的 Pool 对象。

from multiprocessing import Pool, Process, cpu_count

def add(num):
  return num+1

def add_wrap(num):
  ppool = Pool(processes=cpu_count() )
  new_num = ppool.apply_async(add, [num])
  print new_num.get(timeout=3)

test = Process(target=add_wrap, args=(5,)).start()