python 多线程使用queue式例

参考自：https://blog.csdn.net/u011655220/article/details/79037032

from threading import Thread
import time
import random
from queue import Queue
from collections import deque
import socket

#创建队列，设置队列最大数限制为3个
queue = Queue()

# #生产者线程
class Pro_Thread(Thread):
    def run(self):
        while True:
            time.sleep(10)
            #原材料准备，等待被生产
            tasks = deque([1,2,4,3,4])
            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 __init__(self, i):
        super().__init__()
        self.id = i
    def run(self):
        global queue
        while True:
            try:
                #通过get(),这里已经将队列减去了1
                task = queue.get()
                time.sleep(1)
                #这里可能队列数已经空了，但是消费者手里还有正在消费的队列
                #发出完成的信号，不发的话，join会永远阻塞，程序不会停止
            except queue.Empty:
                pass

def main():

    for i in range(5):
        Con_i = Con_Thread(i)
        # 把两个消费者线程列为守护线程，否则主线程结束之后不会销毁该线程，程序不会停止，影响实验结果
        #启动线程
        Con_i.start()

    # 模拟tcp阻塞式样接收数据
    td = Pro_Thread()
    td.start()
    td.join()

    print("主线程结束")

if __name__ == '__main__':
    main()