Python 中消费者 - 生产者模式详解_python生产者消费者示例-CSDN博客

import threading
import time
import queue

# 定义缓冲区
buffer_queue = queue.Queue()

# 定义生产者函数
def producer():
    for i in range(5):
        print(f"生产者生产了数据: {i}")
        buffer_queue.put(i)
        time.sleep(1)

# 定义消费者函数
def consumer():
    while True:
        item = buffer_queue.get()
        if item is None:
            break
        print(f"消费者消费了数据: {item}")
        buffer_queue.task_done()

# 创建生产者和消费者线程
producer_thread = threading.Thread(target=producer)
consumer_thread = threading.Thread(target=consumer)

# 启动线程
producer_thread.start()
consumer_thread.start()

# 等待生产者线程结束
producer_thread.join()

# 向队列中放入 None 表示生产结束
buffer_queue.put(None)

# 等待消费者线程结束
consumer_thread.join()

print("所有数据处理完毕")

代码解释

队列的创建：借助 queue.Queue() 创建一个线程安全的队列 buffer_queue，作为生产者和消费者之间的缓冲区。
生产者函数：producer 函数会循环 5 次，每次生成一个数据并将其放入队列，之后休眠 1 秒。
消费者函数：consumer 函数持续从队列中取出数据，若取出的是 None，则表示生产结束，退出循环。
线程的创建与启动：分别创建生产者和消费者线程，然后启动它们。
线程的同步：等待生产者线程结束后，向队列中放入 None 以通知消费者生产结束，最后等待消费者线程结束。

使用协程实现

Python 的 asyncio 模块可用于实现异步编程，也能用来实现消费者 - 生产者模式。以下是一个示例：

import asyncio

# 定义缓冲区
buffer_queue = asyncio.Queue()

# 定义生产者协程
async def producer():
    for i in range(5):
        print(f"生产者生产了数据: {i}")
        await buffer_queue.put(i)
        await asyncio.sleep(1)

# 定义消费者协程
async def consumer():
    while True:
        item = await buffer_queue.get()
        if item is None:
            break
        print(f"消费者消费了数据: {item}")
        buffer_queue.task_done()

async def main():
    # 创建生产者和消费者任务
    producer_task = asyncio.create_task(producer())
    consumer_task = asyncio.create_task(consumer())

    # 等待生产者任务结束
    await producer_task

    # 向队列中放入 None 表示生产结束
    await buffer_queue.put(None)

    # 等待消费者任务结束
    await consumer_task

# 运行主协程
asyncio.run(main())

print("所有数据处理完毕")