快捷導(dǎo)航

python中queue.Queue之task_done的用法

更新時(shí)間：2024年02月23日 10:45:57 作者：碧落&凡塵

這篇文章主要介紹了python中queue.Queue之task_done的用法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

queue.Queue之task_done用法

from threading import Thread
import time
from queue import Queue
q = Queue()
def run():
    while True:
        msg = q.get()
        print(msg)
        time.sleep(1)
        q.task_done()
 
 
for i in range(10):
    q.put(5-i)
 
for i in range(3):
    t = Thread(target=run)
    t.setDaemon(True)
    t.start()
q.join()

task_done目的是為了告訴隊(duì)列，我當(dāng)前已經(jīng)獲取到隊(duì)列中的內(nèi)容，當(dāng)隊(duì)列獲取到所有線程中的此信號(hào)時(shí)，會(huì)去會(huì)首先會(huì)不斷檢測(cè)是否有地方正在獲取消息內(nèi)容，檢測(cè)到后會(huì)繼續(xù)檢測(cè)隊(duì)列中是否還有消息，若有，則會(huì)分發(fā)消息給get的地方，如果沒有，則會(huì)終端join帶來(lái)的阻塞，繼續(xù)往下執(zhí)行。

如果沒有檢測(cè)到有地方正在get到消息的話，隊(duì)列會(huì)不斷的做這個(gè)檢測(cè)，從而阻塞在join那里，也就是說(shuō)，如果在上述while True中加入了break語(yǔ)句，導(dǎo)致所有線程不再get隊(duì)列消息（實(shí)際線程已結(jié)束）, 那么隊(duì)列將會(huì)一直阻塞在join的地方

Queue.queue 退出與阻塞

#-*-coding:utf-8-*-
import threading
import queue
import time
import random
 
'''
1.創(chuàng)建一個(gè) Queue.Queue() 的實(shí)例，然后使用數(shù)據(jù)對(duì)它進(jìn)行填充。
2.將經(jīng)過(guò)填充數(shù)據(jù)的實(shí)例傳遞給線程類，后者是通過(guò)繼承 threading.Thread 的方式創(chuàng)建的。
3.每次從隊(duì)列中取出一個(gè)項(xiàng)目，并使用該線程中的數(shù)據(jù)和 run 方法以執(zhí)行相應(yīng)的工作。
4.在完成這項(xiàng)工作之后，使用 queue.task_done() 函數(shù)向任務(wù)已經(jīng)完成的隊(duì)列發(fā)送一個(gè)信號(hào)。
5.對(duì)隊(duì)列執(zhí)行 join 操作，實(shí)際上意味著等到隊(duì)列為空，再退出主程序。
'''
 
class jdThread(threading.Thread):
    def __init__(self,index,queue):
        threading.Thread.__init__(self)
        self.index = index
        self.queue = queue
 
    def run(self):
        while True:
            time.sleep(1)
            item = self.queue.get(False,)
            if self.queue.empty():
                print("here")
                break
 
            print("序號(hào)：",self.index,"任務(wù)",item,"完成")
            self.queue.task_done()#task_done方法使得未完成的任務(wù)數(shù)量-1
 
if __name__ == '__main__':
 
    q = queue.Queue(0)
 
    for i in range(6):
        print ("the  i is :%d" % i)
        q.put(i)#put方法使得未完成的任務(wù)數(shù)量+1
 
    '''
    初始化函數(shù)接受一個(gè)數(shù)字來(lái)作為該隊(duì)列的容量，如果傳遞的是
    一個(gè)小于等于0的數(shù)，那么默認(rèn)會(huì)認(rèn)為該隊(duì)列的容量是無(wú)限的.
    '''
    for i in range(2):
        jdThread(i,q).start()#兩個(gè)線程同時(shí)完成任務(wù)
 
print ("I am here ")

Queue.get()默認(rèn)的也是阻塞方式讀取數(shù)據(jù)，隊(duì)列為空時(shí)，不會(huì)拋出 except Queue.Empty ，而是進(jìn)入阻塞直至超時(shí)。加上block=False 的參數(shù), 線程會(huì)以拋出異常的方式退出，從而進(jìn)程也可退出。

Queue.put 跟Queue.get()類似, Queue.put（）默認(rèn)有 block = True 和 timeou 兩個(gè)參數(shù)。當(dāng) block = True 時(shí)，寫入是阻塞式的，阻塞時(shí)間由 timeou 確定。當(dāng)隊(duì)列q被（其他線程）寫滿后，這段代碼就會(huì)阻塞，直至其他線程取走數(shù)據(jù)。Queue.put（）方法加上 block=False 的參數(shù)，為非阻塞方式寫隊(duì)列，當(dāng)隊(duì)列滿時(shí)會(huì)拋出 exception Queue.Full 的異常。