python学习笔记——multiprocessing 多进程组件 进程池Pool

1 进程池Pool基本概述

在使用Python进行系统管理时,特别是同时操作多个文件目录或者远程控制多台主机,并行操作可以节约大量时间,如果操作的对象数目不大时,还可以直接适用Process类动态生成多个进程,几十个尚可,若上百个甚至更多时,手动限制进程数量就显得特别繁琐,此时进程池就显得尤为重要。

进程池Pool类可以提供指定数量的进程供用户调用,当有新的请求提交至Pool中时,若进程池尚未满,就会创建一个新的进程来执行请求;若进程池中的进程数已经达到规定的最大数量,则该请求就会等待,直到进程池中有进程结束,才会创建新的进程来处理该请求。

进程池不用频繁创建和销毁进程

2 进程池Pool的语法

Pool([processes[, initializer[, initargs[, maxtasksperchild[, context]]]]])

processes:使用的工作进程的数量;若processes是None,默认适用os.cpu_count()返回的数量。

initializer:若initializer是None,则每一个工作进程在开始的时候就会调用initializer(*initargs)。

maxtasksperchild:工作进程退出前可以完成的任务数,完成后用一个新的工作进程来替代原进程,让闲置的资源释放,maxtasksperchild默认是None,此意味只要Pool存在工作进程就一直存活

context: 用在制定工作进程启动时的上下文,一般使用multiprocessing.Pool()或者一个context对象的Pool()方法来创建一个池,两种方法都适当的设置了context。

如果主进程退出,则进程池中的所有进程均退出。

使用Pool创建进程池对象,同时进程池中进程已经启动,向进程池中添加事件时,事件排队执行。

实例方法:

p为进程池对象

p.apply():

 apply(func[, args=()[, kwds={}]]) 

该函数用于传递不定参数,主进程会被阻塞直到函数执行结束,实际上这也就说所谓的同步执行。

同步执行,按照加入进程池的顺序执行事件,每次执行完一个再执行另一个,无法获取返回值

p.apply_async()

apply_async(func[, args=()[, kwds={}[, callback=None]]])

与apply用法一样,但它是非阻塞且支持结果返回进行回调;实际上也就是异步执行。

异步执行,同时启动进程池中多个进程执行事件,可以获取事件返回值 — <multiprocessing.pool.ApplyResult object at 0x7f7f6e4357f0>

p.map()

map(func, iterable[, chunksize=None])

Pool类中的map方法,与内置map函数用法基本一致,它融合了map函数和apply_async()函数的功能;它会使进程阻塞直到返回结果。

注意:虽然第二个参数是一个迭代器,但实际应用中,必须在整个队列就绪后,程序才会运行子进程。

p.close():关闭进程池,阻止更多的任务提交到进程池Pool,待任务完成后,工作进程会退出

p.terminate():结束工作进程,不再处理未完成的任务

p.join():等待工作线程的退出,必须在close()或terminate()之后使用,因被终止的进程需要被父进程调用wait(join等价于wait),否则进程会成为僵尸进程。

注意:

(1)使用Pool创建进程池对象,同时进程池中进程已经启动

(2)向进程池对象中添加事件,事件排队执行

(3)如果主进程退出,则进程池中所有进程都退出

3 实例

3.1 基础实例

import multiprocessing as mp

def test():
    pass

p = mp.Pool(processes = 5) # 创建5条进程

for i in range(10):
    p.apply_async(test) # 向进程池添加任务

p.close() # 关闭进程池,不再接受请求
p.join() # 等待所有的子进程结束

说明:

(1)进程池Pool被创建出来后, p.apply_async(test) 语句不停地循环执行,相当于向进程池中提交了10个请求,它们会被放到一个队列中。

(2) p = mp.Pool(5) 执行完毕后创建了5条进程,但尚未给它们分配各自的任务;也就意味着,无论有多少任务,实际的进程数只有5条,每次最多5条进程并行。

(3)当Pool中有进程任务执行完毕后,这条进程资源会被释放,Pool会按先进先出的原则取出一个新的请求给空闲的进程继续执行。

(4)当Pool所有的进程任务完成后,会产生5个僵尸进程,如果主进程/主线程不结束,系统不会自动回收资源,需要调用join函数负责回收。

(5)在创建Pool进程池时,若不指定进程的最大数量,默认创建的进程数为系统的内核数量

(6)如果采用p.apply(test)阻塞方式添加任务,其每次只能向进程池中添加一条任务,然后for循环会被阻塞等待,直到添加的任务被执行完毕,进程池中的5个进程交替执行新来的任务,此时相当于单进程。——该语句需要再深刻理解,尚未完全明白

参考:python的multiprocessing模块进程创建、资源回收-Process,Pool

3.2 apply方式添加任务

import  multiprocessing as mp
import os
from time import sleep

def worker(msg):
    print(os.getpid())
    sleep(2)
    print(msg)
    return msg

#创建进程池对象
p = mp.Pool(processes = 4)#创建4条进程

pool_result = []
for i in range(10):
    msg = 'hello-%d'%i
    r = p.apply(worker,(msg,))    #向进程池中添加事件,该语句为同步执行,    #没有返回值,这种方法用的比较少   r = p.apply(worker,(msg,))    pool_result.append(r)
#获取事件函数的返回值
for r in pool_result:
    print('return:',r) 

p.close()# 关闭进程池,不再接受请求,不能再向里面添加事件
p.join() # 等待进程池中的事件执行完毕,回收进程池

运行

8419
hello-0
8418
hello-1
8420
hello-2
8421
hello-3
8419
hello-4
8418
hello-5
8420
hello-6
8421
hello-7
8419
hello-8
8418
hello-9
return: hello-0
return: hello-1
return: hello-2
return: hello-3
return: hello-4
return: hello-5
return: hello-6
return: hello-7
return: hello-8
return: hello-9

这段代码运行较慢,和进程阻塞有关。相当于单线程!

当将代码(22行)中的 print('return:',r)  修改为 print('return:',r.get())  时

8670
hello-0
8671
hello-1
8672
hello-2
8673
hello-3
8670
hello-4
8671
hello-5
8672
hello-6
8673
hello-7
8670
hello-8
8671
hello-9
Traceback (most recent call last):
  File "test1.py", line 22, in <module>
    print('return:',r.get())
AttributeError: 'str' object has no attribute 'get'

最后报错: AttributeError: 'str' object has no attribute 'get'

3.3 applay_async方式添加任务

import multiprocessing as mp
import os
from time import sleep 

def worker(msg):
    print(os.getpid())
    sleep(2)
    print(msg)
    return msg

#创建进程池对象
p = mp.Pool(processes = 4) #创建4条进程

pool_result = []
for i in range(10):
    msg = 'hello-%d'%i
    r = p.apply_async(worker,(msg,)) #向进程池中添加事件
    pool_result.append(r)

#获取事件函数的返回值
for r in pool_result:
    print('return:',r)

p.close()#关闭进程池,不再接受请求
p.join()# 等待进程池中的事件执行完毕,回收进程池

运行

return: <multiprocessing.pool.ApplyResult object at 0x7f66d0e37d68>
return: <multiprocessing.pool.ApplyResult object at 0x7f66d0e37e80>
return: <multiprocessing.pool.ApplyResult object at 0x7f66d0e37f98>
return: <multiprocessing.pool.ApplyResult object at 0x7f66d0e410f0>
return: <multiprocessing.pool.ApplyResult object at 0x7f66d0e41208>
return: <multiprocessing.pool.ApplyResult object at 0x7f66d0e41320>
return: <multiprocessing.pool.ApplyResult object at 0x7f66d0e41438>
return: <multiprocessing.pool.ApplyResult object at 0x7f66d0e41550>
return: <multiprocessing.pool.ApplyResult object at 0x7f66d0e41668>
return: <multiprocessing.pool.ApplyResult object at 0x7f66d0e41780>
8739
8740
8742
8741
hello-0
hello-3
8742
hello-1
8739
8740
hello-2
8741
hello-5
8739
hello-6
8740
hello-7
hello-4
hello-8
hello-9

注意:

(1)由于这个是异步方式添加任务,所以运行非常快

(2)由于for是内置循环函数,执行效率较高,所以在结果的前10行均为for语句执行结果

(3) r = p.apply_async(worker,(msg,)) 执行结果为进度对象。

(4)由于任务是异步执行,所以在结果中是“乱序”;并不像applay那样有序打印。

同样将代码(22行)中的 print('return:',r) 修改为 print('return:',r.get())  时,

运行结果

8839
8840
8841
8842
hello-0
hello-1
hello-3
8839
hello-2
8842
8841
8840
return: hello-0
return: hello-1
return: hello-2
return: hello-3
hello-4
hello-5
8839
hello-6
8842
hello-7
return: hello-4
return: hello-5
return: hello-6
return: hello-7
hello-9
hello-8
return: hello-8
return: hello-9

在结果中出现“顺序混乱”

这与进程调度及运行时间有所差别有关,当有多个进程并行执行时,每个进程得到的时间片时间不一样,哪个进程接受那个求情以及执行完成时间都死不定的,所以输出会出现乱序的情况。

有时候还会出现两行数据出现在同一行,而下一行却为空行的情况,该情况可能时再执行第一个进程时,刚要打印换行符时,另一个进程也打印出来,这样就有可能本来两行的数据却在同一行打印出来,而两个换行符却次第打印出来,所以就会出现空行的情况。

注意:apply_async()函数本身就可以返回被进程调用的函数返回值。在创建子进程的代码中,若在被调用函数中返回一个值,那么pool.apply_async(func, (msg,))的结果就是返回pool中所有进程的“值的对象”(注意是对象,而不是值本身);同时对比不难发现,pool.apply的结果返回的是被调用函数的返回值,这里是值而不是对象。

比较使用进程池与函数之间

import time
from multiprocessing import Pool 

def run(fn):
    time.sleep(1)
    return fn*fn

test = [1,2,3,4,5,6,7,8]

s = time.time()
for fn in test:
    run(fn)
e = time.time()
print('执行时间:',e - s)

pool = Pool(3)

#使用该模块中的map融合了原map函数和该模块中apply_async函数
r = pool.map(run,test)
pool.close()
pool.join()
e1 = time.time()
print('执行时间:',e1 - e)

运行

执行时间: 8.011977195739746
执行时间: 3.0851333141326904

参考

Python多进程库multiprocessing中进程池Pool类的使用

Python 多进程 multiprocessing.Pool类详解

 

上一篇:python中多进程multiprocessing、多线程threading、线程池threadpool


下一篇:python多进程-----multiprocessing包