|
utex_t mutex;
int idx;
int val;
} shared = {
PTHREAD_MUTEX_INITIALIZER
};
struct {
pthread_mutex_t mutex;
pthread_cond_t cond;
int nready;
} nready = {
PTHREAD_MUTEX_INITIALIZER,
PTHREAD_COND_INITIALIZER
};
void* produce(void*);
void* consume(void*);
int main(int argc, char** argv){
int i;
int nthreads;
int count[MAXTHREAD];
pthread_t tid_produce[MAXTHREAD], tid_consume;
if(argc != 3){
printf("arg error\n");
return 1;
}
nitem = min(MAXITEM,atoi(argv[1]));
nthreads = min(MAXTHREAD, atoi(argv[2]));
for(i = 0; i < nthreads; ++i){
count[i] = 0;
pthread_create(&tid_produce[i], NULL, produce, &count[i]);
}
pthread_create(&tid_consume, NULL, consume, NULL);
for(i = 0; i < nthreads; ++i){
pthread_join(tid_produce[i], NULL);
printf("cout[%d] = %d\n", i, count[i]);
}
pthread_join(tid_consume, NULL);
return 0;
}
void* produce(void* arg){
while(1){
pthread_mutex_lock(&shared.mutex);
if(shared.idx >= nitem){
pthread_mutex_unlock(&shared.mutex);
return NULL;
}
buf[shared.idx] = shared.val;
shared.idx++;
shared.val++;
pthread_mutex_unlock(&shared.mutex);
pthread_mutex_lock(&nready.mutex);
if(nready.nready == 0){
pthread_cond_signal(&nready.cond);
}
nready.nready++;
pthread_mutex_unlock(&nready.mutex);
*((int*) arg) += 1;
}
}
void* consume(void* arg){
int i;
for(i = 0; i < nitem; ++i){
pthread_mutex_lock(&nready.mutex);
while(nready.nready == 0){
pthread_cond_wait(&nready.cond, &nready.mutex);
}
nready.nready--;
pthread_mutex_unlock(&nready.mutex);
if(buf[i] != i){
printf("buf[%d] = %d\n", i, buf[i]);
}
}
printf("buf[%d] = %d\n", nitem-1, buf[nitem-1]);
}
|