python - 如何使用 python 多处理返回一个值？

Question

我有一个用 python 编写的回归测试脚本，它调用许多同样用 python 编写的测试脚本并执行它们。每个测试脚本都会返回 pass 或 fail。我想确保回归测试脚本不会永远等待返回值并在一段时间后停止运行的测试脚本。我遇到了这个 answer它工作得很好。除了一个小警告。 p.join() 不会从测试脚本中返回一个必要的值。我不能使用消息传递或全局变量，因为它是跨多个团队使用的庞大代码库。有人可以提出建议吗？我看过 multiprocessing文档，但我似乎找不到明确的答案。

代码在 Windows 机器上运行，python 版本为 2.7x。

Answer 1

您可以改用Pool，将直接返回辅助函数的结果:

import multiprocessing

def f():
  return "hi"

if __name__ == "__main__":
    pool = multiprocessing.Pool(1) # Just a one-process Pool.
    result = pool.apply_async(f)
    try:
        print(result.get(10)) # Wait 10 seconds
    except multiprocessing.TimeoutError:
        print("Timed out")
        pool.terminate() # Kill all (1, in this case) processes in the pool
        pool.join()