C++11 线程安全的队列模板

最新推荐文章于 2024-04-21 21:11:08 发布

原创最新推荐文章于 2024-04-21 21:11:08 发布 · 523 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #开发语言

C++ 专栏收录该内容

18 篇文章

订阅专栏

本文详细介绍了一个C++11标准库中线程安全的队列(SafeQueue)模板，通过模板实例展示如何利用`std::mutex`、`std::condition_variable`实现数据同步与互斥操作，确保在多线程环境下队列操作的正确性。

C++11 线程安全的队列模板

C++11 标准库版本

主要参考C++11 并发编程第二版

#include <queue>
#include <mutex>
#include <memory>
#include <condition_variable>


template<typename T>
class SafeQueue
{
public:
    SafeQueue() {}
    ~SafeQueue() {}

    void push(T new_data) {
        std::lock_guard<std::m_mutex> lk(m_mut); // 1.全局加锁
        m_dataQueue.push(std::move(new_data)); // 2.push时独占锁
        m_cond.notify_one();
    }

    void wait_and_pop(T& val) {
        std::unique_lock<std::m_mutex> ulk(m_mut); // 3.全局加锁
        m_cond.wait(ulk,[this]() { return !m_dataQueue.empty(); }); // 4.front 和 pop_front时独占锁
        val=std::move(m_dataQueue.front());
        m_dataQueue.pop();
    }

    std::shared_ptr<T> wait_and_pop() {
        std::unique_lock<std::m_mutex> ulk(m_mut);
        m_cond.wait(ulk,[this]() { return !m_dataQueue.empty(); });
        std::shared_ptr<T> val(std::make_shared<T>(std::move(m_dataQueue.front())));
        m_dataQueue.pop();
        return val;
    }

    bool try_pop(T& val) {
        std::lock_guard<std::m_mutex> lk(m_mut);
        if(m_dataQueue.empty())
            return false;
        val=std::move(m_dataQueue.front());
        m_dataQueue.pop();
        return true;
    }

    std::shared_ptr<T> try_pop() {
        std::shared_ptr<T> val;
        std::lock_guard<std::m_mutex> lk(m_mut);
        if(m_dataQueue.empty())
            return val;
        val=std::make_shared<T>(std::move(m_dataQueue.front()));
        m_dataQueue.pop();
        return val;
    }

    bool empty() {
        std::lock_guard<std::m_mutex> lk(m_mut);
        return m_dataQueue.empty();
    }

    int count() {
        std::lock_guard<std::m_mutex> lk(m_mut);
        return m_dataQueue.size();
    }

private:
    std::queue<T> m_dataQueue;
    std::mutex m_mut;
    std::condition_variable m_cond;
}