C++11:基于std::queue和std::mutex构建一个线程安全的队列

C++中的模板std::queue提供了一个队列容器,但这个容器并不是线程安全的,如果在多线程环境下使用队列,它是不能直接拿来用的。
基于它做一个线程安全的队列也并不复杂。基本的原理就是用std::mutex信号量对std::queue进行访问控制,以保证任何一个线程都是独占式访问,下面是完整的代码。

/*
 * threadsafe_queue.h
 *
 *  Created on: 2016年7月26日
 *      Author: guyadong
 */

#ifndef COMMON_SOURCE_CPP_THREADSAFE_QUEUE_H_
#define COMMON_SOURCE_CPP_THREADSAFE_QUEUE_H_
#include <queue>
#include <mutex>
#include <condition_variable>
#include <initializer_list>
namespace gdface {
inline namespace mt{
/*
 * 线程安全队列
 * T为队列元素类型
 * 因为有std::mutex和std::condition_variable类成员,所以此类不支持复制构造函数也不支持赋值操作符(=)
 * */
template<typename T>
class threadsafe_queue{
private:
	// data_queue访问信号量
	mutable std::mutex mut;
	mutable std::condition_variable data_cond;
	using queue_type = std::queue<T>;
	queue_type data_queue;
public:
	using value_type= typename queue_type::value_type;
	using container_type = typename queue_type::container_type;
	threadsafe_queue()=default;
	threadsafe_queue(const threadsafe_queue&)=delete;
	threadsafe_queue& operator=(const threadsafe_queue&)=delete;
	/*
	 * 使用迭代器为参数的构造函数,适用所有容器对象
	 * */
	template<typename _InputIterator>
	threadsafe_queue(_InputIterator first, _InputIterator last){
		for(auto itor=first;itor!=last;++itor){
			data_queue.push(*itor);
		}
	}
	explicit threadsafe_queue(const container_type &c):data_queue(c){}
	/*
	 * 使用初始化列表为参数的构造函数
	 * */
	threadsafe_queue(std::initializer_list<value_type> list):threadsafe_queue(list.begin(),list.end()){
	}
	/*
	 * 将元素加入队列
	 * */
	void push(const value_type &new_value){
		std::lock_guard<std::mutex>lk(mut);
		data_queue.push(std::move(new_value));
		data_cond.notify_one();
	}
	/*
	 * 从队列中弹出一个元素,如果队列为空就阻塞
	 * */
	value_type wait_and_pop(){
		std::unique_lock<std::mutex>lk(mut);
		data_cond.wait(lk,[this]{return !this->data_queue.empty();});
		auto value=std::move(data_queue.front());
		data_queue.pop();
		return value;
	}
	/*
	 * 从队列中弹出一个元素,如果队列为空返回false
	 * */
	bool try_pop(value_type& 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;
	}
	/*
	 * 返回队列是否为空
	 * */
	auto empty() const->decltype(data_queue.empty()) {
		std::lock_guard<std::mutex>lk(mut);
		return data_queue.empty();
	}
	/*
	 * 返回队列中元素数个
	 * */
    auto size() const->decltype(data_queue.size()){
    	std::lock_guard<std::mutex>lk(mut);
    	return data_queue.size();
    }
}; /* threadsafe_queue */
}/* namespace mt */
}/* namespace gdface */
#endif /* COMMON_SOURCE_CPP_THREADSAFE_QUEUE_H_ */

这里只实现了阻塞式的pop函数wait_and_pop,你也可以根据自己的需要对代码进行适当的改造,以符合自己的需求。
(C++11风格代码,在VS2015和gcc5.2.0下编译通过)

  • 8
    点赞
  • 34
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

10km

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值