一种基于copy-on-write的多生产者消费者模型

sculida

于 2018-03-14 12:11:16 发布

阅读量202

点赞数

分类专栏：其它文章标签： copy-on-write 生产者消费者多线程

本文链接：https://blog.youkuaiyun.com/u010607621/article/details/79552799

版权

其它专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一个消费即创建的设计模式，该模式使用智能指针、互斥锁和队列来实现线程安全的数据生产和消费。通过将生产和消费分离，并在消费前进行数据转移，有效地降低了锁的粒度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#ifndef _CREATEONCOSUME
#define _CREATEONCOSUME
#include <vector>
#include <memory>
#include <mutex>
#include <queue>
using namespace std;

template<typename T>
class CreateOnConsume {
public:
	CreateOnConsume()
		: m_forProduce(make_shared<vector<T> >()) {};
	virtual void consumeEachVec(vector<T> &vec) = 0;
	void push(T &t) {
		lock_guard<mutex> lock(m_mtx);
		m_forProduce->push_back(t);
	};
	void consumeAll() {
		move2ForConsume();
		while (!m_forConsume.empty()) {
			auto &head = m_forConsume.front();
			consumeEachVec(*head);
			m_forConsume.pop();
		}
	};
	
protected:
	shared_ptr<vector<T> > m_forProduce;
	mutex m_mtx;
	queue<shared_ptr<vector<T> > > m_forConsume;//?
	void move2ForConsume() {
		lock_guard<mutex> lock(m_mtx);
		if (!m_forProduce->empty()) {
			m_forConsume.push(m_forProduce);
			m_forProduce = make_shared<vector<T> >();
		}
	};
};
#endif

子类只需要重载consumeEachVec即可。

生产过程的锁粒度很低，只是往std::vector push。

消费过程的锁只在move2ForCosume中作用。第一步判断生产用的vector是否为空，第二步搬运这个vector指针，第三步给生产者再创建一个vector。