最近在使用python的队列做一点东西,发现了queue.join(),这和以前使用的多线程thread.join()有什么不一样呢,自己做了个小实验,描述都在代码注释里了
from threading import Thread import time import random from queue import Queue from collections import deque #创建队列,设置队列最大数限制为3个 queue = Queue(3) #生产者线程 class Pro_Thread(Thread): def run(self): #原材料准备,等待被生产 tasks = deque([1, 2, 3, 4, 5, 6, 7, 8]) global queue while True: try: #从原材料左边开始生产 task = tasks.popleft() queue.put(task) print("生产", task, "现在队列数:", queue.qsize()) #休眠随机时间 time.sleep(random.random()) #如果原材料被生产完,生产线程跳出循环 except IndexError: print("原材料已被生产完毕") break #消费者线程 class Con_Thread(Thread): def run(self): global queue while True: if queue.not_empty: #通过get(),这里已经将队列减去了1 task = queue.get() time.sleep(2) #这里可能队列数已经空了,但是消费者手里还有正在消费的队列 #发出完成的信号,不发的话,join会永远阻塞,程序不会停止 queue.task_done() print("消费", task) else: break #r入口方法,主线程 def main(): Pro_1 = Pro_Thread() #把生产线程列为守护线程,否则主线程结束之后不会销毁该线程,程序不会停止,影响实验结果 Pro_1.setDaemon(True) #启动线程 Pro_1.start() for i in range(2): Con_i = Con_Thread() # 把两个消费者线程列为守护线程,否则主线程结束之后不会销毁该线程,程序不会停止,影响实验结果 Con_i.setDaemon(True) #启动线程 Con_i.start() global queue #这里休眠一秒钟,等到队列有值,否则队列创建时是空的,主线程直接就结束了,实验失败,造成误导 time.sleep(1) #接收信号,主线程在这里等待队列被处理完毕后再做下一步 queue.join() #给个标示,表示主线程已经结束 print("主线程结束") if __name__ == '__main__': main()