Windows 同步信号量 SynSemaphore

SynSemaphore.cpp

#include "SynSemaphore.h"
#ifndef WIN32_LEAN_AND_MEAN
#define WIN32_LEAN_AND_MEAN
#endif // !WIN32_LEAN_AND_MEAN
#include <Windows.h>


CSynSemaphore::CSynSemaphore()
{
	m_nHandle = (int)CreateSemaphore(NULL, 0, 1, NULL);
}


CSynSemaphore::~CSynSemaphore()
{
	if (m_nHandle)
	{
		ReleaseSemaphore((HANDLE)m_nHandle, MAXLONG, 0);
		CloseHandle((HANDLE)m_nHandle);
		m_nHandle = NULL;
	}
}

void CSynSemaphore::SetSemaphore()
{
	if (m_nHandle)
	{
		ReleaseSemaphore((HANDLE)m_nHandle, 1, 0);
	}
}

void CSynSemaphore::WaitSemaphore()
{
	if (m_nHandle)
	{
		WaitForSingleObject((HANDLE)m_nHandle, INFINITE);
	}
}

bool CSynSemaphore::WaitSemaphore(int nTimeOut)
{
	if (m_nHandle)
	{
		auto nResult = WaitForSingleObject((HANDLE)m_nHandle, nTimeOut);
		if (nResult == WAIT_TIMEOUT)
		{
			return false;
		}
		else if(WAIT_OBJECT_0 == nResult)
		{
			return true;
		}
	}
	return false;
}

SynSemaphore.h

#pragma once
class CSynSemaphore
{
public:
	CSynSemaphore();
	~CSynSemaphore();

	void SetSemaphore();
	void WaitSemaphore();

	bool WaitSemaphore(int nTimeOut);

private:
	int m_nHandle;
};