python中queue.Queue之task_done的用法

admin Python

2024-03-04 0 741

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 的异常。