python 生产者消费者模型 multiprocessing Process Queue 队列

from multiprocessing import Queue, Process


def consumer(q, name):  # 消费者
    while 1:
        info = q.get()  # 阻塞等待获取数据，如果有数据直接获取，如果没有数据阻塞等待
        if info:  # 收到None作为结束标识
            print(' %s 得到 %s' % (name, info))
        else:  # 当消费者获得队列中数据时，如果获得的是None，就是获得到了生产者不再生产数据的标识
            break  # 此时消费者结束


def producer(q, product):  # 生产者
    for i in range(20):
        info = product + '的产品%s号' % str(i)
        q.put(info)  # 阻塞，如果可以继续往队列中放数据，就直接放，不能放就阻塞等待


if __name__ == '__main__':
    q = Queue(10)  # 队列的最大长度
    p_pro1 = Process(target=producer, args=(q, '美国产'))  # 生产者
    p_pro2 = Process(target=producer, args=(q, '英国产'))
    p_pro3 = Process(target=producer, args=(q, '日本产'))

    p_con1 = Process(target=consumer, args=(q, '消费者01'))  # 消费者
    p_con2 = Process(target=consumer, args=(q, '消费者02'))

    p_l = [p_con1, p_con2, p_pro1, p_pro2, p_pro3]
    [i.start() for i in p_l]  # 开启多进程

    p_pro1.join()
    p_pro2.join()
    p_pro3.join()
    q.put(None)  # 几个消费者就接受几个结束标识 None作为结束标识
    q.put(None)

---

q.get()  # 阻塞等待获取数据，如果有数据直接获取，如果没有数据，阻塞等待
q.put()  # 阻塞，如果可以继续往队列中放数据，就直接放，不能放就阻塞等待

q.get_nowait()  # 不阻塞，如果有数据直接获取，没有数据就报错
q.put_nowait()  # 不阻塞，如果可以继续往队列中放数据，就直接放，不能放就报错