C++11线程安全队列


多线程编程需要实现一个线程安全的队列,上锁,避免多个线程同时读写

代码:

/**
 * 线程安全的队列
 */

#ifndef __THREAD_SAFE_QUEUE__
#define __THREAD_SAFE_QUEUE__
#include 
#include 
#include 
#include 

template 
class thread_safe_queue
{
private:
	mutable mutex 		mut;		//锁
	queue 			data_queue;	//队列
	condition_variable 	data_cond;	//条件变量
public:
	//构造函数
	thread_safe_queue();
	thread_safe_queue &operator=(const thread_safe_queue&)=delete;

	//可以多传递一个额外的条件
	bool wait_and_pop(T &value,atomic_bool &bl);
	bool try_pop(T &value);
	void push(T new_value);
	bool empty() const;

	void notify_all()
	{
		data_cond.notify_all();
	}
};
template 
thread_safe_queue::thread_safe_queue(){}

template 
bool thread_safe_queue::empty() const
{
	lock_guard lock(mut);
	return data_queue.empty();
}

template 
void thread_safe_queue::push(T new_value)
{
	lock_guard lock(mut);
	data_queue.push(new_value);
	data_cond.notify_one();
}


template 
bool thread_safe_queue::wait_and_pop(T &value,atomic_bool &bl)
{
	unique_lock lock(mut);
	
	//避免队列为空时一直等待
	data_cond.wait(lock,[this,&bl](){return (!this->data_queue.empty())||bl; });

	if(bl&&data_queue.empty())
		return false;

	value=data_queue.front();
	data_queue.pop();
	return true;
}
template
bool thread_safe_queue::try_pop(T &value)
{
	lock_guard lock(mut);
	if(empty())
		return false;

	value=data_queue.front();
	data_queue.pop();
	return true;
}

#endif
对于mutex我的理解是在同一个mutex的lock、unlock之间的代码不能同时运行。
condition_variable:线程之间同步的一种方式,通过notify_one --> wait、ontify_all-->wait配合使用 

你可能感兴趣的:(C++)