python生产者和消费者模式实现(三)进程池方式

前端之家收集整理的这篇文章主要介绍了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')

猜你在找的Python相关文章