专栏名称: 马哥Linux运维
马哥linux致力于linux运维培训,连续多年排名第一,订阅者可免费获得学习机会和相关Linux独家实战资料!
目录
相关文章推荐
运维  ·  再见,CDN 巨头:Akamai 宣布 ... ·  昨天  
51好读  ›  专栏  ›  马哥Linux运维

Python的进程

马哥Linux运维  · 公众号  · 运维  · 2019-06-27 22:01

正文


来源:Python绿色通道

ID: Python_channel

作者:By

进程

说明:本文是基于Py2.X环境,

Python实现多进程的方式主要有两种:一种方法是使用os模块中的fork方法; 另一种是使用multiprocessing模块。这两种方法的区别在于前者仅适用于Unix/Linux操作操作。对win是不支持的,而后者则是跨平台的实现方式。

使用os模块中的fork方式实现多进程。

Unix/Linux操作系统提供了一个fork()系统调用,它非常特殊。普通的函数调用,调用一次,返回一次,但是fork()调用一次,返回两次,因为操作系统自动把当前进程(称为父进程)复制了一份(称为子进程),然后,分别在父进程和子进程内返回。

子进程永远返回0,而父进程返回子进程的ID。这样做的理由是,一个父进程可以fork出很多子进程,所以,父进程要记下每个子进程的ID,而子进程只需要调用getppid()就可以拿到父进程的ID。

Python的os模块封装了常见的系统调用,其中就包括fork,可以在Python程序中轻松创建子进程:

import os
print 'Process (%s) start...' % os.getpid()
pid = os.fork()
if pid == 0:
   print 'I am child process (%s) and my parent is %s.' % (os.getpid(), os.getppid())
else:
   print 'I (%s) just created a child process (%s).' % (os.getpid(), pid)
得到:
Process (2450) start...
I (2450) just created a child process (2451).
I am child process (2451) and my parent is 2450.

使用Multiprocessing查模块创建多进程。

multiprocessing模块提供了一个Process类来描述一个进程对象,创建子进程时,只需要传入一个执行函数和函数的参数即可完成一个Process实例的创建,用start()方法启动进程,用join()方法实现进程间的同步。join()方法可以等待子进程结束后再继续往下运行,通常用于进程间的同步。

# -*- coding:utf-8 -*-
from multiprocessing import Process
import os
# 子进程要执行的代码
def run_proc(name):
   print 'Run child process %s (%s)...' % (name, os.getpid())
if __name__ == '__main__':
   print 'Parent process %s.' % os.getpid()
   p = Process(target=run_proc, args=('test',))
   print 'Process will start.'
   p.start()
   p.join()
   print 'Process end.'
得到:
Parent process 2533.
Process will start.
Run child process test (2534)...
Process end.

multiprocessing模块提供了一个pool类来代表进程池对象

Pool可以提供指定数量的进程供用户调用,默认大小是cpu的核数,当有新的请求提交到pool中时,如果池还没有满,那么就会创建一个新的进程用来执行该请求,但如果池的进程数已经达到规定最大值,那么该请求就会等待,直到池中有进程结束才会创建新的进程来处理它。

# -*- coding:utf-8 -*-
from multiprocessing import Pool
import os, time, random
def long_time_task(name):
   print 'Run task %s (%s)...' % (name, os.getpid())
   start = time.time()
   time.sleep(random.random() * 3)
   end = time.time()
   print 'Task %s runs %0.2f seconds.' % (name, (end - start))
if __name__ == '__main__':
   print 'Parent process %s.' % os.getpid()
   p = Pool()
   for i in range(5 ):
       p.apply_async(long_time_task, args=(i,))
   print 'Waiting for all subprocesses done...'
   p.close()
   p.join()
   print 'All subprocesses done.'
得到:
Parent process 2541.
Waiting for all subprocesses done...
Run task 0 (2543)...
Run task 1 (2544)...
Run task 2 (2545)...
Run task 3 (2546)...
Task 0 runs 0.02 seconds.
Run task 4 (2543)...
Task 2 runs 0.60 seconds.
Task 4 runs 1.18 seconds.
Task 3 runs 1.26 seconds.
Task 1 runs 1.66 seconds.
All subprocesses done.

对Pool对象调用join()方法会等待所有子进程执行完毕,调用join()之前必须先调用close(),调用close()之后就不能继续添加新的Process了。

进程间的通信

Process之间肯定是需要通信的,操作系统提供了很多机制来实现进程间的通信。Python的multiprocessing模块包装了底层的机制,提供了Queue、Pipes等多种方式来交换数据。 两者的区别在于Pipe常用于两个进程间的通讯而Queue用于多个进程间实现通讯。

Queue通讯

Queue是多进程安全的队列,可以使用Queue实现多进程之间的数据传输,有两个方法:put和get进行Queue操作。

  • put方法用以插入数据队列中它可以有两个可选参数:blocked和timeout,如果blocked为True(默认值)并且timeout是正值,该方法会阻塞timeout指定的时间,直到该队列有剩余空间,如果超时,会抛出Queue.Full异常,如果blocked为False,但该Queue已满,则会立即抛出Queue.Full异常。

  • Get方法用以从队列读取并且删除一个元素。它可以有两个可选参数:blocked和timeout,如果blocked为True(默认值)并且timeout是正值,那么在等待时间内没有取到任何元素会抛出Queue.Empty异常,如果blocked为False,分两种情况:如果Queue有一个值木口月禾,则立即返回该值,否则如果队列为空,则立即抛出Queuq.Empty异常。

# -*- coding:utf-8 -*-
from multiprocessing import Process, Queue
import os, time, random
# 写数据进程执行的代码:
def write(q):
   for value in ['A', 'B', 'C']:
       print 'Put %s to queue...' % value
       q.put(value)
       time.sleep(random.random())
# 读数据进程执行的代码:
def read(q):
   while True:
       value = q.get(True)
       print 'Get %s from queue.' % value
if __name__ == '__main__':
   # 父进程创建Queue,并传给各个子进程:
   q = Queue()
   pw = Process(target=write, args=(q,))
   pr = Process(target=read, args=(q,))
   # 启动子进程pw,写入:
   pw.start()
   # 启动子进程pr,读取:
   pr.start()
   # 等待pw结束:
   pw.join()
   # pr进程里是死循环,无法等待其结束,只能强行终止:
   pr.terminate()
得到:
Put A to queue...
Get A from queue.
Put B to queue...
Get B from queue.
Put C to queue...
Get C from queue.
Pipes通讯

Pipe常用来在两个进程间进行通信,两个进程分别位于管道的两端。

Pipe方法返回(conn1,conn2)代表一个管道的两个端,Pipe方法有duplex参数,如果duplex参数为True(默认值),那么这个管道是全双工模式,也就是说conn1和conn2均可收发,若duplex为False,conn1只负责接收消息,conn2只负责发送消息。send和recv方法分别是发送和接收消息的方法。例如,在全双工模式下,可以调用conn1.send发送消息,conn1.recv接收消息。如果没有消息可接收,recv方法会一直阻塞。如果管道已经被关闭,那么recv方法会抛出EOFError.

import multiprocessing
import random
import time, os
def proc_send(pipe, urls):
   for url in urls:
       print "process(%s) send:%s" % (os.getpid(), url)
       pipe.send(url)
       time.sleep(random.random())
def proc_recv(pipe






请到「今天看啥」查看全文