一个线程同步类

来源:互联网 发布:淘宝新手购物视频教程 编辑:程序博客网 时间:2024/06/05 21:51

// LockObj.h: interface for the CLockObj class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(LOCKOBJ_H)
#define LOCKOBJ_H

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000
#include "stdtype.h"

class CLockObj
{
public:
 virtual void Lock() = 0;
 virtual void Unlock() = 0;
};

class CLockObjCS: public CLockObj
{
protected:
 CRITICAL_SECTION m_cs;
public:
 CLockObjCS(){
  InitializeCriticalSection(&m_cs);
 };

 CLockObjCS(const CLockObjCS& ){
  InitializeCriticalSection(&m_cs);
 };

 CLockObjCS(const CLockObjCS* ){
  InitializeCriticalSection(&m_cs);
 };

 virtual ~CLockObjCS(){
  DeleteCriticalSection(&m_cs);
 };
 virtual void Lock(){
  EnterCriticalSection(&m_cs);
 };

 virtual void Unlock(){
  LeaveCriticalSection(&m_cs);
 };
};

class CLock
{
protected:
 CLockObj* m_pLockObj;
public:
 CLock(CLockObj *pLockObj):m_pLockObj(pLockObj){
  m_pLockObj->Lock();
 };

 virtual ~CLock( ){
  m_pLockObj->Unlock();
 }
};

#endif

使用方法
CLockObjCS g_mycs;
int g_iCount=0;
ThreadAdd()
{
  CLock lock(&g_mycs);
  ++g_iCount;
}

TheadDec()
{
  CLock lock(&g_mycs);
  --g_iCount;
}

原创粉丝点击