[python] 多进程编程

引言

讲到进程,不得不先说下linuxfork()函数,一个进程调用fork()函数后,系统先给新的进程分配资源,例如存储数据和代码的空间。然后把原来的进程的所有值都复制到新的新进程中,只有少数值与原来的进程的值不同。相当于克隆了一个自己。

import os
import time
# 此代码只能运行于linux/unix
pid = os.fork() # 创建一个子进程
print("test")
if pid == 0:
    print("子进程 {}, 父进程是 {}".format(os.getpid(), os.getppid()))
else:
    print("我是父进程: {}".format(pid))
time.sleep(2)
# 执行结果
# test
# 我是父进程: 6428
# test
# 子进程
# 6428, 父进程是 6427

注意到,fork()后面的代码执行了两次,但行为不一样。如果fork()返回0,说明是子进程。而子进程会将父进程的数据原样拷贝一遍,所以和线程不一样,不能通过全局变量来通信。

由于GIL锁的存在,python的多线程一直被人诟病,在CPU密集型操作中,推荐多进程,在IO密集型操作中,推荐多线程。和多线程的threading模块一样,python为多进程也提供了multiprocessing多进程包。

multiprocessing模块

有了多线程的使用基础,多进程的API很好理解。

普通使用

import multiprocessing
import time
def get_html(n):
    time.sleep(n)
    print("sub_progress success")
    return n

if __name__ == "__main__": #注意,在windows下这句话必须加!!!
    progress = multiprocessing.Process(target=get_html, args=(2,))
    print(progress.pid)
    progress.start() # 启动子进程
    print(progress.pid) # 打印子进程的PID
    progress.join() # 等待子进程结束
    print("main progress end")
# 执行结果
# None
# 5444
# sub_progress success
# main progress end
  1. multiprocessing.Process()传入执行入口函数和参数。
  2. start()方法启动子进程。
  3. progress.pid属性可以获取子进程的ID号,在未启动前,未None,启动后,得到系统分配的ID号。
  4. join()方法等待子进程结束。

进程池

multiprocessing模块中提供Pool类来支持进程池编程。

import multiprocessing
import time
def get_html(n):
    time.sleep(n)
    print("sub_progress success")
    return n

if __name__ == "__main__": #注意,在windows下这句话必须加!!!
    pool = multiprocessing.Pool(multiprocessing.cpu_count())
    result = pool.apply_async(get_html, args=(3,))

    pool.close() # 在join前必须运行close()
    pool.join()
    print(result.get()) # 得到子进程的返回值
# 执行结果  
# sub_progress success
# 3
  1. 使用multiprocessing.Pool()来创建进程池,可以设置进程池的大小,默认为CPU的核心个数,一般进程也不会超过CPU的核心个数,超过了效率也不会高。
  2. apply_async方法传入执行入口函数和参数,传入之后,就进入了子进程等待队列中,等待执行。返回值是ApplyResult对象,存储子进程的结果。
  3. join()方法等待所有子进程的结束,调用前必须使用close()方法。
  4. ApplyResult对象的get()方法,获取子进程的返回值。

multiprocessing模块同样提供了类似于map的方法:

import multiprocessing
import time
def get_html(n):
    time.sleep(n)
    print("{}s sub_progress success".format(n))
    return n

if __name__ == "__main__": #注意,在windows下这句话必须加!!!
    pool = multiprocessing.Pool(multiprocessing.cpu_count())
    # for result in pool.imap_unordered(get_html, [1, 3, 2]):
    for result in pool.imap(get_html, [1, 3, 2]):
        print("{}s sleep success".format(result))
# 执行结果
# 1s sub_progress success
# 1s sleep success
# 2s sub_progress success
# 3s sub_progress success
# 3s sleep success
# 2s sleep success
  1. imap方法,结果输出顺序与传入的可迭代对象的顺序相同。
  2. imap_unordered方法,结果输出顺序与子进程结束的时间顺序相同。

进程池的更好实现

虽然multiprocessing模块中提供Pool类,但是使用进程池的时候,使用concurrent.futures模块中的ProcessPoolExecutor类更加方便。在介绍线程池的时候,介绍了ThreadPoolExecutor的使用,而ProcessPoolExecutor的接口实现和ThreadPoolExecutor完全一样,只用多创建多线程改完创建多进程即可。值得注意的是:在windows下进行多进程编程的时候,主程序一定要加if __name__ == "__main__":

总结

  • 在CPU密集型操作时,使用多进程可以加快速度。
  • 多进程API与多线程API基本相同,在使用线程池的时候,可以使用multiprocessing.Pool,也可以使用Future模块下的ProcessPoolExecutor,推荐后者。
  • 在windows下多进程编程要加if __name__ == "__main__":

参考

Python3高级编程和异步IO并发编程

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 196,264评论 5 462
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 82,549评论 2 373
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 143,389评论 0 325
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,616评论 1 267
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,461评论 5 358
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,351评论 1 273
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,776评论 3 387
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,414评论 0 255
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,722评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,760评论 2 314
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,537评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,381评论 3 315
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,787评论 3 300
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,030评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,304评论 1 252
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,734评论 2 342
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,943评论 2 336

推荐阅读更多精彩内容