python中queue.Queue之task_done的用法

泉鑫

queue.Queue之task_done用法

1. from threading import Thread
   
2. import time
   
3. from queue import Queue
   
4. q = Queue()
   
5. def run():
   
6.     while True:
   
7.         msg = q.get()
   
8.         print(msg)
   
9.         time.sleep(1)
   
10.         q.task_done()
    
11. 
    
12. 
    
13. for i in range(10):
    
14.     q.put(5-i)
    
15. 
    
16. for i in range(3):
    
17.     t = Thread(target=run)
    
18.     t.setDaemon(True)
    
19.     t.start()
    
20. q.join()

复制代码

task_done目的是为了告诉队列， 我当前已经获取到队列中的内容， 当队列获取到所有线程中的此信号时， 会去会首先会不断检测是否有地方正在获取消息内容， 检测到后会继续检测队列中是否还有消息， 若有， 则会分发消息给get的地方， 如果没有， 则会终端join带来的阻塞， 继续往下执行。
如果没有检测到有地方正在get到消息的话， 队列会不断的做这个检测， 从而阻塞在join那里， 也就是说， 如果在上述while True中加入了break语句， 导致所有线程不再get队列消息（实际线程已结束）, 那么队列将会一直阻塞在join的地方

Queue.queue 退出与阻塞

1. #-*-coding:utf-8-*-
   
2. import threading
   
3. import queue
   
4. import time
   
5. import random
   
6. 
   
7. '''
   
8. 1.创建一个 Queue.Queue() 的实例，然后使用数据对它进行填充。
   
9. 2.将经过填充数据的实例传递给线程类，后者是通过继承 threading.Thread 的方式创建的。
   
10. 3.每次从队列中取出一个项目，并使用该线程中的数据和 run 方法以执行相应的工作。
    
11. 4.在完成这项工作之后，使用 queue.task_done() 函数向任务已经完成的队列发送一个信号。
    
12. 5.对队列执行 join 操作，实际上意味着等到队列为空，再退出主程序。
    
13. '''
    
14. 
    
15. class jdThread(threading.Thread):
    
16.     def __init__(self,index,queue):
    
17.         threading.Thread.__init__(self)
    
18.         self.index = index
    
19.         self.queue = queue
    
20. 
    
21.     def run(self):
    
22.         while True:
    
23.             time.sleep(1)
    
24.             item = self.queue.get(False,)
    
25.             if self.queue.empty():
    
26.                 print("here")
    
27.                 break
    
28. 
    
29.             print("序号：",self.index,"任务",item,"完成")
    
30.             self.queue.task_done()#task_done方法使得未完成的任务数量-1
    
31. 
    
32. if __name__ == '__main__':
    
33. 
    
34.     q = queue.Queue(0)
    
35. 
    
36.     for i in range(6):
    
37.         print ("the  i is :%d" % i)
    
38.         q.put(i)#put方法使得未完成的任务数量+1
    
39. 
    
40.     '''
    
41.     初始化函数接受一个数字来作为该队列的容量，如果传递的是
    
42.     一个小于等于0的数，那么默认会认为该队列的容量是无限的.
    
43.     '''
    
44.     for i in range(2):
    
45.         jdThread(i,q).start()#两个线程同时完成任务
    
46. 
    
47. 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 的异常。

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

来源:https://www.jb51.net/python/316203ee1.htm
免责声明：由于采集信息均来自互联网，如果侵犯了您的权益，请联系我们【E-Mail:cb@itdo.tech】 我们会及时删除侵权内容，谢谢合作！