queue.Queue之task_done用法
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | 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 退出与阻塞
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | #-*-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俱乐部。