linux读写锁

 

#ifndef CONFIG_ENV_H
#define CONFIG_ENV_H


#define RWLOCK_NAMESPACE                 RWLOCK

#define NAMESPACE_BEGIN(x)              namespace x {
#define NAMESPACE_END                   }
#define USING_NAMESPACE(x)              using namespace x;


#define BEGIN_RWLOCK_NAMESPACE   NAMESPACE_BEGIN(RWLOCK_NAMESPACE)
#define END_RWLOCK_NAMESPACE             NAMESPACE_END
#define USING_RWLOCK_NAMESPACE   USING_NAMESPACE(RWLOCK_NAMESPACE)


#if defined(_WIN32) && !defined(SEPARATE_COMPILE)
        #ifdef DLL_API_EXPORT
                #define API_EXPORT __declspec(dllexport)
        #else
                #define API_EXPORT __declspec(dllimport)
        #endif
#else
        #define API_EXPORT
#endif

#endif

 

 

#ifndef PTHREAD_RW_LOCK_H_
#define PTHREAD_RW_LOCK_H_


#include "config_env.h"
#include <pthread.h>

BEGIN_RWLOCK_NAMESPACE
class API_EXPORT CReadWriteLock
{
public:
        CReadWriteLock();
        ~CReadWriteLock();


        int ReadAcquire();


        int ReadRelease();


        int WriteAcquire();


        int WriteRelease();
private:
#ifdef _WIN32

#else                   //for posix unix
        pthread_rwlock_t        m_rwl;
#endif


        CReadWriteLock(CReadWriteLock &lock);
        CReadWriteLock & operator=(CReadWriteLock &lock);
};
END_RWLOCK_NAMESPACE

#endif


//---------------------------------------------------------------------------
#include "pthread_rw_lock.h"


USING_RWLOCK_NAMESPACE


CReadWriteLock::CReadWriteLock()
{
#ifdef _WIN32


#else
        pthread_rwlock_init(&m_rwl, NULL);
#endif
}


CReadWriteLock::~CReadWriteLock()
{
#ifdef _WIN32


#else
        pthread_rwlock_destroy(&m_rwl);
#endif
}




CReadWriteLock::CReadWriteLock(CReadWriteLock &lock)
{
}


CReadWriteLock & CReadWriteLock::operator=(CReadWriteLock &lock)
{
}


int CReadWriteLock::ReadAcquire()
{
        int nRet = -1;
#ifdef _WIN32


#else
        nRet = pthread_rwlock_rdlock(&m_rwl);
#endif
        return nRet;
}


int CReadWriteLock::ReadRelease()
{
        int nRet = -1;
#ifdef _WIN32


#else
        nRet = pthread_rwlock_unlock(&m_rwl) ;
#endif
        return nRet;

}

 

int CReadWriteLock::WriteAcquire()
{
        int nRet = -1;
#ifdef _WIN32


#else
        nRet = pthread_rwlock_wrlock(&m_rwl) ;
#endif
        return nRet;
}


int CReadWriteLock::WriteRelease()
{
        int nRet = -1;
#ifdef _WIN32


#else
        nRet = pthread_rwlock_unlock(&m_rwl) ;
#endif
        return nRet;
}

 

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
读写锁是用于多线程环境下保护共享资源的一种机制。在 Linux 内核中,读写锁的实现是基于 spinlock 和原子操作的。下面我将简单介绍一下 Linux 内核中读写锁的源码实现。 读写锁的定义: ```c typedef struct { raw_rwlock_t raw_lock; } rwlock_t; ``` 其中,raw_rwlock_t 是一个原始读写锁类型,它是由内核提供的一个结构体类型,定义在 include/linux/spinlock_types.h 文件中。 raw_rwlock_t 的定义: ```c typedef struct { arch_rwlock_t raw_lock; } raw_rwlock_t; typedef struct { unsigned int lock; } arch_rwlock_t; ``` 其中,arch_rwlock_t 是一个体系结构相关的原始读写锁类型,它的实现由不同的处理器架构提供。 下面是 x86_64 架构下的 arch_rwlock_t 实现: ```c struct __arch_rwlock { unsigned int lock; }; typedef struct __arch_rwlock arch_rwlock_t; ``` 可以看到,在 x86_64 架构下,arch_rwlock_t 只包含一个 unsigned int 类型的 lock 成员变量,用于存储锁状态。 读写锁的初始化: ```c void rwlock_init(rwlock_t *lock) { raw_spin_lock_init(&lock->raw_lock.spinlock); atomic_long_set(&lock->raw_lock.rw_sem, 0); } ``` 其中,raw_spin_lock_init() 用于初始化写锁,atomic_long_set() 用于初始化读计数器。 读锁的获取: ```c void read_lock(rwlock_t *lock) { while (1) { long count = atomic_long_read(&lock->raw_lock.rw_sem); if (count >= 0) { if (atomic_long_cmpxchg(&lock->raw_lock.rw_sem, count, count + 1) == count) { break; } } else { cpu_relax(); } } raw_spin_lock(&lock->raw_lock.spinlock); } ``` 其中,atomic_long_read() 用于读取读计数器的值,如果值大于等于 0,则表示读锁可用,此时使用 atomic_long_cmpxchg() 原子操作来增加读计数器并获取读锁;如果值小于 0,则表示有写锁在使用,此时使用 cpu_relax() 函数等待写锁释放。 读锁的释放: ```c void read_unlock(rwlock_t *lock) { raw_spin_unlock(&lock->raw_lock.spinlock); atomic_long_dec(&lock->raw_lock.rw_sem); } ``` 其中,raw_spin_unlock() 用于释放写锁,atomic_long_dec() 用于减少读计数器的值。 写锁的获取: ```c void write_lock(rwlock_t *lock) { raw_spin_lock(&lock->raw_lock.spinlock); while (1) { long count = atomic_long_read(&lock->raw_lock.rw_sem); if (count == 0) { if (atomic_long_cmpxchg(&lock->raw_lock.rw_sem, 0, -1) == 0) { break; } } else { cpu_relax(); } } } ``` 其中,raw_spin_lock() 用于获取写锁,atomic_long_read() 用于读取读计数器的值,如果值等于 0,则表示读锁未被使用,此时使用 atomic_long_cmpxchg() 原子操作将读计数器的值修改为 -1 并获取写锁;如果值大于 0,则表示有读锁在使用,此时使用 cpu_relax() 函数等待读锁释放。 写锁的释放: ```c void write_unlock(rwlock_t *lock) { atomic_long_set(&lock->raw_lock.rw_sem, 0); raw_spin_unlock(&lock->raw_lock.spinlock); } ``` 其中,atomic_long_set() 用于将读计数器的值设为 0,raw_spin_unlock() 用于释放写锁。 以上就是 Linux 内核中读写锁的源码实现。值得注意的是,在多处理器环境下,读写锁的实现可能会涉及到更复杂的机制,例如写者优先等待、读者优先等待等。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

字正腔圆

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值