更安全的std::condition_variable

 原生std::condition_variable的notify() 和 wait()有以下特性:

1. 仅支持简单的消息等待和消息发送。

2. 当notify()后于调用wait()的时候,消息会丢失,wait()阻塞。

3. 当wait()接收到一次notify()之后,就再也不会阻塞。

 

由于我们的业务所需。实现以下功能:

1. 当notify()后于调用wait()的时候,消息不会丢失,wait()不阻塞。

2. 增加计数。一次notify()对应一次wait()。

3. 线程安全。

#include "base_constructor.h"

#include <mutex>
#include <condition_variable>

namespace naive {

class SafeConditionVariable {

public:

	explicit SafeConditionVariable():_pv(0){}

	void Notify() {
		std::unique_lock<std::mutex> lck(_mtx);
		_pv++;
		_cv.notify_one();
	}

	void Wait() {
		std::unique_lock<std::mutex> lck(_mtx);
		_cv.wait(lck,[this]{ return (_pv > 0); });
		_pv--;
	}
	
	~SafeConditionVariable() {}

private:

	std::mutex				_mtx;
	int						_pv;
	std::condition_variable _cv;

	DISALLOW_COPY_AND_ASSIGN(SafeConditionVariable);
};
}

git地址:https://github.com/SaberMqd/naive/blob/master/include/naive/safe_wait.h

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值