C++如何实现生产者消费者模式_C++多线程同步与互斥经典案例

生产者消费者模式通过共享缓冲区、互斥锁和条件变量实现多线程同步,解决数据生产与消费速度不匹配问题,C++中利用queue、mutex和condition_variable完成线程间协调,确保线程安全与高效通信。

生产者消费者模式是多线程编程中的经典问题,用于解决生产数据与消费数据速度不匹配的问题。在C++中,通常使用互斥锁(std::mutex)、条件变量(std::condition_variable)和队列(std::queue)来实现线程间的同步与互斥。

核心组件说明

要实现生产者消费者模型,需要以下几个关键元素:

  • 共享缓冲区:通常是一个队列,用于暂存生产者生成的数据,供消费者取出处理。
  • 互斥锁(mutex):保护共享资源,防止多个线程同时访问导致数据竞争。
  • 条件变量(condition_variable):用于线程间通信,当缓冲区为空时让消费者等待,当缓冲区有数据时通知消费者;当缓冲区满时让生产者等待,有空间时通知生产者。

代码实现示例

下面是一个基于固定大小缓冲区的生产者消费者模型实现:

#include 
#include 
#include 
#include 
#include 
#include 

std::queue buffer;
std::mutex mtx;
std::condition_variable not_empty;
std::condition_variable not_full;
const int max_buffer_size = 5;

void producer(int id) {
    for (int i = 0; i < 10; ++i) {
        std::unique_lock lock(mtx);
        not_full.wait(lock, []() { return buffer.size() < max_buffer_size; });
        
        buffer.push(i);
        std::cout << "生产者 " << id << " 生产: " << i << std::endl;
        
        not_empty.notify_one();
        std::this_thread::sleep_for(std::chrono::milliseconds(100));
    }
}

void consumer(int id) {
    for (int i = 0; i < 10; ++i) {
        std::unique_lock lock(mtx);
        not_empty.wait(lock, []() { return !buffer.empty(); });
        
        int value = buffer.front();
        buffer.pop();
        std::cout << "消费者 " << id << " 消费: " << value << std::endl;
        
        not_full.notify_one();
        std::this_thread::sleep_for(std::chrono::milliseconds(150));
    }
}

主函数中启动多个生产者和消费者线程:

int main() {
    std::thread p1(producer, 1);
    std::thread p2(producer, 2);
    std::thread c1(consumer, 1);
    std::thread c2(consumer, 2);

    p1.join();
    p2.join();
    c1.join();
    c2.join();

    return 0;
}

关键点解析

该实现中几个重要细节:

  • 使用 std::unique_lock 配合条件变量,支持在等待时释放锁,并在被唤醒后重新获取。
  • 条件变量的 wait 方法接受一个谓词(lambda表达式),避免虚假唤醒问题。
  • 每次修改缓冲区后调用 notify_one() 唤醒一个等待线程,也可以使用 notify_all() 唤醒所有等待线程。
  • 通过 std::this_thread::sleep_for 模拟生产/消费耗时,使输出更清晰。

应用场景与扩展

这种模式广泛应用于任务调度、消息队列、日志处理等场景。可以根据需求进行扩展:

  • 使用循环队列或智能指针管理对象生命周期。
  • 添加停止信号(如标志位)优雅关闭线程。
  • 使用 std::shared_mutex 实现读写优化(若允许多个消费者同时读)。
  • 封装成模板类,支持不同类型的数据。

基本上就这些。掌握这个模型对理解多线程同步机制非常有帮助。