Python进程相关(三)生产者消费者模型

本文介绍了一种使用阻塞队列实现的生产者消费者模型,该模型允许生产者和消费者独立运行,通过队列进行通信,有效平衡两者处理能力。文章详细展示了如何利用Python的multiprocessing模块实现这一模型,并讨论了如何确保所有进程正确结束。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

一种解决问题的方法

生产者和消费者彼此之间不直接通讯,而通过阻塞队列来进行通讯,所以生产者生产完数据之后不用等待消费者处理,直接扔给阻塞队列,消费者不找生产者要数据,而是直接从阻塞队列里取,阻塞队列就相当于一个缓冲区,平衡了生产者和消费者的处理能力。

基于队列实现生产者消费者模型

# -*- coding: utf-8 -*-
"""
Created on Sun Dec 15 10:09:54 2019

@author: clt
"""
from multiprocessing  import Process,Queue
import time 
import random


#下边这种函数耦合的方式,有局限,生产的时候不能吃,吃的时候不能生产
'''
def producer(name):
    for i in range(10):
        res='包子%s'  %i
        time.sleep(random.randint(1,3))
        print ('厨师[%s]生产了<%s>'%(name,res))
        consumer('clt',res)
        
        
        
def consumer(name,res):
    time.sleep(random.randint(1,3))
    print ('吃货[%s]吃了<%s>'%(name,res))
    
'''

def producer(name,q):
    for i in range(10):
        res='包子%s'  %i
        time.sleep(random.randint(1,3))
        q.put(res)
        print ('厨师[%s]生产了<%s>'%(name,res))
        
        
        
def consumer(name,q):
    while True:
        res=q.get()
        time.sleep(random.randint(1,3))
    
        print ('吃货[%s]吃了<%s>'%(name,res))
    
    
if __name__=='__main__':
    #队列
    q=Queue()
    #生产者们
    p1=Process(target=producer,args=('egon',q))
    #消费者们
    c1=Process(target=consumer,args=('clt',q))
    
    p1.start()
    c1.start()
    
    print('主进程')

结果
主进程
厨师[egon]生产了<包子0>
厨师[egon]生产了<包子1>
吃货[clt]吃了<包子0>
吃货[clt]吃了<包子1>
厨师[egon]生产了<包子2>
吃货[clt]吃了<包子2>
厨师[egon]生产了<包子3>
厨师[egon]生产了<包子4>
吃货[clt]吃了<包子3>
厨师[egon]生产了<包子5>
厨师[egon]生产了<包子6>
吃货[clt]吃了<包子4>
厨师[egon]生产了<包子7>
吃货[clt]吃了<包子5>
厨师[egon]生产了<包子8>
吃货[clt]吃了<包子6>
厨师[egon]生产了<包子9>
吃货[clt]吃了<包子7>
吃货[clt]吃了<包子8>
吃货[clt]吃了<包子9>

但上边的代码出现问题,主进程一直卡在那,因为消费者进程一直不死

  • JoinableQueue与守护进程应用
    法一
# -*- coding: utf-8 -*-
"""
Created on Sun Dec 15 10:09:54 2019

@author: clt
"""
from multiprocessing  import Process,Queue
import time 
import random


#下边这种函数耦合的方式,有局限,生产的时候不能吃,吃的时候不能生产
'''
def producer(name):
    for i in range(10):
        res='包子%s'  %i
        time.sleep(random.randint(1,3))
        print ('厨师[%s]生产了<%s>'%(name,res))
        consumer('clt',res)
        
        
        
def consumer(name,res):
    time.sleep(random.randint(1,3))
    print ('吃货[%s]吃了<%s>'%(name,res))
    
'''

def producer(name,q):
    for i in range(10):
        res='包子%s'  %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))
    
    
if __name__=='__main__':
    #队列
    q=Queue()
    #生产者们
    p1=Process(target=producer,args=('egon',q))
    #消费者们
    c1=Process(target=consumer,args=('clt',q))
    
    p1.start()
    c1.start()
    
    
    p1.join()     #保证厨师运行完了
    q.put(None)
    print('主进程')
    
结果:
厨师[egon]生产了<包子0>
厨师[egon]生产了<包子1>
吃货[clt]吃了<包子0>
吃货[clt]吃了<包子1>
厨师[egon]生产了<包子2>
吃货[clt]吃了<包子2>
厨师[egon]生产了<包子3>
吃货[clt]吃了<包子3>
厨师[egon]生产了<包子4>
吃货[clt]吃了<包子4>
厨师[egon]生产了<包子5>
吃货[clt]吃了<包子5>
厨师[egon]生产了<包子6>
厨师[egon]生产了<包子7>
吃货[clt]吃了<包子6>
厨师[egon]生产了<包子8>
吃货[clt]吃了<包子7>
厨师[egon]生产了<包子9>
主进程
吃货[clt]吃了<包子8>
吃货[clt]吃了<包子9>

这次就能保证所有进程运行结束

如果多个生产者和消费者:
法一:有几个生产者就join几次,有几个消费者就发几次None信号
否则主进程就一直不结束

# -*- coding: utf-8 -*-
"""
Created on Sun Dec 15 10:09:54 2019

@author: clt
"""
from multiprocessing  import Process,Queue
import time 
import random

def producer(name,q):
    for i in range(2):
        res='包子%s'  %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))
    
    
if __name__=='__main__':
    #队列
    q=Queue()
    #生产者们
    p1=Process(target=producer,args=('ccc',q))
    p2=Process(target=producer,args=('bbb',q))
    p3=Process(target=producer,args=('aaa',q))
   
    #消费者们
    c1=Process(target=consumer,args=('ccc',q))
    c2=Process(target=consumer,args=('xxx',q))
    c3=Process(target=consumer,args=('yyy',q))
    
    p1.start()
    p2.start()
    p3.start()
    c1.start()
    c2.start()
    c3.start()
    
    
    p1.join()     #保证厨师正常运行完了
    p2.join()
    p3.join()
         
    q.put(None)   #有几个消费者就发几个结束信号,因为放一个就被取一个取走了
    q.put(None)
    q.put(None)
    
    print('主进程')

结果:
厨师[ccc]生产了<包子0>
厨师[bbb]生产了<包子0>
吃货[xxx]吃了<包子0>
厨师[aaa]生产了<包子0>
厨师[ccc]生产了<包子1>
吃货[ccc]吃了<包子0>
厨师[bbb]生产了<包子1>
吃货[xxx]吃了<包子1>
吃货[ccc]吃了<包子1>
吃货[yyy]吃了<包子0>
厨师[aaa]生产了<包子1>
主进程
吃货[xxx]吃了<包子1>

上边方法一太low
法二,是用joinable

# -*- coding: utf-8 -*-
"""
Created on Sun Dec 15 10:09:54 2019

@author: clt
"""
from multiprocessing  import Process,JoinableQueue
import time 
import random


def producer(name,q):
    for i in range(2):
        res='包子%s'  %i
        time.sleep(random.randint(1,3))
        q.put(res)
        print ('厨师[%s]生产了<%s>'%(name,res))
        

        
        
        
def consumer(name,q):
    while True:
        res=q.get()
        time.sleep(random.randint(1,3))
    
        print ('吃货[%s]吃了<%s>'%(name,res))
        q.task_done()  #每取走一个  发一个信号 给join发
    
    
if __name__=='__main__':
    #队列
    q=JoinableQueue()   #根本原理还是发信号  但是是消费者发信号
    #生产者们
    p1=Process(target=producer,args=('ccc',q))
    p2=Process(target=producer,args=('bbb',q))
  
   
    #消费者们
    c1=Process(target=consumer,args=('ccc',q))
    c2=Process(target=consumer,args=('xxx',q))
    c3=Process(target=consumer,args=('yyy',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('主进程')
    
 结果:

厨师[bbb]生产了<包子0>
厨师[ccc]生产了<包子0>
吃货[ccc]吃了<包子0>
厨师[bbb]生产了<包子1>
厨师[ccc]生产了<包子1>
吃货[yyy]吃了<包子1>
吃货[xxx]吃了<包子0>
吃货[ccc]吃了<包子1>
主进程
    
    
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值