前端之家收集整理的这篇文章主要介绍了
python生产者和消费者模式实现(三)进程池方式,
前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
注意:如果要使用Pool(进程池方式)创建进程,就需要使用multiprocessing.Manager()中的 Queue(),而不是multiprocessing.Queue()
import time
import random
from multiprocessing import Pool,Manager
# 生产者
def producer(q,i):
food = 'Spam-%d' % i
time.sleep(random.uniform(1,2))
timeVal = time.strftime("%Y-%m-%d %H:%M:%S",time.localtime())
print('时间:%s\t生产者:%d\t生产了 Spam-%d' % (timeVal,i,i))
q.put(food)
# 消费者
def consumer(q,i):
while True:
food = q.get()
if not food: break
time.sleep(random.uniform(1,2))
timeVal = time.strftime("%Y-%m-%d %H:%M:%S",time.localtime())
print('时间:%s\t消费者: %d\t吃了 %s' % (timeVal,food))
if __name__ == '__main__':
q = Manager().Queue()
producterNum = 50
producterPoolNum = 5
producterPool = Pool(producterPoolNum)
for n in range(1,producterNum + 1):
producterPool.apply_async(producer,(q,n))
consumerPoolNum = 5
consumerPool = Pool(consumerPoolNum)
for n in range(1,consumerPoolNum + 1):
consumerPool.apply_async(consumer,n))
producterPool.close()
producterPool.join()
for n in range(1,consumerPoolNum + 1):
q.put(None)
consumerPool.close()
consumerPool.join()
print('end')