多线程安全的queue

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

template<typename T>
class threadsave_queue {
private:
	mutable std::mutex mut;
	std::queue<T> data_queue;
	std::condition_variable data_cond;
public:
	threadsave_queue() {}
	threadsave_queue(threadsave_queue const& other) {
		std::lock_guard<std::mutex> lk(other.mut);
		data_queue = other.data_queue();
	}
	void push(T new_value) {
		std::lock_guard<std::mutex> lk(mut);
		data_queue.push(new_value);
		data_cond.notify_one();
	}
	void wait_and_pop(T& value) {
		std::unique_lock<std::mutex> lk(mut);
		//--如果data_queue不为空,那么执行后续代码,如果data_queue为空,那么继续等待
		data_cond.wait(lk, [this] {return !data_queue.empty(); });
		value = data_queue.front();
		data_queue.pop();
	}

	std::shared_ptr<T> wait_and_pop() {
		std::unique_lock<std::mutex> lk(mut);
		data_cond.wait(lk, [this] {return !data_queue.empty(); });
		std::shared_ptr<T> res(std::make_shared<T>(data_queue.front()));
		data_queue.pop();
		return res;
	}

	bool empty()const {
		std::lock_guard<std::mutex> lk(mut);
		return data_queue.empty();
	}
};

void make_data(threadsave_queue<int>& tq, int val) {
	tq.push(val);
}
void get_data1(threadsave_queue<int>& tq, int& d1) {
	tq.wait_and_pop(d1);
}
void get_data2(threadsave_queue<int>& tq, int& d1) {
	auto at = tq.wait_and_pop();
	d1 = *at;
}
int main() {
	threadsave_queue<int> q1;
	int d1;
	std::thread t1(make_data, std::ref(q1), 10);
	std::thread t2(get_data1, std::ref(q1), std::ref(d1));
	t1.join();
	t2.join();
	std::cout << d1 << std::endl;
	std::thread t3(make_data, std::ref(q1), 20);
	std::thread t4(get_data2, std::ref(q1), std::ref(d1));
	t3.join();
	t4.join();
	std::cout << d1 << std::endl;
	q1.empty();

}

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值