C++封装EventSelect

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;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值