Queue模組可以用來進行線程間通訊,讓各個線程之間共用資料。現在,我們建立一個隊列,讓
生產者(線程)把新生產的貨物放進去供消費者(線程)使用。
在下列的例子中我們將示範生產者生產貨物,然後把貨物放到一個隊列 之類的資料結構中,生產貨物所要花費的時間
無法預先確定。消費者消耗生產者生產 的貨物的時間也是不確定的。
我們首先用到的還是之前我們建立的多線程通用類myThreads.py
'''<br />Created on 2010-12-16<br />@author: Administrator<br />'''<br />import threading<br />from time import ctime<br />class MyThread(threading.Thread):<br /> def __init__(self,func,args,name=''):<br /> threading.Thread.__init__(self)<br /> self.name=name<br /> self.func=func<br /> self.args=args<br /> def getResult(self):<br /> return self.res<br /> def run(self):<br /> print 'starting',self.name,'at:',ctime()<br /> self.res=apply(self.func,self.args)<br /> print self.name,'finished at:',ctime()
接下來就是我們的主程式,生產者-消費者問題 (prodcons.py)
'''<br />Created on 2010-12-17<br />@author: Administrator<br />'''<br />from random import randint<br />from time import sleep<br />from Queue import Queue<br />from myThread import MyThread<br />def writeQ(queue):<br /> print 'producting object for Q...',<br /> queue.put('test',1)<br /> print "size now",queue.qsize()<br />def readQ(queue):<br /> val=queue.get(1)<br /> print 'consumed object from Q... size now',queue.qsize()<br />def writer(queue,loops):<br /> for i in range(loops):<br /> writeQ(queue)<br /> sleep(randint(1,3))<br />def reader(queue,loops):<br /> for i in range(loops):<br /> readQ(queue)<br /> sleep(randint(2,5))<br />funcs=[writer,reader]<br />nfuncs=range(len(funcs))<br />def main():<br /> nloops=randint(2,5)<br /> q=Queue(32)<br /> threads=[]</p><p> for i in nfuncs:<br /> t=MyThread(funcs[i],(q,nloops),funcs[i].__name__)<br /> threads.append(t)<br /> for i in nfuncs:<br /> threads[i].start()<br /> for i in nfuncs:<br /> threads[i].join()</p><p> print 'all DONE'<br />if __name__=='__main__':<br /> main()
運行如下: