一个轻量级的单写多读锁

一个轻量级的单写多读锁

    与《windows核心编程》上的那个相比最大的优势是体积小, 它只有四个字节(《windows核心编程》上的那个至少是它的10倍), 如果你有大量对象需要进行单写多读访问的话, 它会比较适用. 缺点是它在加锁时使用的等待函数是Sleep, 如果访问冲突很多的话, 效率比较低. 代码如下(很简单, 就不多做解释了):

发现代码里有个bug，参见http://www.cppblog.com/localvar/archive/2008/01/08/132759.html 的第二条 。时间有限，我暂时不改了，请大家自己注意一下。

// 头文件 #ifndef SWMR_LOCK_H #define SWMR_LOCK_H #ifndef SWMR_LOCK_NUMBER_OF_WRITER #define SWMR_LOCK_NUMBER_OF_WRITER        1 #endif // SWMR_LOCK_NUMBER_OF_WRITER typedef volatile long SWMR_LOCK void SwmrLockInitialize( SWMR_LOCK* pLock ); void SwmrLockWriteLock( SWMR_LOCK* pLock ); void SwmrLockWriteUnlock( SWMR_LOCK* pLock ); void SwmrLockReadLock( SWMR_LOCK* pLock ); void SwmrLockReadUnlock( SWMR_LOCK* pLock ); void SwmrLockUninitialize( SWMR_LOCK* pLock ); #endif

--------------------------------------------------------------------------------

// 实现文件 #include "srmrl.h" //////////////////////////////////////////////////////////////////////////////// #define WRITING_FLAG    0x40000000 //////////////////////////////////////////////////////////////////////////////// void SwmrLockInitialize( SWMR_LOCK* pLock ) {     *pLock = 0; } //////////////////////////////////////////////////////////////////////////////// void SwmrLockWriteLock( SWMR_LOCK* pLock ) {     long old, xchg;     do {         old = *pLock; #if( SWMR_LOCK_NUMBER_OF_WRITER > 1 )         if( old & WRITING_FLAG )         {             Sleep( 0 );             continue;         } #endif // ( SWMR_LOCK_NUMBER_OF_WRITER > 1 )         xchg = old | WRITING_FLAG;     } while( _InterlockedCompareExchange( pLock, xchg, old ) != old );     // wait until all readers quit reading     while( old != WRITING_FLAG )     {         Sleep( 0 );         old = *pLock;     } } //////////////////////////////////////////////////////////////////////////////// void SwmrLockWriteUnlock( SWMR_LOCK* pLock ) {     *pLock = 0; } //////////////////////////////////////////////////////////////////////////////// void SwmrLockReadLock( SWMR_LOCK* pLock ) {     long old, xchg;     do {         old = *pLock;         if( old & WRITING_FLAG )         {             Sleep( 0 );             continue;         }         xchg = old + 1;     } while( _InterlockedCompareExchange( pLock, xchg, old ) != old ); } //////////////////////////////////////////////////////////////////////////////// void SwmrLockReadUnlock( SWMR_LOCK* pLock ) {     _InterlockedDecrement( pLock ); } //////////////////////////////////////////////////////////////////////////////// void SwmrLockUninitialize( SWMR_LOCK* pLock ) {     pLock; // has nothing to do }