c++ 阻塞队列的实现

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

namespace WPAYRTL
{
template<class T>
class BlockingQueue
{
public:
	BlockingQueue() {}
	~BlockingQueue() {}

	void offer(std::shared_ptr<T> &e)
	{
		{
			std::unique_lock<std::mutex> lockGard(_myMutex);
			_queue.push(e);
		}
		_myVariable.notify_all();
	}

	std::shared_ptr<T> take()
	{
		std::unique_lock<std::mutex> lockGard(_myMutex);
		_myVariable.wait(lockGard, [&] {return _queue.size() > 0;});
		std::shared_ptr<T> e = _queue.front();
		_queue.pop();
		return e;
	}

	std::shared_ptr<T>take(std::chrono::milliseconds& time)
	{
		std::unique_lock<std::mutex> lockGard(_myMutex);
		bool result = _myVariable.wait_for(lockGard, time, [&] {return _queue.size() > 0;});
		if (!result)
		{
			return nullptr;
		}
		std::shared_ptr<T> e = _queue.front();
		_queue.pop();
		return e;
	}

private:
	std::queue<std::shared_ptr<T>> _queue;
	std::mutex _myMutex;
	std::condition_variable _myVariable;
};

}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值