生产者与消费者

#代码:

#include<stdio.h>
#include<pthread.h>
#define N 20
#define OVER -1
struct strbuff//定义缓冲区结构体
{
int buffer[N];//缓冲数组,每个数组仅存一个整数,装N个数组,装满就取,取出就装...
pthread_mutex_t lock;//互斥体lock,用于对数组互斥操作
int rpos,wpos;//读指针,写指针
pthread_cond_t nempty;//数组未使用条件变量
pthread_cond_t nfull;//数组未使用完条件变量
};
void init(struct strbuff *b)//初始化缓冲区结构体
{
pthread_mutex_init(&b->lock,NULL);//初始化互斥体
pthread_cond_init(&b->nempty,NULL);//初始化非空条件变量
pthread_cond_init(&b->nfull,NULL);//初始化未满条件变量
b->rpos=0;//初始化读指针
b->wpos=0;//初始化写指针
}
void put(struct strbuff *b,int data)//存入一个整数,向缓冲区
{
pthread_mutex_lock(&b->lock);//加锁
if((b->wpos+1)%N==b->rpos)//等待缓冲区未满
pthread_cond_wait(&b->nfull,&b->lock);
b->buffer[b->wpos]=data;//写数据
b->wpos++;//移动位置
//printf("wpos:%d\n",b->wpos);
if(b->wpos>=N)//写满,置零
b->wpos=0;
/*设置缓冲区非空条件变量*/
pthread_cond_signal(&b->nempty);
pthread_mutex_unlock(&b->lock);//解锁
}
int get(struct strbuff *b)//从缓冲区取整数
{
int data;
pthread_mutex_lock(&b->lock);
if(b->wpos==b->rpos)//等待缓冲区装满
pthread_cond_wait(&b->nempty,&b->lock);
data=b->buffer[b->rpos];//读数据
b->rpos++;//移动位置
if(b->rpos>=N)//读完
b->rpos=0;//置零
pthread_cond_signal(&b->nfull);//设置缓冲区未满条件变量
pthread_mutex_unlock(&b->lock);
return data;
}

//测试线程1写入0,1,..,19的整数

struct strbuff *buff;
void *producer(void *data)
{
int n;
for(n=0;n<20;n++)
    {
printf("%d--->\n",n);
put(&buff,n);
    }
put(buff,OVER);
return NULL;
}
void *consumer(void *data)
{
int d;
while(1)
    {
d=get(&buff);
if(d==OVER)
break;
printf("--->%d\n",d);
    }
pthread_exit(NULL);
}
int main()
{
pthread_t th_a,th_b;
init(&buff);
pthread_create(&th_a,NULL,producer,0);
sleep(3);
pthread_create(&th_b,NULL,consumer,0);
pthread_join(th_a,NULL);
pthread_join(th_b,NULL);
return 0;
}