线程-读写锁

当有一个线程已经持有互斥锁时，互斥锁将所有试图进入临界区的线程都阻塞住。但是考虑一种情形，当前持有互斥锁的线程只是要读访问共享资源，而同时有其它几个线程也想读取这个共享资源，但是由于互斥锁的排它性，所有其它线程都无法获取锁，也就无法读访问共享资源了，但是实际上多个线程同时读访问共享资源并不会导致问题。在对数据的读写操作中，更多的是读操作，写操作较少，例如对数据库数据的读写应用。为了满足当前能够允许多个读出，但只允许一个写入的需求，线程提供了读写锁来实现。

 特点

• 如果有其它线程读数据，则允许其它线程执行读操作，但不允许写操作。
• 如果有其它线程写数据，则其它线程都不允许读、写操作。
• 如果某线程申请了读锁，其它线程可以再申请读锁，但不能申请写锁。
• 如果某线程申请了写锁，其它线程不能申请读锁，也不能申请写锁。

函数

#include <pthread.h>  

/* 初始化读写锁 */

int pthread_rwlock_init(pthread_rwlock_t *rwlock, const pthread_rwlockattr_t *attr); 
/* 参数：
    rwlock：指向要初始化的读写锁指针。
    attr：读写锁的属性指针。如果 attr 为 NULL 则会使用默认的属性初始化读写锁，否则使用指定的 attr 初始化读写锁。
        可以使用宏 PTHREAD_RWLOCK_INITIALIZER 静态初始化读写锁，比如：
*/
            pthread_rwlock_t my_rwlock = PTHREAD_RWLOCK_INITIALIZER;
/*
        这种方法等价于使用 NULL 指定的 attr 参数调用 pthread_rwlock_init() 来完成动态初始化，不同之处在于PTHREAD_RWLOCK_INITIALIZER 宏不进行错误检查。
*/

/* 返回值：
    成功：0，读写锁的状态将成为已初始化和已解锁。
    失败：非 0 错误码。
*/


/* 申请读锁 */
int pthread_rwlock_rdlock(pthread_rwlock_t *rwlock ); 
/* 功能：
    以阻塞方式在读写锁上获取读锁（读锁定）。如果没有写者持有该锁，并且没有写者阻塞在该锁上，则调用线程会获取读锁。如果调用线程未获取读锁，则它将阻塞直到它获取了该锁。一个线程可以在一个读写锁上多次执行读锁定。线程可以成功调用 pthread_rwlock_rdlock() 函数 n 次，但是之后该线程必须调用 pthread_rwlock_unlock() 函数 n 次才能解除锁定。
*/
/* 参数：
    rwlock：读写锁指针。
*/
/* 返回值：
    成功：0
    失败：非 0 错误码
*/

int pthread_rwlock_tryrdlock(pthread_rwlock_t *rwlock); 
/*
    用于尝试以非阻塞的方式来在读写锁上获取读锁。如果有任何的写者持有该锁或有写者阻塞在该读写锁上，则立即失败返回。
*/


/* 申请写锁 */
int pthread_rwlock_wrlock(pthread_rwlock_t *rwlock ); 
/* 功能：
    在读写锁上获取写锁（写锁定）。如果没有写者持有该锁，并且没有写者读者持有该锁，则调用线程会获取写锁。如果调用线程未获取写锁，则它将阻塞直到它获取了该锁。
*/
/* 参数：
    rwlock：读写锁指针。
*/
/* 返回值：
    成功：0
    失败：非 0 错误码
*/

int pthread_rwlock_trywrlock(pthread_rwlock_t *rwlock); 
/*
    用于尝试以非阻塞的方式来在读写锁上获取写锁。如果有任何的读者或写者持有该锁，则立即失败返回。
*/


/* 解锁 */
int pthread_rwlock_unlock (pthread_rwlock_t *rwlock); 
/* 参数：
    rwlock：读写锁指针。
*/
/* 返回值：
    成功：0
    失败：非 0 错误码
*/


/* 销毁读写锁,用于销毁一个读写锁，并释放所有相关联的资源
（所谓的所有指的是由 pthread_rwlock_init() 自动申请的资源）  
*/
int pthread_rwlock_destroy(pthread_rwlock_t *rwlock);
/* 参数：
    rwlock：读写锁指针。
*/
/* 返回值：
    成功：0
    失败：非 0 错误码
*/

代码 

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

int data = 0;
/* 定义一个读写锁 */
pthread_rwlock_t rwlock;

void* read_thread1(void* arg)
{
    /* 上读锁，其他地方的wrlock将会阻塞，rdlock不会阻塞 */
    pthread_rwlock_rdlock(&rwlock);
    printf("read1: data = %d\n",data);
    /* 等待2s */
    sleep(2);
    /* 解锁 */
    pthread_rwlock_unlock(&rwlock);
}

void* read_thread2(void* arg)
{
    /* 上读锁，其他地方的wrlock将会阻塞，rdlock不会阻塞 */
    pthread_rwlock_rdlock(&rwlock);
    printf("read2: data = %d\n",data);
    /* 解锁 */
    pthread_rwlock_unlock(&rwlock);
}

void* write_thread1(void* arg)
{
    /* 上写锁，其他地方的wrlock和rdlock都会阻塞 */
    pthread_rwlock_wrlock(&rwlock);
    data = 1;
    printf("wirte1: data = %d\n",data);
    /* 等待4s */
    sleep(4);
    /* 解锁 */
    pthread_rwlock_unlock(&rwlock);
}

void* write_thread2(void* arg)
{
    /* 上读锁，其他地方的wrlock和rdlock都会阻塞 */
    pthread_rwlock_wrlock(&rwlock);
    data = 2;
    printf("wirte2: data = %d\n",data);
    /* 解锁 */
    pthread_rwlock_unlock(&rwlock);
}

int main()
{
    pthread_t pthrd,pthrd1,pthrd2,pthrd3;
    
    /* 初始化读写锁 */
    pthread_rwlock_init(&rwlock,NULL);

    pthread_create(&pthrd1,NULL,write_thread1,NULL);
    pthread_create(&pthrd,NULL,read_thread1,NULL);
    pthread_create(&pthrd3,NULL,write_thread2,NULL);
    pthread_create(&pthrd2,NULL,read_thread2,NULL);

    pthread_join(pthrd,NULL);
    pthread_join(pthrd1,NULL);
    pthread_join(pthrd2,NULL);
    pthread_join(pthrd3,NULL);
    
    /* 销毁读写锁 */
    pthread_rwlock_destroy(&rwlock);

    return 0;
}

╭─lingyun@manjaro ~/Document/codes  
╰─➤  gcc exam.c -lpthread
╭─lingyun@manjaro ~/Document/codes  
╰─➤  ./a.out             
wirte1: data = 1        //这里打印出来后会等待4s，说明 wrlock 会让 wrlock 和 rdlock 都阻塞
read2: data = 1  
read1: data = 1        //这里read1 和 read2 是一起出现的，可见 rdlock 不会让 rdlock 阻塞
wirte2: data = 2        //这里是 read1 和 read2 打印出来后等待2s后才打印出来的，可见 rdlock 会让阻塞 wrlcok