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

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

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

def __enter__(self):
    return self

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

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

from multiprocessing import Pool


def add(x):
    return x + x


if __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 Pool


def add(x):
    return x + x


if __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 Pool


def add(x):
    return x + x


if __name__ == '__main__':
    args = [1, 4, 7, 11]

    with Pool(processes=4) as pool:
        result = pool.map(add, args)

    print(result)