Python使用进程池管理进程

和选用线程池来关系多线程类似，当程序中设置到多进程编程时，Python 提供了更好的管理多个进程的方式，就是使用进程池。

进程池可以提供指定数量的进程给用户使用，即当有新的请求提交到进程池中时，如果池未满，则会创建一个新的进程用来执行该请求；反之，如果池中的进程数已经达到规定最大值，那么该请求就会等待，只要池中有进程空闲下来，该请求就能得到执行。

Python multiprocessing 模块提供了 Pool() 函数，专门用来创建一个进程池，该函数的语法格式如下：

multiprocessing.Pool( processes )

其中，processes 参数用于指定该进程池中包含的进程数。如果进程是 None，则默认使用 os.cpu_count() 返回的数字（根据本地的 cpu 个数决定，processes 小于等于本地的 cpu 个数）。

注意，Pool() 函数只是用来创建进程池，而 multiprocessing 模块中表示进程池的类是 multiprocessing.pool.Pool 类。该类中提供了一些和操作进程池相关的方法，如表 1 所示。

表 1 multiprocessing 模块 Pool 类常用方法
方法名	功能
apply( func[, args[, kwds]] )	将 func 函数提交给进程池处理。其中 args 代表传给 func 的位置参数，kwds 代表传给 func 的关键字参数。该方法会被阻塞直到 func 函数执行完成。
apply_async( func[, args[, kwds[, callback[, error_callback]]]] )	这是 apply() 方法的异步版本，该方法不会被阻塞。其中 callback 指定 func 函数完成后的回调函数，error_callback 指定 func 函数出错后的回调函数。
map( func, iterable[, chunksize] )	类似于 Python 的 map() 全局函数，只不过此处使用新进程对 iterable 的每一个元素执行 func 函数。
map_async( func, iterable[, chunksize[, callback[, error_callback]]] )	这是 map() 方法的异步版本，该方法不会被阻塞。其中 callback 指定 func 函数完成后的回调函数，error_callback 指定 func 函数出错后的回调函数。
imap( func, iterable[, chunksize] )	这是 map() 方法的延迟版本。
imap_unordered( func, iterable[, chunksize] )	功能类似于 imap() 方法，但该方法不能保证所生成的结果（包含多个元素）与原 iterable 中的元素顺序一致。
starmap( func, iterable[,chunksize] )	功能类似于 map() 方法，但该方法要求 iterable 的元素也是 iterable 对象，程序会将每一个元素解包之后作为 func 函数的参数。
close()	关闭进程池。在调用该方法之后，该进程池不能再接收新任务，它会把当前进程池中的所有任务执行完成后再关闭自己。
terminate()	立即中止进程池。
join()	等待所有进程完成。

下面程序演示了进程池的创建和使用。

from multiprocessing import Pool
import time
import os

def action(name='http://c.biancheng.net'):
    print(name,' --当前进程：',os.getpid())
    time.sleep(3)
if __name__ == '__main__':
    #创建包含 4 条进程的进程池
    pool = Pool(processes=4)
    # 将action分3次提交给进程池
    pool.apply_async(action)
    pool.apply_async(action, args=('http://c.biancheng.net/python/', ))
    pool.apply_async(action, args=('http://c.biancheng.net/java/', ))
    pool.apply_async(action, kwds={'name': 'http://c.biancheng.net/shell/'})
    pool.close()
    pool.join()

程序执行结果为：

http://c.biancheng.net --当前进程： 14396
http://c.biancheng.net/python/ --当前进程： 5432
http://c.biancheng.net/java/ --当前进程： 11080
http://c.biancheng.net/shell/ --当前进程： 10944

除此之外，我们可以使用 with 语句来管理进程池，这意味着我们无需手动调用 close() 方法关闭进程池。例如：

from multiprocessing import Pool
import time
import os

def action(name='http://c.biancheng.net'):
    time.sleep(3)
    return (name+' --当前进程：%d'%os.getpid())
if __name__ == '__main__':
    #创建包含 4 条进程的进程池

    with Pool(processes=4) as pool:
        adds = pool.map(action, ('http://c.biancheng.net/python/', 'http://c.biancheng.net/java/', 'http://c.biancheng.net/shell/'))
    for arc in adds:
        print(arc)

程序执行结果为：

http://c.biancheng.net/python/ --当前进程：24464
http://c.biancheng.net/java/ --当前进程：22900
http://c.biancheng.net/shell/ --当前进程：23324

本文标题：Python使用进程池管理进程

本文地址：https://www.hosteonscn.com/4646.html