Linux c++,用信号量实现消费者生产者队列(代码可直接通过编译)

//用信号量实现的一个消费者生产者队列,
#include <iostream>
#include <pthread.h>
#include <semaphore.h>
#include <errno.h>
#include <queue>
using namespace std;
const int WAITTIME = 10000;
class Task{};
class MessageQue{
public:
  MessageQue(){}
  ~MessageQue(){}
  int Init(int Max){
    int ret = 0;
    ret = pthread_mutex_init(&m_mutex,NULL);
    if(ret == -1){
      goto Errcode;
    }
    ret = sem_init(&m_isEmpty,0,0);
    if(ret == -1){
      goto Errcode;
    }
    ret = sem_init(&m_isFull,0,Max);
    if(ret == -1){
      goto Errcode;
    }
    return 0;
  Errcode:
    //log
    return -1;
  }
  int Wait2Pop(Task** p){
    while(1){
      int ret = 0;
      //等待队列不为空即被唤醒
      ret = sem_wait(&m_isEmpty);
      if(ret < 0){
        if(errno == EINTR)
          continue;
        else
          goto Errcode;
      }
      else{
        break;
      }
    }
    pop(p);
    return 0;
  Errcode:
    return -1;
  }
  int Wait2Push(Task* p){
    while(1){
      int ret = 0;
      //队列不满即可被唤醒
      ret = sem_wait(&m_isFull);
      if(ret < 0){
        if(errno == EINTR)
          continue;
        else
          goto Errcode;
      }
      else{
        break;
      }
    }
    push(p);
    return 0;
  Errcode:
    return -1;
  }
private:
  int pop(Task** p){
    pthread_mutex_lock(&m_mutex);
    *p = qTask.front();
    qTask.pop();
    //通知队列不满
    sem_post(&m_isFull);
    pthread_mutex_unlock(&m_mutex);
  }
  int push(Task* p){
    pthread_mutex_lock(&m_mutex);
    qTask.push(p);
    //通知队列不空
    sem_post(&m_isEmpty);
    pthread_mutex_unlock(&m_mutex);
  }
  queue<Task*> qTask;
  sem_t m_isEmpty;
  sem_t m_isFull;
  pthread_mutex_t m_mutex;
};

int main()
{
  MessageQue q;
}
 

猜你喜欢

转载自blog.csdn.net/wWX336815/article/details/83585209