c++11 线程安全的队列实现

最新推荐文章于 2024-10-12 21:07:13 发布

mymodian

最新推荐文章于 2024-10-12 21:07:13 发布

阅读量1.5w

点赞数 16

分类专栏： c++11 文章标签： c11 队列 threadsafe

本文链接：https://blog.csdn.net/mymodian9612/article/details/53608084

版权

本文介绍了如何使用C++11的mutex和condition_variable结合自定义链表实现线程安全的队列，以提高并发性能。通过优化加锁策略，使得在保证线程安全的同时，减少了锁的粒度，从而提升了队列操作的效率。具体实现包括初始化时创建一个空节点，push_back操作只需对tail加锁，pop_front操作则对head和tail加锁但tail加锁时间短。

摘要由CSDN通过智能技术生成

用c++11的mutex和condition_variable配合STL的queue可以很方便地实现线程安全的队列，queue用来存取元素，存取元素时用mutex进行全局加锁，存取完成再进行解锁。

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

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

	void push(T new_data)
	{
		std::lock_guard<std::mutex> lk(mut);            // 1.全局加锁
		data_queue.push(std::move(new_data));           // 2.push时独占锁
		cond.notify_one();
	}
	void wait_and_pop(T& val)
	{
		std::unique_lock<std::mutex> ulk(mut);                    // 3.全局加锁
		cond.wait(ulk,[this]() { return !data_queue.empty(); });  // 4.front 和 pop_front时独占锁
		val=std::move(data_queue.front());
		data_queue.pop();
	}
	std::shared_ptr<T> wait_and_pop()
	{
		std::unique_lock<std::mutex> ulk(mut);
		cond.wait(ulk,[this]() { return !data_queue.empty(); });
		std::shared_ptr<T> val(std::make_shared<T>(std::move(data_queue.front())));
		data_queue.pop();
		return val;
	}
	bool try_pop(T& val)
	{
		std::lock_guard<std::mutex> lk(mut);
		if(data_queue.empty())
			return false;
		val=std::move(data_queue.front());
		data_queue.pop();
		return true;
	}
	std::shared_ptr<T> try_pop()
	{
		std::shared_ptr<T> val;
		std::lock_guard<std::mutex> lk(mut);
		if(data_queue.empty())
			return val;
		val=std::make_shared<T>(std::move(data_queue.front()));
		data_queue.pop();
		return val;
	}
	bool empty()
	{
		std::lock_guard<std::mutex> lk(mut);
		return data_queue.empty();
	}
private:
	std::queue<T> data_queue;
	std::mutex mut;
	std::condition_variable cond;
};

但是