c++实现的一对锁的封装(挺有意思的)

来源:互联网 发布:it was until 编辑:程序博客网 时间:2024/06/07 01:44

       在使用锁的时候,出现的最多的问题是由于忘记对锁的释放从而出现的各种死锁问题。所以经常会去检查对应的锁是否在之后的代码中得到合适的释放。前段时间学习到很巧妙的实现,即在构造函数中实现对锁的锁定,在析构函数中实现对锁的释放,这样只需要在使用锁的时候,定义一个锁,接下来即进入临界区,最后在退出对应的作用域的时候,自动调用析构函数,对应的锁自动释放, 对于一些锁的简单使用,还是比较不错的,个人觉得挺有意思的,拿出来和大家分享下。

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

  1. class ThreadRWLock
  2. {
  3. public:
  4.        ThreadRWLock(){
  5.               pthread_rwlockattr_t attr;
  6.               pthread_rwlockattr_init(&attr);
  7.               pthread_rwlock_init(&_rw_lock, &attr); 
  8.         }
  9.        ~ThreadRWLock(){
  10.              pthread_rwlock_destory(&_rw_lock);
  11.          }
  12.         int AcquireRead(void){
  13.              return pthread_rwlock_rdlock(&_rw_lock);
  14.         }
  15.         int AcquireWrite(void){
  16.              return pthread_rwlock_wrlock(&_rw_lock);
  17.         }
  18.         int Release(void){
  19.             return pthread_rwlock_unlock(&_rw_lock);
  20.         }
  21. private:
  22.        pthread_rwlock_t  _rw_lock;
  23. };

----------------------------------------------------------------------------------------------------------------------------------------------------------------
//读锁的实现
  1. class ReadLocker
  2. {
  3. public:
  4.   explicit ReadLocker(ThreadRWLock& rwlock): _lock_point(&rwlock)
  5.   {
  6.       _lock_point->AcquireRead();
  7.       _is_lock = true;
  8.   }
  9.   ~ReadLocker(){
  10.     if (_is_lock)
  11.         _lock_point->Release();
  12.   }
  13.   void Lock(){
  14.     if (!_is_lock)
  15.         _lock_point->AcquireRead();
  16.        _is_lock = true;
  17.     }
  18.   void Unlock(){
  19.       if (_is_lock){
  20.          _lock_point->Release();
  21.       }  
  22.       _is_lock = false;
  23.   }

  24. private:
  25.   ThreadRWLock * _lock_point;
  26.   bool is_lock;
  27. };
//读锁的使用示例:
  1. void fun(void){
  2.       ThreadRWLock   lock;
  3.       ReadLocker        readlock(lock); //自动获取读锁
  4.      /*
  5.          临界区
  6.      */
  7.      //退出作用域自动释放读锁
  8. }
----------------------------------------------------------------------------------------------------------------------------------------------------------------
//写锁的实现:
  1. class WriteLocker
  2. {
  3. public:
  4.   explicit WriteLocker(ThreadRWLock& rwlock)
  5.               : _lock_point(rwlock){
  6.               _lock_point->AcquireWrite();
  7.               _is_lock = true;
  8.   }
  9.   explicit WriteLocker(ThreadRWLock* rwlock)
  10.             :_lock_point(rwlock){
  11.              _lock_point->AcquireWrite();
  12.               _is_lock = true;
  13.   }
  14.   ~WriteLocker(){
  15.      if(_is_lock)
  16.          _lock_point->Release();
  17.         _is_lock = false;   
  18.       }
  19.    void Lock(){
  20.       if (!_is_lock)
  21.            _lock_point->AcquireWrite();
  22.           _is_lock = true;
  23.    }
  24.    void Unlock(){
  25.      if (_is_lock)
  26.          _lock_point->Release();
  27.         _is_lock = false;
  28.    }
  29. private:
  30.   ThreadRWLock * _lock_point;
  31.   bool _is_lock;
  32. };
//写锁的使用示例(与读锁类似)
  1. void fun(void){
  2.       ThreadRWLock   lock;
  3.       WriteLocker        writelock(lock); //自动获取写锁
  4.       /*
  5.          临界区
  6.        */
  7.       //退出作用域自动释放写锁
  8. }
----------------------------------------------------------------------------------------------------------------------------------------------------------------