python 多線程優先隊列Queue
Queue模塊允許創建指定長度的隊列. 下面是Queue模塊的常用方法:
get():刪除并返回隊列中的一個項目
put(): 添加項目到隊列
qsize() : 返回隊列中元素的個數
empty(): 隊列為空返回True否則返回False
full(): 隊列已滿返回True,負責返回False
#!/usr/bin/pythonimport Queue import threading import time
exitFlag = 0
class myThread (threading.Thread): def init(self, threadID, name, q): threading.Thread.init(self) self.threadID = threadID self.name = name self.q = q def run(self): print "Starting " + self.name process_data(self.name, self.q) print "Exiting " + self.name
def process_data(threadName, q): while not exitFlag: queueLock.acquire() if not workQueue.empty(): data = q.get() queueLock.release() print "%s processing %s" % (threadName, data) else: queueLock.release() time.sleep(1)
threadList = ["Thread-1", "Thread-2", "Thread-3"] nameList = ["One", "Two", "Three", "Four", "Five"] queueLock = threading.Lock() workQueue = Queue.Queue(10) threads = [] threadID = 1
Create new threads
for tName in threadList: thread = myThread(threadID, tName, workQueue) thread.start() threads.append(thread) threadID += 1
Fill the queue
queueLock.acquire() for word in nameList: workQueue.put(word) queueLock.release()
Wait for queue to empty
while not workQueue.empty(): pass
Notify threads it's time to exit
exitFlag = 1
Wait for all threads to complete
for t in threads: t.join() print "Exiting Main Thread" </pre>
輸出結果如下:Starting Thread-1 Starting Thread-2 Starting Thread-3 Thread-1 processing One Thread-2 processing Two Thread-3 processing Three Thread-1 processing Four Thread-2 processing Five Exiting Thread-3 Exiting Thread-1 Exiting Thread-2 Exiting Main Thread