multiprocessing.Queue 多个进程生产和多个进程消费怎么处理

在这个示例中,我们创建了一个队列 q,并通过 multiprocessing.Manager().Queue() 来确保队列可以在多个进程之间共享。我们定义了 consumerproducer 函数,分别用于从队列中获取数据和向队列中放入数据。

在主进程中,我们创建了多个消费者和生产者进程,并将它们启动。生产者进程将数据放入队列,消费者进程从队列中取出数据并处理。生产者进程完成后,我们向队列发送 None 作为结束信号,告知消费者没有更多数据。每个消费者在接收到 None 后会停止工作。

注意,我们在 consumer 函数中使用了 queue.task_done() 来标记任务完成。这是可选的,但在使用 join() 方法等待队列中的所有任务完成时很有用。

这个模式允许多个生产者并发地向队列中放入数据,同时多个消费者并发地从队列中取出并处理数据,直到所有生产者完成生产,消费者接收到结束信号。

当使用 multiprocessing.Queue 进行多个生产者和多个消费者的场景时,队列可以很好地协调这些进程。以下是一个示例,展示了如何创建多个生产者和多个消费者,它们共享同一个队列:

python 复制代码
# encoding:utf-8
import multiprocessing
import time
import random

def consumer(queue):
    """
    作者:阙辉
    """
    while True:
        item = queue.get()  # 从队列中获取数据
        if item is None:
            print(f"Consumer {multiprocessing.current_process().name} received end signal.")
            queue.task_done()  # 标记任务完成
            break
        print(f"Consumer {multiprocessing.current_process().name} received {item}")
        time.sleep(random.uniform(0.5, 1.5))  # 模拟处理时间
        queue.task_done()  # 标记任务完成

def producer(queue, items):
    """
    作者:阙辉
    """
    for item in items:
        print(f"Producer {multiprocessing.current_process().name} sent {item}")
        queue.put(item)
        time.sleep(random.uniform(0.5, 1.5))  # 模拟生产时间

if __name__ == '__main__':
    manager = multiprocessing.Manager()
    q = manager.Queue()  # 使用 Manager.Queue 来支持多个生产者和消费者模式

    # 创建多个消费者进程
    consumers = [multiprocessing.Process(target=consumer, args=(q,)) for _ in range(4)]

    # 创建多个生产者进程
    producers = [multiprocessing.Process(target=producer, args=(q, range(20))) for _ in range(4)]

    # 启动所有消费者进程
    for c in consumers:
        c.start()

    # 启动所有生产者进程
    for p in producers:
        p.start()

    # 等待所有生产者完成
    for p in producers:
        p.join()

    # 发送结束信号,告知所有消费者没有更多数据
    for _ in consumers:
        q.put(None)

    # 等待所有消费者完成
    for c in consumers:
        c.join()

    print("All tasks completed.")
相关推荐
懒大王爱吃狼1 小时前
Python教程:python枚举类定义和使用
开发语言·前端·javascript·python·python基础·python编程·python书籍
瓜牛_gn1 小时前
mysql特性
数据库·mysql
秃头佛爷2 小时前
Python学习大纲总结及注意事项
开发语言·python·学习
奶糖趣多多2 小时前
Redis知识点
数据库·redis·缓存
阿伟*rui2 小时前
配置管理,雪崩问题分析,sentinel的使用
java·spring boot·sentinel
CoderIsArt3 小时前
Redis的三种模式:主从模式,哨兵与集群模式
数据库·redis·缓存
深度学习lover3 小时前
<项目代码>YOLOv8 苹果腐烂识别<目标检测>
人工智能·python·yolo·目标检测·计算机视觉·苹果腐烂识别
XiaoLeisj4 小时前
【JavaEE初阶 — 多线程】单例模式 & 指令重排序问题
java·开发语言·java-ee
paopaokaka_luck4 小时前
【360】基于springboot的志愿服务管理系统
java·spring boot·后端·spring·毕业设计
dayouziei4 小时前
java的类加载机制的学习
java·学习