多线程同步问题C++实现:生产者-消费者、读者-写者

生产者-消费者问题

  • 全局互斥锁:mutex

  • 条件变量:cv_produce, cv_consume

  • 共享队列:queue

  • 共享队列的最大容量:MAX_SIZE

  • 注意细节:

    • 使用uniqu_lock配合条件变量
    • cv_produce, cv_consume一开始处于wait等待状态,等待退出条件为共享队列是否为空或已满(排除虚假唤醒的情况)
    • consumer在消费完成之后使用cv_produce通知producer
    • producer在生成完成之后使用cv_consume通知consumer
  • 基础实现

    #include <iostream>
    #include <queue>
    #include <thread>
    #include <mutex>
    #include <condition_variable>
    
    std::mutex mutex;
    std::condition_variable cv_produce, cv_consume;
    std::queue<int> buffer;
    const int MAX_SIZE = 20;
    
    void consumer()
    {
        while (true)
        {
            std::this_thread::sleep_for(std::chrono::milliseconds(3000));
            std::unique_lock<std::mutex> lock(mutex);
            // 防止虚假唤醒,也可以使用while判断
            cv_consume.wait(lock, []()
                         { return !buffer.empty(); });
            std::cout << "consumer " << std::this_thread::get_id() << " consume:";
            int n = buffer.front();
            buffer.pop();
            std::cout << n << ", remain size:" << buffer.size() << std::endl;
            cv_produce.notify_all(); // 此处为生产者通知
        }
    }
    
    void producer(int id)
    {
        while (true)
        {
            std::this_thread::sleep_for(std::chrono::milliseconds(1000));
            std::unique_lock<std::mutex> lock(mutex);
            // 防止虚假唤醒,也可以使用while判断
            cv_produce.wait(lock, []()
                         { return buffer.size() < MAX_SIZE; });
            std::cout << "producer " << std::this_thread::get_id() << " produce:";
            buffer.push(id);
            std::cout << id << ", remain size:" << buffer.size() << std::endl;
            cv_consume.notify_all(); // 此处为消费者通知
        }
    }
    
    int main(int argc, char const *argv[])
    {
        std::vector<std::thread> consumer_threads(2), producer_threads(2);
        for (int i = 0; i < 2; ++i)
        {
            consumer_threads[i] = std::thread(consumer);
            producer_threads[i] = std::thread(producer, i + 1);
        }
        for (int i = 0; i < 2; ++i)
        {
            consumer_threads[i].join();
            producer_threads[i].join();
        }
        return 0;
    }
    
  • 使用结构体封装,包括缓存队列、锁、条件变量

    #include <iostream>
    #include <queue>
    #include <thread>
    #include <mutex>
    #include <condition_variable>
    
    const int TOTAL_ITEMS = 20; // 生产总个数
    
    // 缓存结构体, 使用循环队列当做缓存
    struct Buffer_Struct
    {
        std::queue<int> buffer;
        std::mutex mutex;
        std::condition_variable cv_produce;
        std::condition_variable cv_consume;
    } buffer_struct;
    
    using Buffer = struct Buffer_Struct;
    
    void produce_item(Buffer *buff, int item)
    {
        std::unique_lock<std::mutex> lock(buff->mutex);
        buff->cv_produce.wait(lock, [&buff]()
                              { return buff->buffer.size() < TOTAL_ITEMS; });
        buff->buffer.push(item);
        buff->cv_consume.notify_all();
    }
    
    int consume_item(Buffer *buff)
    {
        int res;
        std::unique_lock<std::mutex> lock(buff->mutex);
        buff->cv_consume.wait(lock, [&buff]()
                              { return !buff->buffer.empty(); });
        res = buff->buffer.front();
        buff->buffer.pop();
        buff->cv_produce.notify_all();
        return res;
    }
    
    void producer()
    {
        for (int i = 0; i < TOTAL_ITEMS; ++i)
        {
            std::this_thread::sleep_for(std::chrono::milliseconds(500));
            std::cout << "produce the " << i << "th item..." << std::endl;
            produce_item(&buffer_struct, i);
        }
    }
    
    void consumer()
    {
        static int cnt = 0;
        while (true)
        {
            std::this_thread::sleep_for(std::chrono::milliseconds(1000));
            int item = consume_item(&buffer_struct);
            std::cout << "consume the " << item << "th item" << std::endl;
            if (++cnt == TOTAL_ITEMS)
                break;
        }
    }
    
    int main(int argc, char const *argv[])
    {
        std::thread consume_thread(consumer);
        std::thread produce_thread(producer);
        consume_thread.join();
        produce_thread.join();
        return 0;
    }
    

读者-写者问题

  • 写者是比较简单的,它和任何进程互斥,用互斥信号量的P操作、V操作即可解决。

  • 读者必须实现与写者互斥的同时还要实现与其他读者的同步

    • 利用读者计数器实现
  • 读者优先版本

    #include <iostream>
    #include <string>
    #include <vector>
    #include <thread>
    #include <mutex>
    
    class ReadWriteLock
    {
    private:
        std::mutex readMtx;
        std::mutex writeMtx;
        int readCnt; // 读者的个数
    
    public:
        ReadWriteLock() : readCnt(0) {}
        void readLock()
        {
            readMtx.lock();     // 互斥访问readCnt变量
            if (++readCnt == 1) // 读者数量+1
            {
                writeMtx.lock(); //当第一个读进程读共享文件时, 阻止写进程写
            }
            readMtx.unlock(); //释放互斥变量readCnt
        }
    
        void readUnlock()
        {
            readMtx.lock();     // 互斥访问readCnt变量
            if (--readCnt == 0) // 读者数量-1
            {
                writeMtx.unlock(); //当最后一个读进程读完共享文件,允许写进程写
            }
            readMtx.unlock(); //释放互斥变量 readCnt
        }
    
        void writeLock()
        {
            writeMtx.lock();
        }
    
        void writeUnlock()
        {
            writeMtx.unlock();
        }
    };
    
    static std::string content;
    ReadWriteLock rwlock;
    
    void write(char c)
    {
        rwlock.writeLock();
        content += c;
        std::cout << "write content: " << content << std::endl;
        rwlock.writeUnlock();
    }
    
    void read()
    {
        rwlock.readLock();
        std::cout << "read content: " << content << std::endl;
        rwlock.readUnlock();
    }
    
    int main(int argc, char const *argv[])
    {
        std::vector<std::thread> writer_threads;
        std::vector<std::thread> reader_threads;
    
        for (int i = 0; i < 10; ++i)
            writer_threads.push_back(std::thread(write, 'A' + i));
    
        for (int i = 0; i < 100; ++i)
            reader_threads.push_back(std::thread(read));
    
        for (int i = 0; i < 10; ++i)
            writer_threads[i].join();
    
        for (int i = 0; i < 100; ++i)
            reader_threads[i].join();
        return 0;
    }
    
  • 写者优先(或公平?)

    #include <iostream>
    #include <string>
    #include <vector>
    #include <thread>
    #include <mutex>
    
    class ReadWriteLock
    {
    private:
        std::mutex readMtx;
        std::mutex writeMtx;
        std::mutex mutex;
        int readCnt; // 已加读锁的个数
    
    public:
        ReadWriteLock() : readCnt(0) {}
    
        void readLock()
        {
            mutex.lock();       // 在无写进程请求时进入
            readMtx.lock();     // 互斥访问readCnt变量
            if (++readCnt == 1) // 读者数量+1
            {
                writeMtx.lock(); //当第一个读进程读共享文件时, 阻止写进程写
            }
            readMtx.unlock(); //释放互斥变量readCnt
            mutex.unlock();   //恢复对共享文件的访问
        }
    
        void readUnlock()
        {
            readMtx.lock();     // 互斥访问readCnt变量
            if (--readCnt == 0) // 读者数量-1
            {
                writeMtx.unlock(); //当最后一个读进程读完共享文件,允许写进程写
            }
            readMtx.unlock(); //释放互斥变量 readCnt
        }
    
        void writeLock()
        {
            mutex.lock(); //在无写进程请求时进入
            writeMtx.lock();
        }
    
        void writeUnlock()
        {
            writeMtx.unlock();
            mutex.unlock(); //恢复对共享支件的访问
        }
    };
    
    static std::string content;
    ReadWriteLock rwlock;
    
    void write(char c)
    {
        rwlock.writeLock();
        content += c;
        std::cout << "write content: " << content << std::endl;
        rwlock.writeUnlock();
    }
    
    void read()
    {
        rwlock.readLock();
        std::cout << "read content: " << content << std::endl;
        rwlock.readUnlock();
    }
    
    int main(int argc, char const *argv[])
    {
        std::vector<std::thread> writer_threads;
        std::vector<std::thread> reader_threads;
    
        for (int i = 0; i < 10; ++i)
            writer_threads.push_back(std::thread(write, 'A' + i));
    
        for (int i = 0; i < 100; ++i)
            reader_threads.push_back(std::thread(read));
    
        for (int i = 0; i < 10; ++i)
            writer_threads[i].join();
    
        for (int i = 0; i < 100; ++i)
            reader_threads[i].join();
        return 0;
    }
    
  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值