多线程同步之读者写者问题

#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>

#define M 6        //number of readers
#define N 2        //number of writers
int readCount = 0; //current number of readers
pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;    //用于互斥地修改readCount变量
pthread_mutex_t rw = PTHREAD_MUTEX_INITIALIZER;      //用于读、写以及写、写之间的互斥

void* write(void *arg)
{
    pthread_mutex_lock(&rw);
    printf("writer %d is writing\n", arg);
    sleep(1);
    printf("writer %d is leaving\n", arg);
    pthread_mutex_unlock(&rw);
}

void* read(void *arg)
{
    pthread_mutex_lock(&mutex);
    if (readCount == 0)
        pthread_mutex_lock(&rw);
    ++readCount;
    pthread_mutex_unlock(&mutex);

    printf("reader %d is reading\n", arg);
    sleep(1);
    printf("reader %d is leaving\n", arg);

    pthread_mutex_lock(&mutex);
    --readCount;
    if (readCount == 0)
        pthread_mutex_unlock(&rw);
    pthread_mutex_unlock(&mutex);
}

int main()
{
    pthread_t readers[M], writers[N];
    int i;
    for (i = 0; i < M; ++i)
        pthread_create(&readers[i], NULL, read, (void*)i);
    for (i = 0; i < N; ++i)
        pthread_create(&writers[i], NULL, write, (void*)i);
    sleep(10);
    return 0;
}