linux下c++實現簡單的生產者消費者隊列模式

引言

生產者消費者是一個經典的模式

利用生產者，消費者和緩衝區降低了生產者和消費者之間的的耦合度

便於對生產者和消費者的修改

下面記錄的是一個經典的單一生產者多消費者的模式

設計思路

以隊列做為緩衝區，實現產品的FIFO

生產者調用緩衝區的push函數，將產品加入緩衝區

消費者調用緩衝區的pop函數，將產品從緩衝區取出

因為生產者與消費者分屬於不同的線程，所以要設置鎖

類的聲明

class CacheQueue
{
private:
/**
* @brief 緩衝隊列
*/
queue<int>* _requests; /**
* @brief 互斥鎖
**/
pthread_mutex_t _mutex; /**
* @brief Queue not full conditional object
**/
pthread_cond_t _not_full_cond; /**
* @brief Queue not empty conditional object
**/
pthread_cond_t _not_empty_cond; uint32_t _bufSize; public: ChacheQueue(); void SetMaxLength(uint32_t bufSize);
/**
* @brief 向隊列添加產品
* @param [in] req: 待添加的產品
**/
void Push(int req); /**
* @brief 從隊列中取出一個產品
* @param [return] : 從隊列中取出的產品
**/
int Pop(uint32_t timeout); /**
* @brief 析構函數
**/
~CacheQueue();
};

重要的函數是Push和Pop，生產者調用Push向緩衝區添加產品，消費者則調用Pop函數獲取產品

線程條件_not_full_cond表示隊列不滿，可以添加產品

線程條件_not_empty_cond表示隊列不空，可以獲取產品

Push函數

void CacheQueue::Push(int req)
{
/**
* 上鎖
*/
pthread_mutex_lock(&_mutex);/**
* 如果隊列滿，等待信號
*/
while (_requests->size() == _bufSize)
{
pthread_cond_wait(&_not_full_cond, &_mutex);
}
_requests->push(req);/**
* 發送非空信號
*/
pthread_cond_signal(&_not_empty_cond);/**
* 解鎖
*/
pthread_mutex_unlock(&_mutex);
}

Pop函數

int CacheQueue::Pop(uint32_t timeout)
{
int ret = 0;
int req = NO_DATA;
/**
* 上鎖
*/
pthread_mutex_lock(&_mutex);
/**
* 若隊列空等待指定時間
*/
struct timeval now;
struct timespec timepass;
gettimeofday(&now, NULL);
timepass.tv_sec = now.tv_sec + timeout;
timepass.tv_nsec = 0;
while (ret == 0 && _requests->empty())
{
ret = pthread_cond_timedwait(&_not_empty_cond, &_mutex, &timepass);
}
/**
* 沒有數據，返回沒有數據標識
*/
if(ret!=0)
{
pthread_mutex_unlock(&_mutex);
return req;
}
/**
* 返回數據，發送隊列非滿信號
*/
req = _requests->front();
_requests->pop();
pthread_cond_signal(&_not_full_cond);
/**
* 解鎖
*/
pthread_mutex_unlock(&_mutex);
return req;
}