多线程学习系列四 生产者和消费者 互斥所和条件变量实现

// producer_consumer.cpp

//////////////////////////////////////////////////////////////////////

// 有一个生产者在生产产品，这些产品将提供给若干个消费者去消费，为了使生产者和消费者能并发执行，

// 在两者之间设置一个有多个缓冲区的缓冲池，生产者将它生产的产品放入一个缓冲区中，消费者可以从缓

// 冲区中取走产品进行消费，所有生产者和消费者都是异步方式运行的，但它们必须保持同步，即不允许消

// 费者到一个空的缓冲区中取产品，也不允许生产者向一个已经装满产品且尚未被取走的缓冲区中投放产品。

//////////////////////////////////////////////////////////////////////

#include <pthread.h>

#include <stdlib.h>

#include <unistd.h>

#include <iostream>

using namespace std;

const int BUFFER_LENGTH = 3;

int buffer[BUFFER_LENGTH];

int front = 0, rear = -1; // 缓冲区的前端和尾端

int size = 0;

pthread_mutex_t mymutex = PTHREAD_MUTEX_INITIALIZER;

pthread_cond_t empty_cond = PTHREAD_COND_INITIALIZER;

pthread_cond_t full_cond = PTHREAD_COND_INITIALIZER;

bool producer_wait = false;

bool consumer_wait = true;

void *producer(void *arg);

void *consumer(void *arg);

int main(int argc, char **argv)

{

pthread_t producer_id[2];

pthread_t consumer_id;

pthread_create(&producer_id[0], NULL, producer, NULL);

pthread_create(&producer_id[1], NULL, producer, NULL);

pthread_create(&consumer_id, NULL, consumer, NULL);

pthread_join(producer_id[0],NULL);

pthread_join(producer_id[1],NULL);

pthread_join(consumer_id,NULL);

sleep(1);

return 0;

}

void *producer(void *arg)

{

int i=0;

while(i++<12)

{

pthread_mutex_lock(&mymutex);

if (size == BUFFER_LENGTH) // 如果缓冲区已满，等待; 否则，添加新产品

{

printf("buffer is full. producer is waiting...\n");

producer_wait = true;

pthread_cond_wait(&full_cond, &mymutex);

producer_wait = false;

}

// 往尾端添加一个产品

rear = (rear + 1) % BUFFER_LENGTH;

buffer[rear] = rand() % BUFFER_LENGTH;

printf("producer[%ld] produces the item %d: %d\n",(long)pthread_self(), rear, buffer[rear]);

++size;

if(consumer_wait)

pthread_cond_signal(&empty_cond);

pthread_mutex_unlock(&mymutex);

}

sleep(2);

return (void *)0;

}

void *consumer(void *arg)

{

int i=0;

while(i++<6)

{

pthread_mutex_lock(&mymutex);

if (size == 0) // 如果缓冲区已空，等待; 否则，消费产品

{

printf("buffer is empty. consumer is waiting...\n");

consumer_wait = true;

pthread_cond_wait(&empty_cond, &mymutex);

consumer_wait = false;

}

// 从前端消费一个产品

printf("consumer[%ld] consumes an item %d: %d\n",(long)pthread_self(), front, buffer[front]);

front = (front + 1) % BUFFER_LENGTH;

--size;

if(producer_wait)

pthread_cond_signal(&full_cond);

pthread_mutex_unlock(&mymutex);

}

return (void *)0;

}

运行结果：