Python的学习之旅———生产者消费者模型

生产者消费者模型

from multiprocessing import Process,Queue,JoinableQueue
import time
import random

def producer(name,food,q):
    for i in range(3):
        res='%s%s' %(food,i)
        time.sleep(random.randint(1,3))
        q.put(res)
        print('%s 生产了 %s' %(name,res))

def consumer(name,q):
    while True:
        res=q.get()
        if res is None:break
        time.sleep(random.randint(1, 3))
        print('%s 吃了 %s' %(name,res))
        q.task_done()

if __name__ == '__main__':
    q=JoinableQueue() #q.join()
    p1=Process(target=producer,args=('egon','泔水',q))
    p2=Process(target=producer,args=('贱哥','屎',q))
    c1=Process(target=consumer,args=('alex',q))
    c2=Process(target=consumer,args=('alex',q))
    c3=Process(target=consumer,args=('alex',q))
    c1.daemon=True
    c2.daemon=True
    c3.daemon=True

    p1.start()
    p2.start()
    c1.start()
    c2.start()
    c3.start()
    
    p1.join()
    p2.join()
    q.join()
    print('主')

关键点：

q.task_done()           消费者发通知，一个数据已经被处理完了 ，队列里有一个计数 生产了了多少个就计数多少个 比如10，taskdone 就-1.

q=JoinableQueue() #q.join()      可等待队列 join的是队列，队列里被取干净就是队列结束

先要等 生产者都生产完了

也就是 p1.join() 和p2.join() 都执行完了

再写q.join()   q.join() 执行完是收集到了足够的 q.task_done() 

这时候消费者 还或者，但是已经不可能在收到 队列里的的数值，所以把消费者设置成守护进程。