基于windows api实现的共享锁/独占锁

众所周知，windows平台上实现线程同步，或者说资源的加锁与解锁的方法有内核事件、临界区、互斥量、信号量，甚至interlocked系列函数等多种手段。但是在日常的编程中，我们使用这些手段对 “多个线程同时对同一个资源进行读写” 的时候，在读写之前先要对资源假锁，读写完之后要对资源解锁。

设想这样一种情况，有一个ftp服务器，每天有很频繁的对这个ftp服务的文件进行下载，但是几乎好几天才会对这些文件进行更新。在我们每一次对文件下载的时候，读取文件的时候都要对文件进行加锁，以保证同时没有其他人对文件进行写入。但是这些加锁的行为，在99%的时候都是不会有人同时写入文件的，只有1%的情况下会有人同时也要写入文件。这样的话，我们多锁就大大的浪费了，而且你在加锁的同时，别人即使也只是读取文件，也需要等你先解锁。

解决这个问题的办法是，对文件的读取设置共享锁，多个线程可以同时读文件，不会互相阻塞。再设置独占锁，当要对文件进行写入的时候，加上独占锁，这样别的线程此时不能读也不能写。

windows提供了一个称为slim 的共享/独占锁来解决这个问题。但是呢，slim只在vista和window server 2008才支持。在之前的版本上没有支持。于是，我就w利用现有的线程同步手段，来模拟达到slim这一个共享/独占锁的功能，代码封装如下：

</pre><pre name="code" class="cpp">//共享和独占锁（读不锁，写锁），适用于资源的读的频率比写的频率高的情况
//共享锁: 大家都可以同时读，但是不能写。
//独占锁: 就是只有一个人独占使用，不管是读还是写
//规定：acquire和release必须成对出现，不支持嵌套以及互相嵌套
//缺点：需要对加锁过程本身进行临界区控制，会带来细微的性能损失
#ifdef __cplusplus
extern "C" {
#endif
struct SELock //Shared & Exclusive lock
{   
    RTL_CRITICAL_SECTION sec_shared,sec_exclusive; //对加锁代码本身进行临界区控制
    HANDLE exclusive_evt;
    HANDLE shared_evt;
    volatile long shared_count;
};

//初始化一个SE锁
_inline void InitializeSELock(SELock *lock)
{
    InitializeCriticalSection(&lock->sec_shared);
    InitializeCriticalSection(&lock->sec_exclusive);
    lock->exclusive_evt = CreateEventW(NULL,TRUE,TRUE,NULL);
    lock->shared_evt = CreateEventW(NULL,TRUE,TRUE,NULL);
    lock->shared_count = 0;
}

//清理一个SE锁
_inline void DeleteSELock(SELock *lock)
{    
    DeleteCriticalSection(&lock->sec_shared);
    DeleteCriticalSection(&lock->sec_exclusive);
    CloseHandle(lock->exclusive_evt);
    CloseHandle(lock->shared_evt);
    lock->shared_count = 0;
}

//请求共享锁，用于读
_inline void AcquireSELockShared(SELock *lock)
{
    EnterCriticalSection(&lock->sec_exclusive);   
    EnterCriticalSection(&lock->sec_shared);     
    WaitForSingleObject(lock->exclusive_evt,INFINITE); //等待独占锁    
    ++lock->shared_count;
    if(lock->shared_count)
        ResetEvent(lock->shared_evt); //打开共享锁      
    LeaveCriticalSection(&lock->sec_shared);
    LeaveCriticalSection(&lock->sec_exclusive);
}

//释放共享锁
_inline void ReleaseSELockShared(SELock *lock)
{
    EnterCriticalSection(&lock->sec_shared);    
    --lock->shared_count;    
    if(!lock->shared_count)
        SetEvent(lock->shared_evt); //关闭共享锁        
    LeaveCriticalSection(&lock->sec_shared);    
}

//请求独占锁
_inline void AcquireSELockExclusive(SELock *lock)
{
    EnterCriticalSection(&lock->sec_exclusive);    
    WaitForSingleObject(lock->exclusive_evt,INFINITE); //等待独占锁
    WaitForSingleObject(lock->shared_evt,INFINITE); //等待共享锁
    ResetEvent(lock->exclusive_evt); //打开独占锁
    LeaveCriticalSection(&lock->sec_exclusive);
}

//释放独占锁
_inline void ReleaseSELockExclusive(SELock *lock)
{
    SetEvent(lock->exclusive_evt); //关闭独占锁
}


#ifdef __cplusplus
}
#endif

基于windows api实现的共享锁/独占锁,布布扣,bubuko.com