本文實例講述了Python多進程機制。分享給大家供大家參考。具體如下:
在以前只是接觸過PYTHON的多線程機制,今天搜了一下多進程,相關文章好像不是特別多。看了幾篇,小試了一把。程序如下,主要內容就是通過PRODUCER讀一個本地文件,一行一行的放到隊列中去。然后會有相應的WORKER從隊列中取出這些行。
?
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 import multiprocessing import os import sys import Queue import time def writeQ(q,obj): q.put(obj,True,None) print "put size: ",q.qsize() def readQ(q): ret = q.get(True,1) print "get size: ",q.qsize() return ret def producer(q): time.sleep(5) #讓進行休息幾秒 方便ps命令看到相關內容 pid = os.getpid() handle_file = '/home/dwapp/joe.wangh/test/multiprocess/datafile' with open(handle_file,'r') as f: #with...as... 這個用法今天也是第一次看到的 for line in f: print "producer <" ,pid , "> is doing: ",line writeQ(q,line.strip()) q.close() def worker(q): time.sleep(5) #讓進行休息幾秒 方便ps命令看到相關內容 pid = os.getpid() empty_count = 0 while True: try: task = readQ(q) print "worker <" , pid , "> is doing: " ,task ''' 如果這里不休眠的話 一般情況下所有行都會被同一個子進程讀取到 為了使實驗效果更加清楚 在這里讓每個進程讀取完 一行內容時候休眠5s 這樣就可以讓其他的進程到隊列中進行讀取 ''' time.sleep(5) except Queue.Empty: empty_count += 1 if empty_count == 3: print "queue is empty, quit" q.close() sys.exit(0) def main(): concurrence = 3 q = multiprocessing.Queue(10) funcs = [producer , worker] for i in range(concurrence-1): funcs.append(worker) for item in funcs: print str(item) nfuncs = range( len(funcs) ) processes = [] for i in nfuncs: p = multiprocessing.Process(target=funcs[i] , args=(q,)) processes.append(p) print "concurrence worker is : ",concurrence," working start" for i in nfuncs: processes[i].start() for i in nfuncs: processes[i].join() print "all DONE" if __name__ == '__main__': main()新聞熱點
疑難解答