【Linux多线程】条件变量之生产者-消费者模型

线程同步典型的案例即为生产者消费者模型，而借助条件变量来实现这一模型，是比较常见的一种方法。
假定有两个线程，一个模拟生产者行为，一个模拟消费者行为。两个线程同时操作一个共享资源,生产者向其中添加产品，消费者从中消费掉产品（生产者向队列中插入数据，消费者则在生产者发出队列准备好（有数据了）后接收消息，然后取出数据进行处理）。
实现的关键点在以下几个方面：
    生产者和消费者都对条件变量的使用加了锁；
    消费者调用pthread_cond_wait,等待队列是否准备好的信息，注意参数有两个，一个是pthread_cond_t，另外一个是pthread_mutex_t

代码：

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <pthread.h>
 
#define PRO_NUM 2   //生产线程数量
#define CON_NUM 2   //消费线程数量
pthread_cond_t cond;
pthread_cond_t n_empty;
pthread_mutex_t mutex;
 
int g_num = 0;      //产品数量
int empty_num = 3;  //生产的空位数量
 
//productor
void* pro_route(void* arg)
{
    int id =*(int*)arg;
    free(arg);
 
    while(1)
    {
        pthread_mutex_lock(&mutex);
        while(empty_num <= 0)
        {
            printf("生产线程%d等待。\n", id);
            pthread_cond_wait(&n_empty, &mutex);
            printf("有空位,数量为%d\n", empty_num);
        }
        printf("生产线程%d生产\n", id);
        ++g_num;
        --empty_num;
        printf("生产品%d完成\n", g_num);
        sleep(rand()%3);
        pthread_cond_signal(&cond);
        pthread_mutex_unlock(&mutex);
        sleep(rand()%3);
    }
}
//consumer
void *con_route(void* arg)
{
    int id =*(int*)arg;
    free(arg);
 
    while(1)
    {
        pthread_mutex_lock(&mutex);
        while(g_num <= 0)
        {
            printf("消费线程%d等待。。\n", id);
            pthread_cond_wait(&cond, &mutex);
            printf("第%d产品到了！！\n", g_num);
        }
        printf("消费线程%d消费 产品%d\n", id , g_num);
        --g_num;
        ++empty_num;
        sleep(rand()%2);
        printf("消费线程%d消费完成\n", id);
        pthread_cond_signal(&n_empty);
        pthread_mutex_unlock(&mutex);
        sleep(rand()%3);
    }
}
int main( )
{
    int i;
    srand(getpid());
 
    pthread_t tids[PRO_NUM + CON_NUM];
    //互斥量，条件变量初始化
    pthread_mutex_init(&mutex, NULL);
    pthread_cond_init(&cond, NULL);   //条件变量1
    pthread_cond_init(&n_empty, NULL);//条件变量2
 
    //创建生产者线程
    for(i =0; i< PRO_NUM; ++i)
    {
        int* p = (int*)malloc(sizeof(int)); //传入参数相当作线程编号
        *p = i;
        pthread_create(&tids[i], NULL, pro_route, p);
    }
    //创建消费者线程
    for(i =0; i< CON_NUM; ++i)
    {
        int* p = (int*)malloc(sizeof(int)); //消费线程编号
        *p = i;
        pthread_create(&tids[i], NULL, con_route, p);
    }
    for(i =0; i< PRO_NUM + CON_NUM; ++i) //回收线程
    {
        pthread_join(tids[i], NULL);
    }
    //互斥量，条件变量销毁
    pthread_mutex_destroy(&mutex);
    pthread_cond_destroy(&cond);
    pthread_cond_destroy(&n_empty);
 
    return 0;
}

gcc执行结果