python支持多线程多进程编程,标准库模块threading使得python的多线程编程变得容易,支持很多功能,在python2.6后,python新添加了多进程的标准库模块multiprocessing,功能非常类似threading。
下面介绍multiprocessing模块的一些常用类和方法。
1、multiprocessing.Process
该类原型如下:
class multiprocessing.Process(group=None, target=None, name=None, args=(), kwargs={})
group参数通常为None,仅仅是为兼容threading.Thread。
target为可调用对象,被run()函数调用,一般用法就是将函数传给该参数,交给子进程运行。
name为进程名,如不指定,默认为Process-1,Process-2....这样的名字,数字随自己查数量依次递增。
args为target调用的参数元祖。
kwargs为target调用的关键字参数字典。
Process类生成一个process对象,并运行在一个新的单独的进程内。
一个简单的使用例子:
from multiprocessing import Process
import time
def timetask(times):
time.sleep(times)
print time.localtime()
if __name__ == '__main__':
arg = 5
p = Process(target = timetask, args = (arg,))
p.start()
p.join()
每个process对象最多只能调用一次start()方法,join([timeout])方法会阻塞调用process对象的进程,直到timeout时间超时,或者process进程退出。如果timeout设置为None,则无超时时间。对于linux操作系统的进程管理,父进程会等待子进程退出,并收回子进程的资源,然后父进程再退出。如果父进程先于子进程退出,则子进程会被init进程接管。如果父进程没有退出也没有回收子进程资源,则子进程会一直停留在僵死状态。这里join方法就是阻塞父进程,等待子进程执行完毕。
如果在子进程里面要接受标准输入(如调用raw_input函数),这时候会有问题。因为标准输入是被父进程占用,因此这里子进程无法从标准输入读取数据。
一个解决办法是将标准输入描述符,传入子进程函数:
from multiprocessing import Process
import sys, os
import time
def intask(x, fileno):
sys.stdin = os.fdopen(fileno)
while True:
in_char = raw_input("Enter to continue, Q to quit: ")
if in_char.upper() == 'Q':
break
print x * x
if __name__ == '__main__':
fn = sys.stdin.fileno()
arg = 5
p = Process(target = intask, args = (arg, fn))
p.start()
p.join()
开启多个进程并行执行:
from multiprocessing import Process
import sys, os
import time
def timetask(times):
time.sleep(times)
print time.localtime()
def works(func, arg, worknum):
proc_record = []
for i in range(worknum):
p = Process(target = func, args = (arg,))
p.start()
proc_record.append(p)
for p in proc_record:
p.join()
if __name__ == '__main__':
arg = 5
procs = 4
works(timetask, arg, procs)
程序同时开启4个进程并行执行timetask函数,这里需要注意的是先把所有的进程对象执行了start()方法后,再执行join(),如果对每个进程对象一起执行start()和join(),那么单个进程对象start()后,会马上阻塞父进程,导致无法马上对下一个进程对象执行start()。结果就是子进程并不是并行运行,而是依次运行。
有时间再更新multiprocessing模块的其他类和方法