forked from LosaDev/ls-2019-source
101 lines
2.0 KiB
C++
101 lines
2.0 KiB
C++
// cIocpQueue.cpp: implementation of the cIocpQueue class.
|
|
//
|
|
//////////////////////////////////////////////////////////////////////
|
|
|
|
#include "StdAfx.h"
|
|
#include "cIocpQueue.h"
|
|
|
|
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////
|
|
// Construction/Destruction
|
|
//////////////////////////////////////////////////////////////////////
|
|
|
|
cIocpQueue::cIocpQueue() : m_count(0), m_completionPort(NULL)
|
|
{
|
|
}
|
|
|
|
cIocpQueue::~cIocpQueue()
|
|
{
|
|
Cleanup();
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////
|
|
// Startup/Cleanup
|
|
//////////////////////////////////////////////////////////////////////
|
|
BOOL cIocpQueue::Startup( int32 timeout )
|
|
{
|
|
if( !m_completionPort )
|
|
{
|
|
m_completionPort = CreateIoCompletionPort( INVALID_HANDLE_VALUE,
|
|
NULL,
|
|
0,
|
|
0 );
|
|
if( !m_completionPort ) return FALSE;
|
|
|
|
m_timeout = timeout;
|
|
return TRUE;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
void cIocpQueue::Cleanup()
|
|
{
|
|
if( m_completionPort )
|
|
{
|
|
CloseHandle( m_completionPort );
|
|
m_completionPort = NULL;
|
|
}
|
|
}
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////
|
|
// Enqueue/Dequeue
|
|
//////////////////////////////////////////////////////////////////////
|
|
BOOL cIocpQueue::Enqueue( const DWORD completionKey, const DWORD bytes )
|
|
{
|
|
if( m_completionPort )
|
|
{
|
|
if(PostQueuedCompletionStatus( m_completionPort,
|
|
bytes,
|
|
completionKey,
|
|
NULL ))
|
|
{
|
|
InterlockedIncrement( &m_count );
|
|
return TRUE;
|
|
}
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
LPDWORD cIocpQueue::Dequeue()
|
|
{
|
|
DWORD bytes = 0;
|
|
return Dequeue( bytes );
|
|
}
|
|
|
|
LPDWORD cIocpQueue::Dequeue( DWORD &bytes )
|
|
{
|
|
LPDWORD completionKey = NULL;
|
|
LPOVERLAPPED overlapped = NULL;
|
|
|
|
if( m_completionPort )
|
|
{
|
|
if(GetQueuedCompletionStatus( m_completionPort,
|
|
&bytes,
|
|
reinterpret_cast<LPDWORD>(&completionKey),
|
|
&overlapped,
|
|
m_timeout ))
|
|
{
|
|
InterlockedDecrement( &m_count );
|
|
}
|
|
}
|
|
|
|
return completionKey;
|
|
}
|
|
|
|
|
|
|