一个简单但很好用的线程锁

最新推荐文章于 2024-07-25 00:24:08 发布

fuzhufang

最新推荐文章于 2024-07-25 00:24:08 发布

阅读量706

点赞数

分类专栏： C/C++ 文章标签： class null

本文链接：https://blog.csdn.net/fuzhufang/article/details/4233342

版权

C/C++ 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

#ifndef __LOCK_H__
#define __LOCK_H__

#ifdef _WIN32
#include <WinSock2.h>
#else
#include <pthread.h>
#endif

class CLock
{
public:
class CMutex
{
  friend class CLock;
public:
  CMutex()
  {
#ifdef _WIN32
   InitializeCriticalSection(&m_cs);
#else
   pthread_mutex_init(&m_tmt, NULL);
#endif
  }

  ~CMutex()
  {
#ifdef _WIN32
   DeleteCriticalSection(&m_cs);
#else
   pthread_mutex_destroy(&m_tmt);
#endif
  }

private:
  void lock()
  {
#ifdef _WIN32
   EnterCriticalSection(&m_cs);
#else
   pthread_mutex_lock(&m_tmt);
#endif
  }

  void unlock()
  {
#ifdef _WIN32
   LeaveCriticalSection(&m_cs);
#else
   pthread_mutex_unlock(&m_tmt);
#endif
  }

private:
#ifdef _WIN32
  CRITICAL_SECTION m_cs;
#else
  pthread_mutex_t  m_tmt;
#endif
};

explicit CLock(CMutex& mutex) : m_mutex(mutex) {m_mutex.lock();}
~CLock() {m_mutex.unlock();}

private:
  CMutex&    m_mutex;
};