Python通过上下文管理器使用进程池multiprocessing.Pool

进程池工具multiprocessing.Pool返回的对象其实是个上下文管理器。那么我们怎么知道multiprocessing.Pool对象是上下文管理器呢？我们看查看一下multiprocessing.Pool对象的属性，可以看到__enter__和__exit__函数，因此multiprocessing.Pool对象无疑是上下文管理器。

我们看一下__enter__和__exit__函数，__exit__函数里面是调用列terminate函数，所以如果通过上下文管理器方式使用就不用显式调用terminate函数了。

def __enter__(self):    return selfdef __exit__(self, exc_type, exc_val, exc_tb):    self.terminate()

通过上下文管理器使用Pool实例：

from multiprocessing import Pooldef add(x):    return x + xif __name__ == '__main__':    args = [1, 4, 7, 11]    with Pool(processes=4) as pool:        result = pool.map(add, args, 4)    print(result)

普通方法使用Pool实例：

from multiprocessing import Pooldef add(x):    return x + xif __name__ == '__main__':    args = [1, 4, 7, 11]    pool = Pool(processes=5)    result = pool.map(add, args)    pool.terminate()    pool.join()  # terminate之后调用,否则报错    print(result)

PS: 线程池工具multiprocessing.dummy.Pool类似，也是可以用上下文管理器。

from multiprocessing.dummy import Pooldef add(x):    return x + xif __name__ == '__main__':    args = [1, 4, 7, 11]    with Pool(processes=4) as pool:        result = pool.map(add, args)    print(result)