lyfz_repos/经典版源码/Others/ClientTest/CritSection.h

#ifndef __CRITSECTION_20160228__
#define __CRITSECTION_20160228__

/************************************************************************/
/*  Copyright (C), 2016-2020, [IT], 保留所有权利;
/*  模 块 名：ThreadSection;
/*  描    述：线程使用的临界值保护对象;
/*
/*  版    本：[V];	
/*  作    者：[IT];
/*  日    期：[2/23/2016];
/*
/*
/*  注    意：;
/*
/*  修改记录：[IT];
/*  修改日期：;
/*  修改版本：;
/*  修改内容：;
/************************************************************************/
class ThreadSection
{
public:
    ThreadSection(){
        HRESULT hr = Init();
        (hr);
    }

    ~ThreadSection(){
        DeleteCriticalSection(&_CriticalSection);
    }

    /// Enter/Lock a critical section
    bool Lock(){
        bool result = false;
        __try
        {
            EnterCriticalSection(&_CriticalSection);
            result = true;
        }
        __except(STATUS_NO_MEMORY == GetExceptionCode())
        {
        }
        return result;
    }

    /// Leave/Unlock a critical section
    bool Unlock(){
        bool result = false;
        __try
        {
            LeaveCriticalSection(&_CriticalSection);
            result = true;
        }
        __except(STATUS_NO_MEMORY == GetExceptionCode())
        {
        }
        return result;
    }

private:
    /// Initialize critical section
    HRESULT Init() throw(){
        HRESULT hRes = E_FAIL;
        __try
        {
            InitializeCriticalSection(&_CriticalSection);
            hRes = S_OK;
        }
        // structured exception may be raised in low memory situations
        __except(STATUS_NO_MEMORY == GetExceptionCode())
        {
            hRes = E_OUTOFMEMORY;
        }
        return hRes;
    }

    ThreadSection(const ThreadSection & tSection);
    ThreadSection &operator=(const ThreadSection & tSection);
    CRITICAL_SECTION _CriticalSection;
};

// 垫片类;
class AutoThreadSection
{
public:
    /// auto section - constructor
    AutoThreadSection(__in ThreadSection* pSection){
		//if ( pSection != NULL ){
			_pSection = pSection;
			_pSection->Lock();
		//}
    }

    ~AutoThreadSection(){
      // if ( _pSection != NULL ) 
		   _pSection->Unlock();
    }
private:
    AutoThreadSection(const AutoThreadSection & tSection);
    AutoThreadSection &operator=(const AutoThreadSection & tSection);
    ThreadSection * _pSection;
};

#endif // __CRITSECTION_20160228__