#include "StdAfx.h"
|
#include "WaitThreadPools.h"
|
|
CWaitThreadPools::CWaitThreadPools(int nThreadCount) : CThreadPools(nThreadCount)
|
{
|
m_pWait = NULL;
|
m_pWaitCallback = WaitCallback;
|
|
m_hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);
|
}
|
|
CWaitThreadPools::~CWaitThreadPools(void)
|
{
|
CloseWaitThread();
|
|
CloseHandle(m_hEvent);
|
}
|
|
VOID CALLBACK CWaitThreadPools::WaitCallback(PTP_CALLBACK_INSTANCE pInstance, PVOID pParameter, PTP_WAIT pWait, TP_WAIT_RESULT WaitResult)
|
{
|
// Instance, Parameter, and Work not used in this example.
|
UNREFERENCED_PARAMETER(pInstance);
|
UNREFERENCED_PARAMETER(pParameter);
|
UNREFERENCED_PARAMETER(pWait);
|
UNREFERENCED_PARAMETER(WaitResult);
|
|
return;
|
}
|
|
BOOL CWaitThreadPools::CreateWaitThread(PVOID pParameter)
|
{
|
if (NULL==m_pPool || NULL==m_pCleanupGroup) return FALSE;
|
|
TP_CALLBACK_ENVIRON* pCallBackEnviron = (TP_CALLBACK_ENVIRON*)&m_CallBackEnviron;
|
|
PTP_WAIT pWait = CreateThreadpoolWait((PTP_WAIT_CALLBACK)WaitCallback, pParameter, pCallBackEnviron);
|
if (NULL==pWait) return FALSE;
|
|
SetThreadpoolWait(pWait, m_hEvent, NULL);
|
|
return TRUE;
|
}
|
|
void CWaitThreadPools::CloseWaitThread()
|
{
|
if (NULL==m_pWait) return;
|
|
WaitForThreadpoolWaitCallbacks(m_pWait, TRUE);
|
|
CloseThreadpoolWait(m_pWait);
|
|
m_pWait = NULL;
|
}
|
|
BOOL CWaitThreadPools::SetEvent1()
|
{
|
if (m_hEvent==NULL) return FALSE;
|
|
return SetEvent(m_hEvent);
|
}
|
|
BOOL CWaitThreadPools::ResetEvent1()
|
{
|
if (m_hEvent==NULL) return FALSE;
|
|
return ResetEvent(m_hEvent);
|
}
|
