插件窝 干货文章 python中queue.Queue之task_done的用法

python中queue.Queue之task_done的用法

queue 队列 阻塞 线程 265    来源:    2024-10-17

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目的是为了告诉队列, 我当前已经获取到队列中的内容, 当队列获取到所有线程中的此信号时, 会去会首先会不断检测是否有地方正在获取消息内容, 检测到后会继续检测队列中是否还有消息, 若有, 则会分发消息给get的地方, 如果没有, 则会终端join带来的阻塞, 继续往下执行。

如果没有检测到有地方正在get到消息的话, 队列会不断的做这个检测, 从而阻塞在join那里, 也就是说, 如果在上述while True中加入了break语句, 导致所有线程不再get队列消息(实际线程已结束), 那么队列将会一直阻塞在join的地方

Queue.queue 退出与阻塞

#-*-coding:utf-8-*-
import threading
import queue
import time
import random
 
'''
1.创建一个 Queue.Queue() 的实例,然后使用数据对它进行填充。
2.将经过填充数据的实例传递给线程类,后者是通过继承 threading.Thread 的方式创建的。
3.每次从队列中取出一个项目,并使用该线程中的数据和 run 方法以执行相应的工作。
4.在完成这项工作之后,使用 queue.task_done() 函数向任务已经完成的队列发送一个信号。
5.对队列执行 join 操作,实际上意味着等到队列为空,再退出主程序。
'''
 
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("序号:",self.index,"任务",item,"完成")
            self.queue.task_done()#task_done方法使得未完成的任务数量-1
 
if __name__ == '__main__':
 
    q = queue.Queue(0)
 
    for i in range(6):
        print ("the  i is :%d" % i)
        q.put(i)#put方法使得未完成的任务数量+1
 
    '''
    初始化函数接受一个数字来作为该队列的容量,如果传递的是
    一个小于等于0的数,那么默认会认为该队列的容量是无限的.
    '''
    for i in range(2):
        jdThread(i,q).start()#两个线程同时完成任务
 
print ("I am here ")

Queue.get()默认的也是阻塞方式读取数据,队列为空时,不会抛出 except Queue.Empty ,而是进入阻塞直至超时。 加上block=False 的参数, 线程会以抛出异常的方式退出,从而进程也可退出。

Queue.put 跟Queue.get()类似, Queue.put()默认有 block = True 和 timeou 两个参数。当 block = True 时,写入是阻塞式的,阻塞时间由 timeou 确定。当队列q被(其他线程)写满后,这段代码就会阻塞,直至其他线程取走数据。Queue.put()方法加上 block=False 的参数,为非阻塞方式写队列,当队列满时会抛出 exception Queue.Full 的异常。

总结

以上为个人经验,希望对您有所帮助。