Linux下使用互斥量和条件量实现读写锁（优先分配写锁）

目前只实现了加读锁的功能，后续会逐步完善。

定义自己的头文件pthread_rwlock.h

#ifndef _PTHREAD_RWLOCK_H
#define _PTHREAD_RWLOCK_H
typedef struct
{
    pthread_mutex_t rw_mutex;
    pthread_cond_t rw_condreaders;
    pthread_cond_t rw_condwriters;
    int            rw_magic;
    int            rw_nwaitreaders;
    int            rw_nwaitwriters;
    int            rw_refcount;//0 >n -1
}my_pthread_rwlock_t;

#define RW_MAGIC   0x20190324


#define MY_PTHREAD_RWLOCK_INITIALIZER {PTHREAD_MUTEX_INITIALIZER,PTHREAD_COND_INITIALIZER,PTHREAD_COND_INITIALIZER,RW_MAGIC, 0, 0, 0}
typedef int my_pthread_rwlockattr_t;

int my_pthread_rwlock_init(my_pthread_rwlock_t *rw, my_pthread_rwlockattr_t *attr);
int my_pthread_rwlock_destroy(my_pthread_rwlock_t *rw);

int my_pthread_rwlock_rdlock(my_pthread_rwlock_t *rw)
{
    int result;
    if(rw->rw_magic != RW_MAGIC)
        return -1;
    if((result = pthread_mutex_lock(&rw->rw_mutex)) != 0)
        return result;
    while(rw->rw_refcount < 0 || rw->rw_nwaitwriters > 0 )
    {
        rw->rw_nwaitreaders++;
        result = pthread_cond_wait(&rw->rw_condreaders, &rw->rw_mutex);
        rw->rw_nwaitreaders--;
        if(result != 0)
            break;
    }
    if(result == 0)
        rw->rw_refcount++;
    pthread_mutex_unlock(&rw->rw_mutex);
    return result;
}
int my_pthread_rwlock_wrlock(my_pthread_rwlock_t *rw);

int my_pthread_rwlock_unlock(my_pthread_rwlock_t *rw);


int my_pthread_rwlock_tryrdlock(my_pthread_rwlock_t *rw);
int my_pthread_rwlock_trywrlock(my_pthread_rwlock_t *rw);
#endif

测试函数test.cpp

#include<stdio.h>
#include<pthread.h>
#include"pthread_rwlock.h"

my_pthread_rwlock_t rw = MY_PTHREAD_RWLOCK_INITIALIZER;
void* thread_fun1(void *arg)
{
    my_pthread_rwlock_rdlock(&rw);
    printf("this is thread_fun1.\n"); 
}
void* thread_fun2(void *arg)
{
    my_pthread_rwlock_rdlock(&rw);
    printf("this is thread_fun2.\n"); 

}
int main()
{
    pthread_t tid1, tid2;
    pthread_create(&tid1, NULL, thread_fun1, NULL);
    pthread_create(&tid2, NULL, thread_fun2, NULL);
    pthread_join(tid1,NULL);
    pthread_join(tid2,NULL);
    return 0;
}

运行的结果

 

此程序的关键code在于，那个while循环。只有当前没有线程持有写锁，并且也没有写锁排队等待分配写锁（要保证优先分配写锁），才能够跳出循环，否则将会一直阻塞（期间while也会一直判断循环的条件是否满足）且当thread_cond_wait阻塞被真正的唤醒的时候，才能给当前调用的线程分配读锁。

目前只是实现了原来读写锁的加读锁（共享锁）的功能，后续需要逐步完善。