多进程、多线程
这周接到1个需求,项目经理觉得我这边构造数据太过缓慢,由于数据量过大,以前数据构造完后将其导入MPP,利用copy_from速度还是很快的,一般为10W/s。现在换成Kafka消息队列,又由于python库自带的原因(这个是组里大神告诉我的),无法像java开发利用list导数据,因此我只能一条一条以json的形式将数据放入消息队列,一般为800条/s。原先构造5000W数据需要7H,经过优化后暂时需要3.5H,性能提升一倍。有点小开心,话不多说,石海同学开始将一下个人对于标题的理解。。。。
进一步优化利用多进程的JoinableQueue,一边一直产生数据,另一边一直消费数据,现在5000W数据大约需要1.5H,性能较最初提升了近五倍。
历史
上古年代:在很久很久以前,当时主流的磁盘操作系统MS-DOS是只支持单任务操作的,就打个比方,如果我想在电脑上听音乐和看电影,是不能同时开启的,只能先听音乐后看电影,或者位置互换。
2002:横空出世的Intel Pentium 4HT处理器,提出了cpu多线程(SMT),其支持一个cpu进行多任务开启。
————你总不能要求Guido1989年为了打发圣诞节假期发明的一种编译语言还要设计一下多线程的部分。
2006:在秋季的英特尔信息技术峰会上,Inter总裁宣布,其在11月份将会交付第一台4核cpu处理器,而这距离双核发布还不到1年,支持多核时代就此拉开序幕。
————为什么说python多线程是历史遗留问题,因为在当时想要在两个或者更多的处理器对同一个对象运行时,为了保护线程数据完整性和状态同步,最简单的方法就是加锁,于是出现了GIL这把超级大锁。
2006-至今:GPU也浩浩荡荡发展了十几年了,过程我就不细说了(因为不懂),总之对于图象类任务GPU和CPU不是一个量级上的,以前看过一个视频,很好的解释了CPU和GPU的区别。比如我要画一幅图,cpu需要一笔一划将他画出来,而GPU是直接在脑子里构思好,一炮就把图打出来了。因此对于神经网络这个模型,多个神经元同时进行计算,GPU比CPU快太多。。。。。
编译器、解释器、IDE
编译器:对于C、C++这类语言需要先编译后运行,其中编译的时候就是编译器。
解释器:对于python、PHP、JavaScript等就是利用解释器,’一边编译一边解释‘,不过速度会比较慢,因此产生一种叫预编译的东西,python在运行时就先生成pyc的二进制临时文件,在出结果。
预编译:Java、C#,运行前将源代码编译成中间代码,然后再解释器运行,这种执行效率比编译运行会有效率一些,避免重复编译。
IDE:集成开发环境(IDE,Integrated Development Environment ),用于提供程序开发环境的应用程序,python常用的就是pycharm和jupyter notebook。
GIL锁
概念:GIL全称Global Interpreter Lock,其并不是python的特性,由于Cpython是大部分环境下默认的python执行环境,而在实现python解释器会自动上锁。
目的:确保每个进程只有一个线程运行,所以在外面我们一般说python的多线程是伪线程。因为不管你有几个核,你用多线程也只能跑一个核。
进程、线程、协程的利用
进程:拥有代码和打开的文件资源、数据资源、独立的内存空间。
线程:从属于进程,是程序的实际执行者,线程拥有自己的栈空间。
协程:从属于线程,是一种比线程更加轻量级的存在。
总结:
对操作系统来说,线程是最小的执行单元,进程是最小的资源管理单元。
多进程
我觉得最简单的讲,多进程就是你在任务窗口看建了几个任务。比如你电脑上有16个核,理论上你可以开16个进程,每个进程占满一个cpu。对python而言由于没有多线程的利用,所有在单进程无法满足需求时,自然得利用多进程。
单任务单进程
1 | from multiprocessing import Process |
我还看到一种是用run不用start的,但大概看了下没什么特殊的,不过这个join只能用于start.所以建议还是都用start,这个join是阻塞的意思,简单来说:主进程和其他子进程结束的话,都等着,等我结束了才能结束。
举个例子:
1 | from multiprocessing import Process |
如果这么写,主进程你走你的,work2子进程你给我等一下。等我run完你在run。
注意:每个子进程之间的join不是串行的,是并行的。既无论你多少个子进程,谁最后结束,谁关闭文件。
单任务多进程
比如你要启动50个进程,然后写50个子进程就太慢了。因此,对于多进程有一种新的形式。
1 | def long_time_task(name): |
可以看到task1和task4的进程是一样的,因为只启动了4个进程,因此第5个进程需要等待其他进程结束才开始运行。
注意:close不加直接写join是会报错的。
多任务多进程
我这边用的是经典的生产者和消费者模型,及一个模型构造生产者,一个模型构造消费者,两者通过自带的JoinableQueue进行通信。(不要用deque里面的队列!!)
1 | import time |
具体实现情况
- 启动一个生产者,和一个消费者(这里可以用多进程消费),看具体时间
- 生产者结束后将JoinableQueue进行阻塞,直到队列全部被消费后,才结束进程。
Queue与JoinableQueue的区别
- Queue在消费者中需要if判断队列,不然的话就陷入死循环。而jionableQueue不用,其只需要将队列堵塞后,队列消费完程序才解决。
- Queue很难控制,因为如果只是判断队列是否存在,就需要考虑产生和消耗队列的时间。因此我认为还是JoinableQueue好
多线程
开头就说了,python虽然线程是真正的线程,但在解释器执行的时候会遇到GIL锁,导致其无论多少核,只能跑满一个核,所以对于比较简单的可以运行多线程,好像在爬虫里面运用多线程的比较多。
那为什么有的程序一个进程一开用了N个cpu,那是因为他进程中存在利用c扩写的库,他将关键的部分用C/C++写成python扩展,其他还用python写。一般计算密集型的程序都会用C代码编写并通过扩展的方式集成到python脚本(numpy)。在扩展中完全可以用C创建原生线程,而且不用锁GIL,充分利用CPU的计算资源。
1 | from threading import Thread |
问题
1 | 能否利用Pool批量构造生产者和消费者, |