进程和线程
简介
多任务
现代操作系统包括Mac OS X,UNIX,Linux,Windows等,它们都是支持“多任务”的操作系统。
什么叫“多任务”呢?简单地说,就是操作系统可以同时运行多个任务。打个比方,你一边在用浏览器上网,一边在听MP3,一边在用Word赶作业,这就是多任务,至少同时有3个任务正在运行。还有很多任务悄悄地在后台同时运行着,只是桌面上没有显示而已。 现在多核CPU已经非常普及了,使用不同的核执行不同任务自然也不是什么难事。但是,即使是过去的单核CPU,也可以执行多任务。我们知道,CPU执行代码都是顺序执行的,那么单核CPU是怎么执行多任务的呢?
答案就是操作系统轮流让各个任务交替执行,任务1执行0.01秒,切换到任务2,任务2执行0.01秒,再切换到任务3,执行0.01秒…… 这样反复执行下去。表面上每个任务都是交替执行的,但是,由于CPU的执行速度实在是太快了,我们实际感觉到的就变成了所有任务都在同时执行。
真正的并行执行多任务只能在多核CPU上实现,但是,由于任务数量往往是远多于CPU的核心数量的,要给每个任务都分配一个CPU内核是不现实的,所以实际上操作系统会自动把很多任务轮流调度到每个核心上执行。
进程和线程的含义
对于操作系统来说,一个任务就是一个进程(Process),比如打开一个浏览器就是启动一个浏览器进程,打开一个记事本就启动了一个记事本进程,打开两个记事本就启动了两个记事本进程,打开一个Word就启动了一个Word进程。
有些进程还不止同时干一件事,比如Word,它可以同时进行打字、拼写检查、打印等事情。在一个进程内部,要同时干多件事,就需要同时运行多个“子任务”,我们把进程内的“子任务”称为线程(Thread)。
由于每个进程至少要干一件事,所以,一个进程至少有一个线程。当然,像Word这种复杂的进程可以有多个线程,多个线程可以同时执行,多线程的执行方式和多进程是一样的,也是由操作系统在多个线程之间快速切换,让每个线程都短暂地交替运行,看起来就像同时执行一样。当然,真正同时执行多线程需要多核CPU才可能实现。
我们前面编写的所有的Python程序,都是执行单任务的进程,并且只有一个线程。如果我们要同时执行多个任务怎么办?有三种解决方案:
- 多进程模式:启动多个进程,每个进程只有一个线程,但多个进程可以一块执行多个任务;
- 多线程模式:启动一个进程,在一个进程内启动多个线程,多个线程可以一块执行多个任务;
- 多进程+多线程模式:启动多个进程,每个进程再启动多个线程,这样同时执行的任务就更多了,当然这种模型更复杂,实际很少采用。
注意,同时执行的多个任务之间可能是关联的,需要相互通信和协调。比方说,任务1必须先暂停等待任务2完成后才能继续执行,而任务3和任务4需要操作同一个文件所以不能同时执行。因此,多进程和多线程的程序的复杂度要远远高于我们前面写的单进程单线程的程序。
因为复杂度高,调试困难,所以,不是迫不得已,我们也不想编写多任务。但是,有很多时候,没有多任务还真不行。想想在电脑上看电影,就必须由一个线程播放视频,另一个线程播放音频,否则,单线程实现的话就只能先把视频播放完再播放音频,或者先把音频播放完再播放视频,这显然是不行的。
Python既支持多进程,又支持多线程,在接下来的这一章,我们会讨论如何编写这两种多任务程序。
小结
线程是最小的执行单元,进程由至少一个线程组成。如何调度进程和线程,完全由操作系统决定,程序自己不能决定自己什么时候被执行,执行多长时间。
多进程和多线程的程序涉及到同步、数据共享等问题,编写起来更复杂。
多进程
fork函数
要让Python程序实现多进程(multiprocessing),我们先得了解操作系统的相关知识。
Unix/Linux操作系统提供了一个 fork()
系统调用函数,它非常特殊。普通的函数在被调用时,调用一次只会返回一次。但是 fork()
函数调用一次会返回两次,因为此时操作系统会自动把当前进程(称为父进程)复制一份(称为子进程),然后分别在父进程和子进程内进行返回。
子进程永远返回0,而父进程返回子进程的ID。这样做的理由是,一个父进程可以fork出很多子进程,所以,父进程要记下每个子进程的ID,而子进程只需要调用 getppid()
就可以拿到父进程的ID。
Python的 os
模块封装了常见的系统调用,其中就包括fork,可以在Python程序中轻松创建子进程:
1 | import os |
运行结果如下:
1 | Process (876) start... |
这段代码在执行第一个print时只有一个进程(876),因此只打印一次。然后在执行 fork()
之后,当前进程(876)被复制出一个子进程(877)。当前进程会先返回,返回子进程ID(877),然后往下走进入if-else代码块,进入else语句进行打印;其后子进程(877)返回0,同样往下走,进入if-else代码块,进入if语句进行打印。
由于Windows没有fork调用,上面的代码在Windows上无法运行。而Mac系统是基于BSD(Unix的一种)内核,所以,在Mac下运行是没有问题的~
有了fork调用,一个进程在接到新任务时就可以复制出一个子进程来处理新任务,常见的Apache服务器就是由父进程监听端口,每当有新的http请求时,就fork出子进程来处理新的http请求。
multiprocessing
如果你打算编写多进程的服务程序,Unix/Linux无疑是正确的选择。由于Windows没有fork调用,难道在Windows上无法用Python编写多进程的程序?
由于Python是跨平台的,自然也应该提供跨平台的多进程支持。multiprocessing
模块就是跨平台版本的多进程模块。
multiprocessing
模块提供了一个 Process
类来代表一个进程对象,下面的例子演示了启动一个子进程并等待其结束:
1 | from multiprocessing import Process |
执行结果如下:
1 | Parent process 928. |
创建子进程时,只需要传入一个执行函数和该函数的参数就可以创建一个 Process
实例。用 start()
方法就可以启动这个子进程,这样创建进程要比 fork()
更简单和灵活。
join()
方法可以等待子进程结束后再继续往下运行,通常用于进程间的同步。
进程池
如果要启动大量的子进程,我们可以用进程池的方式来批量创建子进程:
1 | from multiprocessing import Pool |
执行结果如下:
1 | Parent process 669. |
代码解读:
对 Pool
对象调用 join()
方法会等待所有子进程执行完毕,调用 join()
之前必须先调用 close()
,调用 close()
之后就不能往进程池中继续添加新的 Process
了。
请注意输出的结果,task 0,1,2,3这四个子进程是立刻执行的,而task 4要等待前面某个task完成后才执行,这是因为我们定义进程池时定义了大小为4,也即最多同时执行4个进程,所以第5个进程就要等进程池里有空位了才能开始。如果改成:
1 | p = Pool(5) |
就可以同时跑5个进程。
当没有传入参数时,Pool的默认大小是CPU的核数,所以如果电脑是4核CPU则默认进程池大小为4,如果电脑是8核CPU则默认进程池大小为8。
子进程的输入和输出
很多时候,子进程和父进程要执行的不是同一个任务。我们创建了子进程后,还需要控制子进程的输入和输出。
subprocess
模块可以让我们非常方便地启动一个子进程,并且控制其输入和输出。
下面的例子演示了如何在Python代码中运行命令 nslookup www.python.org
,这和命令行直接运行的效果是一样的:
1 | import subprocess |
运行结果:
1 | $ nslookup www.python.org |
看看帮助文档中 subprocess
模块的 call()
函数的描述:
Run command with arguments. Wait for command to complete or timeout, then return the returncode attribute.
它可以帮助我们建立一个子进程来执行系统调用函数,并且允许传入函数,调用后会等待运行结束并最终返回调用函数的返回值,之后才继续执行后续代码。可以只传入一个列表,列表第一个元素为系统调用的名称,第二个元素为参数。
如果子进程执行的过程中还需要其他输入,我们使用 subprocess
模块的 Popen
类初始化子进程,并通过它的 communicate()
方法来实现进程执行过程中的多次输入:
1 | import subprocess |
上面的代码相当于在命令行执行命令 nslookup
,然后手动输入:
1 | set q=mx |
运行结果如下:
1 | $ nslookup |
进程间通信
Process之间肯定是需要通信的,操作系统提供了很多机制来实现进程间的通信。Python的 multiprocessing
模块包装了底层的机制,提供了 Queue
、Pipes
等多种方式来交换数据。
我们以 Queue
为例,在父进程中创建两个子进程,第一个子进程往 Queue
里写数据,第二个子进程从 Queue
里读数据:
1 | from multiprocessing import Process, Queue |
运行结果如下:
1 | Process to write: 50563 |
在Unix/Linux下,multiprocessing
模块封装了 fork()
调用,使我们不需要关注 fork()
的细节。由于Windows没有fork调用,因此,multiprocessing
需要“模拟”出fork的效果,父进程的所有Python对象都必须先通过 pickle
序列化再传到子进程去。所以,如果 multiprocessing
在Windows下调用失败了,就要先考虑是不是序列化失败了。
小结
在Unix/Linux下,可以使用 fork()
调用实现多进程。
要实现跨平台的多进程,可以使用 multiprocessing
模块。
进程间通信可以通过 Queue
、Pipes
等实现的。
多线程
多线程简单实现
多任务可以由多进程完成,也可以由一个进程内的多线程完成。
我们前面提到了进程是由若干线程组成的,一个进程至少有一个线程。
由于线程是操作系统直接支持的执行单元,因此,高级语言通常都内置多线程的支持,Python也不例外,并且,Python的线程是真正的Posix Thread,而不是模拟出来的线程。
Python的标准库提供了两个模块:_thread
和 threading
,_thread
是低级模块,threading是高级模块,对 _thread
进行了封装。绝大多数情况下,我们只需要使用 threading
这个高级模块。
使用 threading
时,我们可以创建 Thread
实例来建立新的线程,然后调用 start()
启动该线程:
1 | import time, threading |
执行结果如下:
1 | thread MainThread is running... |
由于任何进程默认就会启动一个线程,我们把该线程称为主线程,主线程又可以启动新的线程,Python的 threading
模块有个 current_thread()
函数,它永远返回当前线程的实例。主线程实例的名字叫 MainThread
,子线程的名字在创建时指定,我们用 LoopThread
命名子线程。名字仅仅在打印时用来显示,没有其他意义,如果不起名字Python就会自动给线程命名为 Thread-1
,Thread-2
……
线程锁
多线程和多进程最大的不同在于,多进程中,同一个变量,各自有一份拷贝存在于每个进程中,互不影响。而多线程中,所有变量都由所有线程共享,所以,任何一个变量都可以被任何一个线程修改,因此,线程之间共享数据最大的危险在于多个线程同时改一个变量,把内容给改乱了。
来看看多个线程同时操作一个变量怎么把内容给改乱了:
1 | import time, threading |
注意,change_it()
函数中,使用了global关键字来声明使用的 balance
变量是在函数定义外部的,这样在函数内部的修改也会反映到函数外部。
balance
,初始值为0,并且启动两个线程,先加后减,理论上结果应该为0,但是,由于线程的调度是由操作系统决定的,当t1、t2交替执行时,只要循环次数足够多,在某个时刻,线程t1和t2同时对 balance
变量进行修改,那么最终结果就不一定是0了。比方说:
时间 | 线程 | balance | 操作 |
---|---|---|---|
00 | t1 | 0 | +5 |
01 | t1 | 5 | -5 |
02 | t2 | 0 | +8 |
03 | t2 | 8 | -8 |
04 | t1 | 0 | +5 |
05 | t2 | 5 | +8 |
06 | t1 | 13 | -8 |
07 | t2 | 5 | -5 |
08 | t1,t2 | 0 | +5,+8 |
09 | t1 | 8 | -5 |
10 | t2 | 3 | -8 |
11 | t1 | -5 | +5 |
12 | t2 | 0 | +8 |
13 | t1 | 8 | -5 |
14 | t2 | 3 | -8 |
假设在08时刻,t1和t2同时执行 balance = balance + n
这个语句,右边的 balance
都取0,那么t1执行完该语句时 balance
为5,而t2执行完时 balance
为8,假设t2后执行完毕,那么 balance
就取8,这时就产生错误了,并且错误会继续累积,当循环次数达到一定规模时,这种错误的累积会非常可怕。
为什么会产生这样的错误呢?其实 呀,这时因为高级语言的一条语句在CPU执行时需要转换为多条汇编指令,即使一个简单的计算:
1 | balance = balance + n |
也需要分为两步:
- 计算
balance + n
,存入临时变量中; - 将临时变量的值赋给
balance
。
可以看成:
1 | x = balance + n |
临时变量 x
是一个局部变量,两个线程各自都有自己的 x
,所以当代码正常执行时:
初始值 balance = 0
1 | t1: x1 = balance + 5 # x1 = 0 + 5 = 5 |
结果 balance = 0
但是t1和t2是交替运行的,如果操作系统以下面的顺序执行t1、t2:
初始值 balance = 0
1 | t1: x1 = balance + 5 # x1 = 0 + 5 = 5 |
结果 balance = -8
,自然就不对了
究其原因,是因为修改 balance
需要多条语句,而执行这几条语句时,线程可能中断,其他线程可能也会对 balance
进行了修改,从而导致多个线程把同一个对象的内容改乱了。所以,我们必须确保一个线程在修改 balance
的时候,别的线程一定不能改。
如果我们要确保 balance
计算正确,就要给 change_it()
上一把锁,当某个线程开始执行 change_it()
时,我们说,该线程获得了锁,因此其他线程不能同时执行 change_it()
,只能等待锁被释放,线程获得该锁以后才能改。由于锁只有一个,无论多少线程,同一时刻最多只有一个线程持有该锁,所以,不会造成修改的冲突。创建一个锁可以通过 threading.Lock()
来实现:
1 | balance = 0 |
当多个线程同时执行 lock.acquire()
时,只有一个线程能成功地获取锁,然后继续执行代码,其他线程就继续等待直到获得锁为止。
获得锁的线程用完后一定要释放锁,否则那些苦苦等待锁的线程将永远等待下去,成为死线程。所以我们用 try...finally
来确保锁一定会被释放。
锁的好处就是确保了某段关键代码只能由一个线程从头到尾完整地执行,坏处当然也很多,首先是阻止了多线程并发执行,包含锁的某段代码实际上只能以单线程模式执行,效率就大大地下降了。其次,由于可以构造多个不同的锁,不同的线程持有不同的锁,在试图获取对方持有的锁时,可能会造成死锁,导致多个线程全部挂起,既不能执行,也无法结束,只能靠操作系统强制终止。
死循环与多核CPU
如果你拥有一个多核CPU,肯定就会想到不同的核应该可以同时执行不同的多个线程。
如果写一个死循环的话,会出现什么情况呢?
打开Mac OS X的Activity Monitor,或者Windows的Task Manager,都可以监控某个进程的CPU使用率。我们可以监控到一个死循环线程会100%占用一个CPU。如果有两个死循环线程,在多核CPU中,可以监控到会占用200%的CPU,也就是占用两个CPU核心。要想把N核CPU的核心全部跑满,就必须启动N个死循环线程。
试试用Python写个死循环:
1 | import threading, multiprocessing |
启动与CPU核心数量相同的N个线程,可以监控到CPU占用率仅有102%,也就是仅使用了一核。但是用C、C++或Java来改写相同的死循环,直接可以把全部核心跑满,4核就跑到400%,8核就跑到800%,为什么Python不行呢?
因为Python的线程虽然是真正的线程,但解释器执行代码时,有一个GIL锁:Global Interpreter Lock,任何Python线程执行前,必须先获得GIL锁,然后,每执行100条字节码,解释器就自动释放GIL锁,让别的线程有机会执行。这个GIL全局锁实际上把所有线程的执行代码都给上了锁,所以,多线程在Python中只能交替执行,即使100个线程跑在100核CPU上,也只能用到1个核。
GIL是Python解释器设计的历史遗留问题,通常我们用的解释器是官方实现的CPython,要真正利用多核,除非重写一个不带GIL的解释器。
所以,在Python中,可以使用多线程,但不要指望能有效利用多核。如果一定要通过多线程利用多核,那只能通过C扩展来实现,不过这样就失去了Python简单易用的特点。
不过,也不用过于担心,Python虽然不能利用多线程实现多核任务,但可以通过多进程实现多核任务。多个Python进程有各自独立的GIL锁,互不影响。
小结
多线程编程,模型复杂,容易发生冲突,必须用锁加以隔离,同时,又要小心死锁的发生。
Python解释器由于设计时有GIL全局锁,导致了多线程无法利用多核。多线程并发在Python中就是一个美丽的梦。
ThreadLocal
局部变量的传递
在多线程环境下,每个线程都有自己的数据。一个线程使用自己的局部变量比使用全局变量好,因为局部变量只有线程自己能看见,不会影响其他线程,而全局变量的修改必须加锁。
但是局部变量也有问题,就是在函数调用的时候,传递起来很麻烦:
1 | def process_student(name): |
每个函数一层一层调用都这么传参数那还得了?用全局变量?也不行,因为每个线程处理不同的 Student
对象,所以不能共享。
如果用一个全局的 dict
存放所有的 Student
对象,以线程自身作为 key
和存取对应的 Student
对象如何呢?
1 | global_dict = {} |
这种方式理论上是可行的,它最大的优点是消除了 std
对象在每层函数中的传递问题,但是,每个函数获取 std
的代码有点丑。
ThreadLocal的用法
有没有更简单的方式?ThreadLocal
应运而生,不用查找 dict
,ThreadLocal
可以自动帮我们做这件事:
1 | import threading |
执行结果:
1 | Hello, Alice (in Thread-A) |
全局变量 local_school
就是一个 ThreadLocal
对象,每个线程都可以对它进行读写,线程之间互不影响。你可以把 local_school
看成全局变量,但每个属性如 local_school.student
都是线程的局部变量,可以任意读写而互不干扰,也不用管理锁的问题,ThreadLocal
内部会处理。也可以把 local_school
理解为一个全局的 dict
,不但可以绑定 local_school.student
,还可以绑定其他变量,如 local_school.teacher
等等。
ThreadLocal
最常用的地方就是为每个线程绑定一个不同的数据库连接,HTTP请求,用户身份信息等,这样一个线程的所有调用到的处理函数都可以非常方便地访问这些资源。
小结
ThreadLocal
虽然是全局的,但每个线程都只能读写自己线程的独立副本,互不干扰。ThreadLocal
解决了参数在一个线程中各个函数之间互相传递的问题。
进程 vs. 线程
比较多进程和多线程
我们介绍了多进程和多线程,这是实现多任务最常用的两种方式。现在,我们来讨论一下这两种方式的优缺点。
首先,要实现多任务,通常我们会设计主从模式(Master-Worker模式),Master负责分配任务,Worker负责执行任务,因此,多任务环境下,通常是一个Master,多个Worker。
- 如果用多进程实现Master-Worker,主进程就是Master,其他进程就是Worker。
- 如果用多线程实现Master-Worker,主线程就是Master,其他线程就是Worker。
多进程模式最大的优点就是稳定性高,因为一个子进程崩溃了,不会影响主进程和其他子进程。(当然主进程挂了所有进程就全挂了,但是Master进程只负责分配任务,挂掉的概率低)著名的Apache最早就是采用多进程模式。
多进程模式的缺点是创建进程的代价大,在Unix/Linux系统下,用fork调用还行,在Windows下创建进程开销巨大。另外,操作系统能同时运行的进程数也是有限的,在内存和CPU的限制下,如果有几千个进程同时运行,操作系统连调度都会成问题。
多线程模式通常比多进程快一点,但是也快不到哪去,而且,多线程模式致命的缺点就是任何一个线程挂掉都可能直接造成整个进程崩溃,因为所有线程共享进程的内存。在Windows上,如果一个线程执行的代码出了问题,你经常可以看到这样的提示:“该程序执行了非法操作,即将关闭”,其实往往是某个线程出了问题,但是操作系统会强制结束整个进程。
在Windows下,多线程的效率比多进程要高,所以微软的IIS服务器默认采用多线程模式。由于多线程存在稳定性的问题,IIS的稳定性就不如Apache。为了缓解这个问题,IIS和Apache现在又有多进程+多线程的混合模式,真是把问题越搞越复杂。
效率问题
无论是多进程还是多线程,只要数量一多,效率肯定上不去,为什么呢?
我们打个比方,假设你不幸正在准备中考,每天晚上需要做语文、数学、英语、物理、化学这5科的作业,每项作业耗时1小时。
如果你先花1小时做语文作业,做完了,再花1小时做数学作业,这样,依次全部做完,一共花5小时。这种每次只完成一个任务的方式称为单任务模型,或者批处理任务模型。
假设你打算切换到多任务模型,可以先做1分钟语文,再切换到数学作业,做1分钟,再切换到英语,以此类推,只要切换速度足够快,这种方式就和单核CPU执行多任务是一样的了,以幼儿园小朋友的眼光来看,你就正在同时写5科作业。
但是,切换作业是有代价的,比如从语文切到数学,要先收拾桌子上的语文书本、钢笔(这叫保存现场),然后,打开数学课本、找出圆规直尺(这叫准备新环境),才能开始做数学作业。操作系统在切换进程或者线程时也是一样的,它需要先保存当前执行的现场环境(CPU寄存器状态、内存页等),然后,把新任务的执行环境准备好(恢复上次的寄存器状态,切换内存页等),才能开始执行。这个切换过程虽然很快,但是也需要耗费时间。如果有几千个任务同时进行,操作系统可能就主要忙着切换任务,根本没有多少时间去执行任务了,这种情况最常见的就是硬盘狂响,点窗口无反应,系统处于假死状态。
所以,多任务一旦多到一个限度,就会消耗掉系统所有的资源,结果效率急剧下降,所有任务都做不好。
计算密集型 vs. IO密集型
是否采用多任务的第二个考虑是任务的类型。我们可以把任务分为计算密集型和IO密集型。
计算密集型任务的特点是要进行大量的计算,消耗CPU资源,比如计算圆周率、对视频进行高清解码等等,全靠CPU的运算能力。这种计算密集型任务虽然也可以用多任务完成,但是任务越多,花在任务切换的时间就越多,CPU执行任务的效率就越低,所以,要最高效地利用CPU,计算密集型任务同时进行的数量应当等于CPU的核心数。
计算密集型任务由于主要消耗CPU资源,因此,代码运行效率至关重要。Python这样的脚本语言运行效率很低,完全不适合计算密集型任务。对于计算密集型任务,最好用C语言编写。
第二种任务的类型是IO密集型,涉及到网络、磁盘IO的任务都是IO密集型任务,这类任务的特点是CPU消耗很少,任务的大部分时间都在等待IO操作完成(因为IO的速度远远低于CPU和内存的速度)。对于IO密集型任务,任务越多,CPU效率越高,但也有一个限度。常见的大部分任务都是IO密集型任务,比如Web应用。
IO密集型任务执行期间,99%的时间都花在IO上,花在CPU上的时间很少,因此,用运行速度极快的C语言替换用Python这样运行速度极低的脚本语言,完全无法提升运行效率。对于IO密集型任务,最合适的语言就是开发效率最高(代码量最少)的语言,脚本语言是首选,C语言最差。
异步IO
考虑到CPU和IO之间巨大的速度差异,一个任务在执行的过程中大部分时间都在等待IO操作,单进程单线程模型会导致别的任务无法并行执行,因此,我们才需要多进程模型或者多线程模型来支持多任务并发执行。
现代操作系统对IO操作已经做了巨大的改进,最大的特点就是支持异步IO。如果充分利用操作系统提供的异步IO支持,就可以用单进程单线程模型来执行多任务,这种全新的模型称为事件驱动模型,Nginx就是支持异步IO的Web服务器,它在单核CPU上采用单进程模型就可以高效地支持多任务。在多核CPU上,可以运行多个进程(数量与CPU核心数相同),充分利用多核CPU。由于系统总的进程数量十分有限,因此操作系统调度非常高效。用异步IO编程模型来实现多任务是一个主要的趋势。
所谓异步IO,其实这个异步是指使用者发起IO请求后并不马上得到结果。比方说有一些需要处理的数据放在磁盘上,使用者预先知道这些数据的位置,所以预先发起异步IO读请求。而等到真正需要用到这些数据的时候,再等待异步IO完成,对数据进行处理。这样做的话,在发起IO请求到实际使用数据这段时间内,程序还可以继续做其他事情。
对应到Python语言,单进程的异步编程模型称为协程,有了协程的支持,就可以基于事件驱动编写高效的多任务程序。我们会在后面讨论如何编写协程。
分布式进程
简介
在线程和进程中,应当优先选择进程,因为进程更稳定,而且,进程可以分布到多台机器上,而线程最多只能分布到同一台机器的多个CPU上。
Python的 multiprocessing
模块不但支持多进程,其中 managers
子模块还支持把多进程分布到多台机器上。一个服务进程可以作为调度者,将任务分布到其他多个进程中,依靠网络通信。由于 managers
模块封装很好,我们不必了解网络通信的细节,就可以很容易地编写分布式多进程程序。
举个例子:如果我们已经有一个通过 Queue
通信的多进程程序在同一台机器上运行,现在,由于处理任务的进程任务繁重,希望把发送任务的进程和处理任务的进程分布到两台机器上。怎么用分布式进程实现?
原有的 Queue
可以继续使用,但是,通过 managers
模块把 Queue
通过网络暴露出去,就可以让其他机器的进程访问 Queue
了。
实现方法
我们先看服务进程,服务进程负责启动 Queue
,把 Queue
注册到网络上,然后往 Queue
里面写入任务:
1 | # task_master.py |
请注意,当我们在一台机器上写多进程程序时,创建的 Queue
可以直接拿来用,但是,在分布式多进程环境下,添加任务到 Queue
不可以直接对原始的 task_queue
进行操作,那样就绕过了 QueueManager
的封装,必须通过 manager.get_task_queue()
获得的 Queue
接口添加任务。
然后,在另一台机器上启动任务进程(本机上启动也可以):
1 | # task_worker.py |
任务进程要通过网络连接到服务进程,所以要指定服务进程的IP。
现在,可以试试分布式进程的工作效果了。先启动 task_master.py
服务进程:
1 | $ python3 task_master.py |
task_master.py
进程发送完任务后,开始等待 result
队列的结果。现在启动 task_worker.py
进程:
1 | $ python3 task_worker.py |
task_worker.py
进程结束,在 task_master.py
进程中会继续打印出结果:
1 | Result: 3411 * 3411 = 11634921 |
这个简单的Master/Worker模型有什么用?其实这就是一个简单但真正的分布式计算,把代码稍加改造,启动多个worker,就可以把任务分布到几台甚至几十台机器上,比如把计算 n*n
的代码换成发送邮件,就实现了邮件队列的异步发送。
Queue
对象存储在哪?注意到 task_worker.py
中根本没有创建 Queue
的代码,所以,Queue
对象是存储在 task_master.py
进程中的:
而 Queue
之所以能通过网络访问,就是通过 QueueManager
实现的。由于 QueueManager
管理的不止一个 Queue
,所以,要给每个 Queue
的网络调用接口起个名字,比如 get_task_queue
。
authkey
有什么用?这是为了保证两台机器正常通信,不被其他机器恶意干扰。如果 task_worker.py
的 authkey
和 task_master.py
的 authkey
不一致,肯定连接不上。
小结
Python的分布式进程接口简单,封装良好,适合需要把繁重任务分布到多台机器的环境下。
注意 Queue
的作用是用来传递任务和接收结果,每个任务的描述数据要尽量小。比如发送一个处理日志文件的任务,就不要发送几百兆的日志文件本身,而是发送日志文件存放的完整路径,由Worker进程再去共享的磁盘上读取文件。