c++11实现生产者和消费者

爱被打了一巴掌 2023-01-13 15:51 142阅读 0赞

#include <iostream>
    #include <list>
    #include <thread>
    #include <mutex>
    #include <condition_variable>
    #include <unistd.h>
    
    using namespace std;
    
    mutex mtx;
    condition_variable cv_producer;
    condition_variable cv_consumer;
    const int MAX_NUMBER = 10; 
    int resource_count = 0;
    list<string> my_store;
    
    void ProducerThread()
    {
        for (int count = 0; count < 10; count++) {
            unique_lock<mutex> lck(mtx);
            if (MAX_NUMBER <= resource_count) {
               cv_producer.wait(lck);
            }   
            resource_count++;
            string s = "resource:" + to_string(count);
            my_store.push_back(s);
            cout << "producer produce one item: " << s << " count:" << resource_count << endl;
            cv_consumer.notify_one();
            usleep(1);
        }   
    }
    
    void ConsumerThread(int id) 
    {
        for (int count = 0; count < 5; count++) {
            unique_lock<mutex> lck(mtx);
            if (0 >= resource_count) {
               cv_consumer.wait(lck);
            }   
            string s = my_store.front();
            my_store.pop_front();
            resource_count--;
            cout << "consumer " << id << " consume one item: " << s << " count:" << resource_count << endl;
            cv_producer.notify_one();
        }   
    }
    
    int main()
    {
        thread producer_thread(ProducerThread);
        thread consumer_thread1(ConsumerThread, 1); 
        thread consumer_thread2(ConsumerThread, 2); 
    
        producer_thread.join();
        consumer_thread1.join();
        consumer_thread2.join();
    
        cout << "main ended " << endl;
        return 0;
    }

输出如下：

producer produce one item: resource:0 count:1
    producer produce one item: resource:1 count:2
    consumer 2 consume one item: resource:0 count:1
    producer produce one item: resource:2 count:2
    producer produce one item: resource:3 count:3
    producer produce one item: resource:4 count:4
    producer produce one item: resource:5 count:5
    producer produce one item: resource:6 count:6
    producer produce one item: resource:7 count:7
    producer produce one item: resource:8 count:8
    producer produce one item: resource:9 count:9
    consumer 2 consume one item: resource:1 count:8
    consumer 2 consume one item: resource:2 count:7
    consumer 2 consume one item: resource:3 count:6
    consumer 2 consume one item: resource:4 count:5
    consumer 1 consume one item: resource:5 count:4
    consumer 1 consume one item: resource:6 count:3
    consumer 1 consume one item: resource:7 count:2
    consumer 1 consume one item: resource:8 count:1
    consumer 1 consume one item: resource:9 count:0
    main ended