无锁队列之单生产者单消费者

现实应用场景中经常会用到单生产者单消费者或多生产者多消费者模型,例如一个写线程,接收到数据写入缓冲区,另一个线程读出并处理。为了尽可能减少锁所占用的时间,可使用gcc的一些原子操作来代替pthread_mutex_t或pthread_spinlock_t。

核心思想:预分配一段连续的内存,用min_seq_表示读的位置,max_seq_表示写的位置,当写满或者读完时返回错误;如果不同步有可能发生的错误是:读线程读取max_seq_时,刚好写线程在修改max_seq_,min_seq_同上,故使用__sync_add_and_fetch 函数自增0或者1来读取或者修改变量。

注意事项:分配的内存大小一定要足够大,否则可能会造成数据的丢失。
就好比一件仓库,有人不停往里存放东西,有人不停往外取,当仓库较小时,空间满时,可能会出现存放失败,那么这件商品就丢失了。

实现如下:

#ifndef SPSC_H
#define SPSC_H
//single producer single consumer

#include <stdint.h>
#include <stdlib.h>
#include <string.h>

//if Data is fixed length, you can use Data directly; else you must use Data*
template<typename Data>
class Spsc
{
public:
  Spsc():buf_(NULL), size_(0), max_seq_(0), min_seq_(0){}
  Spsc(int size):buf_(NULL), size_(size), max_seq_(0), min_seq_(0){}

  ~Spsc()
  {
    if(NULL != buf_)
    {
      free(buf_);
      buf_ = NULL;
    }
  }

  bool Init();
  bool Init(int size);
  bool Push(Data& data);

  //note:if data is pointer type, don't forget delete it
  bool Pop(Data& data);

private:
  char* buf_;
  int size_;
  uint64_t max_seq_;//push seq
  uint64_t min_seq_;//pop seq
};

template<typename Data>
bool Spsc<Data>::Init()
{
  if
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值