windows-nt/Source/XPSP1/NT/public/sdk/inc/thrdbase.h
2020-09-26 16:20:57 +08:00

108 lines
2.1 KiB
C++

//***************************************************************************
//
// Copyright (c) 1997-2001 Microsoft Corporation, All Rights Reserved
//
// ThrdBase.h
//
// Purpose: Definition of ThreadBase class
//
//***************************************************************************
#if _MSC_VER > 1000
#pragma once
#endif
#ifndef __THREADBASE_H__
#define __THREADBASE_H__
class POLARITY CThreadBase
{
public:
enum THREAD_SAFETY_MECHANISM
{
etsmFirst = 0,
etsmSerialized = 0,
etsmPriorityRead,
etsmPriorityWrite,
etsmLast
};
// Construction/Destruction
CThreadBase( THREAD_SAFETY_MECHANISM etsm = etsmSerialized );
virtual ~CThreadBase();
// Thread Safe Ref/Counting functions
LONG AddRef( void );
LONG Release( void );
// Provide Readable Read/Write accessors should
// we not want to serialize at a later date. Note
// that timeouts have no meaning unless we're
// doing a non-serialized implementation.
BOOL BeginRead( DWORD dwTimeOut = INFINITE );
void EndRead( void );
BOOL BeginWrite( DWORD dwTimeOut = INFINITE );
void EndWrite( void );
protected:
virtual void OnFinalRelease( void );
// Thread Safety functions
private:
CRITICAL_SECTION m_cs;
LONG m_lRefCount;
THREAD_SAFETY_MECHANISM m_etsm;
// Private thread safety functions. We can maybe promote
// these to protected if we see a need to later, however
// for right now, everyone should specify if they mean
// to read or write when they wish to access data that
// may change.
void Lock( void );
void Unlock( void );
};
inline BOOL CThreadBase::BeginRead( DWORD dwTimeout /*=INFINITE*/ )
{
EnterCriticalSection( &m_cs );
return TRUE;
}
inline void CThreadBase::EndRead( void )
{
LeaveCriticalSection( &m_cs );
}
inline BOOL CThreadBase::BeginWrite( DWORD dwTimeout /*=INFINITE*/ )
{
EnterCriticalSection( &m_cs );
return TRUE;
}
inline void CThreadBase::EndWrite( void )
{
LeaveCriticalSection( &m_cs );
}
inline void CThreadBase::Lock( void )
{
EnterCriticalSection( &m_cs );
}
inline void CThreadBase::Unlock( void )
{
LeaveCriticalSection( &m_cs );
}
#endif