#include "StdAfx.h"
#include "WHDataLocker.h"

//////////////////////////////////////////////////////////////////////////////////

//构造函数
CWHDataLocker::CWHDataLocker(CCriticalSection & CriticalSection, bool bLockAtOnce) 
	: m_CriticalSection(CriticalSection)
{
	//设置变量
	m_nLockCount=0;

	//锁定对象
	if (bLockAtOnce==true)
	{
		Lock();
	}

	return;
}

//析构函数
CWHDataLocker::~CWHDataLocker()
{
	//解除锁定
	while (m_nLockCount>0)
	{
		UnLock();
	}

	return;
}

//锁定函数
VOID CWHDataLocker::Lock()
{
	//锁定对象
	m_CriticalSection.Lock();

	//设置变量
	m_nLockCount++;

	return;
}

//解锁函数
VOID CWHDataLocker::UnLock()
{
	//效验状态
	ASSERT(m_nLockCount>0);
	if (m_nLockCount==0) return;

	//设置变量
	m_nLockCount--;

	//解除锁定
	m_CriticalSection.Unlock();

	return;
}

//////////////////////////////////////////////////////////////////////////////////