class CEventSelectModel
{
public:
CEventSelectModel();
~CEventSelectModel();
static DWORD WINAPI EventSelectDelProc(LPVOID lparamer);
BOOL CreateEventThread();
BOOL GetThreadState();
void SetThreadState(BOOL bFlag);
BOOL AddSock(SOCKET ClientSock);
BOOL DeleteSock(int Index);
int GetClientCount(); //获取当前客户端数量
SOCKET GetClientSock(int Index);
WSAEVENT GetClientEvent(int Index);
WSAEVENT* GetClientEventAry();
//void ReleaseEventSelectModel();//释放自己
private:
int m_nClientCount;
SOCKET m_ClientSock[MAXIMUM_WAIT_OBJECTS];//64
WSAEVENT m_ClientEvent[MAXIMUM_WAIT_OBJECTS];
BOOL m_bThreadState;
};
#include "stdafx.h"
#include "EventSelectModel.h"
//
// Construction/Destruction
//
CEventSelectModel::CEventSelectModel()
{
m_nClientCount = 0;
m_bThreadState = FALSE;
}
CEventSelectModel::~CEventSelectModel()
{
}
DWORD WINAPI CEventSelectModel::EventSelectDelProc(LPVOID lparamer)
{
int nResult = 0;
int nIndex = 0;
WSANETWORKEVENTS NetWorkEvents = { 0 };
TCHAR szRecvBuf[2048] = "";
CEventSelectModel *pEventSelectModel = (CEventSelectModel*)lparamer;
while(true)
{
nResult = WSAWaitForMultipleEvents(pEventSelectModel->GetClientCount(),pEventSelectModel->GetClientEventAry(),FALSE,1000,FALSE);
if( WSA_WAIT_FAILED == nResult || nResult == WSA_WAIT_TIMEOUT)
{
if( 0 == pEventSelectModel->GetClientCount())
{
break;
}
continue;
}
nIndex = nResult;
WSAEnumNetworkEvents(pEventSelectModel->GetClientSock(nIndex),pEventSelectModel->GetClientEvent(nIndex),&NetWorkEvents);
if(NetWorkEvents.lNetworkEvents && FD_READ)
{
nResult = recv(pEventSelectModel->GetClientSock(nIndex),szRecvBuf,2048,0);
if( 0 == nResult || SOCKET_ERROR == nResult)
{
pEventSelectModel->DeleteSock(nIndex);
}
else
{
szRecvBuf[nResult] = '\0';
cout << "Client" << pEventSelectModel->GetClientSock(nIndex) << ":" << szRecvBuf << endl;
send(pEventSelectModel->GetClientSock(nIndex),szRecvBuf,strlen(szRecvBuf),0);
}
}
else
{
pEventSelectModel->DeleteSock(nIndex);
}
}
pEventSelectModel->SetThreadState(FALSE);
return 0;
}
BOOL CEventSelectModel::AddSock(SOCKET ClientSock)
{
if( INVALID_SOCKET != ClientSock)
{
m_ClientSock[m_nClientCount] = ClientSock;
m_ClientEvent[m_nClientCount] = WSACreateEvent();
WSAEventSelect(ClientSock,m_ClientEvent[m_nClientCount],FD_READ|FD_CLOSE);
m_nClientCount++;
if( FALSE == m_bThreadState) // 创建线程
{
CreateEventThread();
m_bThreadState = TRUE;
}
}
return FALSE;
}
int CEventSelectModel::GetClientCount() //获取当前客户端数量
{
return m_nClientCount;
}
SOCKET CEventSelectModel::GetClientSock(int Index)
{
return m_ClientSock[Index];
}
WSAEVENT CEventSelectModel::GetClientEvent(int Index)
{
return m_ClientEvent[Index];
}
WSAEVENT* CEventSelectModel::GetClientEventAry()
{
return m_ClientEvent;
}
BOOL CEventSelectModel::DeleteSock(int Index)
{
if( Index >= 0 && Index < MAXIMUM_WAIT_OBJECTS)
{
closesocket(m_ClientSock[Index]);
WSACloseEvent(m_ClientEvent[Index]);
m_ClientSock[Index] = m_ClientSock[m_nClientCount - 1];
m_ClientEvent[Index] = m_ClientEvent[m_nClientCount - 1];
m_ClientSock[m_nClientCount - 1] = NULL;
m_ClientEvent[m_nClientCount - 1] = NULL;
m_nClientCount--;
return TRUE;
}
return FALSE;
}
BOOL CEventSelectModel::CreateEventThread()
{
CloseHandle(CreateThread(NULL,0,CEventSelectModel::EventSelectDelProc,this,0,NULL));
return TRUE;
}
BOOL CEventSelectModel::GetThreadState()
{
return m_bThreadState;
}
void CEventSelectModel::SetThreadState(BOOL bFlag)
{
m_bThreadState = bFlag;
}