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 的异常。
总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持IT俱乐部。