www.pudn.com > ChatUseIOCP.rar > DtSocketBase.cpp

#include "StdAfx.h" 
#include "dtsocketbase.h" 
 
Datatal::DtSocketBase::DtSocketBase(void) 
{ 
	m_bReconnect = false; 
 
	//Create events 
	m_hNewDataEvent = CreateEvent(NULL, TRUE, FALSE, NULL); 
	m_nBufferPos = 0; 
	m_lOutBuffers.pFirst = NULL; 
	m_lOutBuffers.pLast = NULL; 
} 
 
Datatal::DtSocketBase::~DtSocketBase(void) 
{ 
	Disconnect(true); 
	CloseHandle(m_hNewDataEvent); 
} 
 
bool Datatal::DtSocketBase::Disconnect(bool bStopThread /* = false */) 
{ 
	SetState(DTSS_DISCONNECTED); 
 
	m_CritWrite.Lock(); 
	while (m_lOutBuffers.pFirst) 
		m_lOutBuffers.RemoveFirst(); 
	m_CritWrite.Unlock(); 
 
	if (bStopThread) 
	{ 
		SetReconnect(false); 
		StopThread(); 
	} 
 
	return true; 
} 
 
void Datatal::DtSocketBase::WriteLog(int nImportance, const char* pszGroup, LPCTSTR lpszFormat, ...) 
{ 
	char		szFormattedString[1124]; 
	va_list		argptr; 
 
	//Initialize  
	memset(szFormattedString, 0, sizeof(szFormattedString)); 
 
	//Format Data 
	va_start(argptr,lpszFormat); 
	vsprintf(szFormattedString,lpszFormat,argptr); 
	va_end(argptr); 
 
	HandleWriteLog(nImportance, pszGroup, szFormattedString); 
} 
 
 
bool Datatal::DtSocketBase::Send(char* szBuffer, size_t nBufSize) 
{ 
 
	//Lock outbuffer 
	m_CritWrite.Lock(); 
	WriteLog(Datatal::LP_NORMAL, "Send", "Appending new buffer, size: %d", nBufSize); 
	m_lOutBuffers.Append(szBuffer, nBufSize); 
	m_CritWrite.Unlock(); 
 
	//Signal thread that new data exist. 
	if (!SetEvent(m_hNewDataEvent)) 
		WriteLog(Datatal::LP_HIGH, "Send", "Set hNewDataEvent Failed!"); 
 
	return true; 
}