在生产者写入缓冲区时,首先要获得互斥锁,并且判断缓冲区是否为满。也就是判断写指针+1后是否等于读指针,如果相等则进入等待状态,等候条件变量notfull发出信号,如果判断条件不等。则向缓冲区写一个整数,同时设置条件变量notempty,最后释放互斥锁!
当消费者从缓冲区读数时候,首先要获得互斥锁,并且判断缓冲区是否为空,也就是判断写指针是否等于读指针,如果相等则进入等待状态,等候条件变量notempty发出信号,如果判断结果不等,则从缓冲区读出一个整数,同时设置条件变量notfull,最后释放互斥锁。
#include<stdio.h>
#include<pthread.h>
#define BUFFER_SIZE 8
struct prodcons{
int buffer[BUFFER_SIZE];
pthread_mutex_t lock;
int readpos,writepos;
pthread_cond_t notempty;//缓冲区未空判断
pthread_cond_t notfull;//缓冲区未满判断
};
void init(struct prodcons *b)
{
pthread_mutex_init(&b->lock,NULL);
pthread_cond_init(&b->notempty,NULL);
pthread_cond_init(&b->notfull,NULL);
b->readpos=0;
b->writepos=0;
}
void put(struct prodcons *b,int data)
{
if( (b->writepos + 1 ) % BUFFER_SIZE == b->readpos)
{
pthread_cond_wait(&b->notfull,&b->lock);
printf("wait for not full!\n");
}
b->buffer[b->writepos]=data;
b->writepos++;
if(b->writepos >=BUFFER_SIZE)
b->writepos = 0;
pthread_cond_signal(&b->notempty);
pthread_mutex_unlock(&b->lock);
}
int get(struct prodcons *b)
{
int data;
pthread_mutex_lock(&b->lock);
if(b->writepos == b->readpos)
{
pthread_cond_wait(&b->notempty,&b->lock);
printf("wait for not empty!\n");
}
data = b->buffer[b->readpos];
b->readpos++;
if(b->readpos >= BUFFER_SIZE)
b->readpos = 0;
pthread_cond_signal(&b->notfull);
pthread_mutex_unlock(&b->lock);
return data;
}
#define OVER (-1)
struct prodcons buffer;
void *producer(void *data)
{
int n;
for(n=0;n<1000;n++)
{
printf("put----->%d \n",n);
put(&buffer,n);
}
put(&buffer,OVER);
return NULL;
}
void *consumer(void *data)
{
int d;
while(1)
{
d = get(&buffer);
if(d == OVER)
break;
printf("%d------>get\n",d);
}
return NULL;
}
int main(void)
{
pthread_t th_a,th_b;
void *retval;
init(&buffer);
pthread_create(&th_a,NULL,producer,0);
pthread_create(&th_b,NULL,consumer,0);
pthread_join(th_a,&retval);
pthread_join(th_b,&retval);
return 0;
}
Makefile如下::
CC=gcc
OBJS2=mYpthead.o
EXEC2=mYpthead
all:$(EXEC2)
$(EXEC2):$(OBJS2)
$(CC) -o $@ $(OBJS2) -lpthread
clean:
rm -f $(OBJS2) $(EXEC2)
<script type="text/javascript" id="wumiiRelatedItems"> </script>