C++模板实现的队列多线程安全

直接给代码,没什么好说的。

#pragma once
#include <iostream>
#include <queue> 
#include <mutex>
#include <memory>
#include <condition_variable>
template<typename T>
class Threadsafe_queue
{
private :
    mutable std::mutex mut;
    std::queue<std::shared_ptr<T>> data_queue;
    std::condition_variable data_cond;
public:
    Threadsafe_queue(){}
    void wait_end_pop(T& value)
    {
        std::unique_lock<std::mutex> lk(mut);
        //等待队列不为空
        data_cond.wait(lk,[this]{return !data_queue.empty();});
        value=std::move(*data_queue.front());
        data_queue.pop();
    }
    bool try_pop(T& value)
    {
        std::lock_guard<std::mutex> lk(mut);
        if(data_queue.empty())
            return false;
        value=std::move(*data_queue.front());
        data_queue.pop();
        return true;
    }

    std::shared_ptr<T> wait_end_pop()
    {
        std::unique_lock<std::mutex> lk(mut);
        data_cond.wait(lk,[this]{return !data_queue.empty();});
        std::shared_ptr<T> res=data_queue.front();
        data_queue.pop();
        return res;
    }

    std::shared_ptr<T> try_pop()
    {
        std::lock_guard<std::mutex> lk(mut);
        if(data_queue.empty())
            return std::shared_ptr<T>();
        std::shared_ptr<T> res=data_queue.front();
        data_queue.pop();
        return res;
    }

    void push(T new_value)
    {
        std::shared_ptr<T> data(
            std::make_shared<T>(std::move(new_value)));
        std::lock_guard<std::mutex> lk(mut);
        data_queue.push(data);
        data_cond.notify_one();
    }

    bool empty() const
    {
        std::lock_guard<std::mutex> lk(mut);
        return data_queue.empty();
    }
};

使用方法

#include <thread>
Threadsafe_queue<int> mQueue=Threadsafe_queue<int>();
int main(int argc,char*argv[])
{
    //lambada表达式
    std::thread mThread([]
        {
            while(1)
            {
                int a;
                mQueue.wait_end_pop(a);
                std::cout<<a<<std::endl;
                if(a==1) break;
            }
        });
    mThread.detach();
    int j=1000;
    while(j--)
    {
        mQueue.push(j);
    }
}

提示:纯手打可能 大概也许 应该没手误。。。。。

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
当设计一个线程安全模板队列时,你需要考虑以下几个方面: 1. 数据访问的互斥保护:使用互斥量(`std::mutex`)来保护对队列的读写操作,确保同一时间只有一个线程可以访问队列。 2. 条件变量的使用:使用条件变量(`std::condition_variable`)来实现线程的等待和唤醒机制,以避免忙等待和降低 CPU 占用。条件变量可以用于等待队列非空的条件和等待队列为空的条件。 3. 内存管理:确保在队列中存储的元素的内存管理是正确的,避免内存泄漏或悬空指针的问题。 下面是一个基本的线程安全队列的示例代码: ```cpp #include <queue> #include <mutex> #include <condition_variable> template<typename T> class ThreadSafeQueue { public: ThreadSafeQueue() = default; void Push(T value) { std::lock_guard<std::mutex> lock(m_mutex); m_queue.push(std::move(value)); m_cv.notify_one(); // 唤醒一个等待的线程 } T Pop() { std::unique_lock<std::mutex> lock(m_mutex); m_cv.wait(lock, [this] { return !m_queue.empty(); }); T value = std::move(m_queue.front()); m_queue.pop(); return value; } bool Empty() const { std::lock_guard<std::mutex> lock(m_mutex); return m_queue.empty(); } private: std::queue<T> m_queue; mutable std::mutex m_mutex; std::condition_variable m_cv; }; ``` 在上述代码中,`ThreadSafeQueue` 类使用了一个 `std::queue` 来存储元素,并使用 `std::mutex` 来保护对队列的读写操作。`std::condition_variable` 用于实现线程等待和唤醒。 `Push()` 函数用于向队列中添加元素,会先获取互斥锁,然后将元素推入队列,并通过条件变量的 `notify_one()` 唤醒一个等待的线程。 `Pop()` 函数用于从队列中取出元素,如果队列为空,则会等待条件变量的通知,直到队列非空。一旦满足条件,就会取出队列的前端元素,并将其从队列中移除。 `Empty()` 函数用于检查队列是否为空,同样需要获取互斥锁来保护对队列的访问。 通过使用互斥锁和条件变量,这个模板队列可以在多线程环境下安全地进行读写操作。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值