Esempio n. 1
0
/*************************************************************************
	Agent lord working thread
*************************************************************************/
unsigned int System::lordWorkingThread(void)
{
	while(true)
	{
		//----------------------------------------
		// wait cmd sign
		::WaitForSingleObject(m_hLordCmdSign, INFINITE);

		// enter critical section
		::EnterCriticalSection(&m_csLordCmdQueue);
		if(m_qLordCmd.empty())
		{
			// leave critical section
			::LeaveCriticalSection(&m_csLordCmdQueue);
			continue;
		}
		// get cmd
		LordCmd theCmd = m_qLordCmd.front();
		m_qLordCmd.pop();

		if(m_qLordCmd.empty()) ::ResetEvent(m_hLordCmdSign);
		// leave critical section
		::LeaveCriticalSection(&m_csLordCmdQueue);

		//----------------------------------------
		//Process cmd
		switch(theCmd.type)
		{
		// create agents...
		case LCT_CreateAgent:
			{
				bool	bStartNow = LOWORD(theCmd.nParam2)==1;
				int		nCounts = (int)HIWORD(theCmd.nParam2);
				AgentCreateParam* pCreateparam = (AgentCreateParam*)(INT_PTR)theCmd.nParam1;

				// 保存登录用户名
				char		buf[128];
				int			iParam	= -1;
				std::string	sParam;
				for( int j=0; j<(int)pCreateparam->vParams.size(); j++ )
				{
					if( pCreateparam->vParams[j].first == "AccountName" )
					{
						iParam		= j;
						sParam		= pCreateparam->vParams[j].second;
						break;
					}
				}

				// 形如000X时,特殊处理
				int			iNum	= atoi( sParam.substr( sParam.length()-4 ).c_str() );

				for( int i=0; i<nCounts; i++ )
				{
					if( i == 1 )
					{
						sParam		= sParam.substr( 0, sParam.length()-4 );
					}

					// 一组多用户命名规则
					if( i > 0 && iParam >= 0 )
					{
						sprintf( buf, "*****@*****.**", i + iNum );
						pCreateparam->vParams[iParam].second	= sParam + buf;
					}
					else
					{
						sprintf( buf, "@game.hopecool.com" );
						pCreateparam->vParams[iParam].second	= sParam + buf;
					}

					Agent* pAgent = m_pAgentManager->createAgent(*pCreateparam);

					if(bStartNow) 
					{
						if( pAgent->start() == false )
							m_hUnStartAgents.push_back( pAgent->getID() );
						::Sleep(AGENT_AROUSE_TIME_INTERVAL);
					}

					::SendMessage(m_hNotifyWnd, WMA_AGENT_NUM, m_pAgentManager->getAgentNum(), 0);
				}
				//delete org createparam
				delete pCreateparam; pCreateparam=0;

				if( m_hUnStartAgents.size() > 0 ) pushLordCommand( LCT_ArouseStartFail );
			}
			break;
		// arouse agent...
		case LCT_ArouseAll:
			{
				//Start all agent
				AgentManager::AgentIterator agent = m_pAgentManager->getAgentIterator();
				while (!agent.isAtEnd())
				{
					agent.getCurrentValue()->start();
					++agent;
					::Sleep(AGENT_AROUSE_TIME_INTERVAL);
				}
				if( m_hUnStartAgents.size() > 0 ) pushLordCommand( LCT_ArouseStartFail );
			}
			break;
		// stop all agent...
		case LCT_StopAll:
			{
				// resume all first
				::SetEvent(m_hResumeHandle);

				// send stop sign
				int nAgentNum = m_pAgentManager->getAgentNum();
				int nAgentTeam = nAgentNum / MAXIMUM_WAIT_OBJECTS + 1;
				HANDLE (*hAllWorkThread)[MAXIMUM_WAIT_OBJECTS] = new HANDLE[nAgentTeam][MAXIMUM_WAIT_OBJECTS];
				AgentManager::AgentIterator agent = m_pAgentManager->getAgentIterator();
				int nIndex = 0;
				int nOffset = 0;
				while (!agent.isAtEnd())
				{
					hAllWorkThread[nOffset][nIndex++] = agent.getCurrentValue()->getWorkThread();
					if( nIndex == MAXIMUM_WAIT_OBJECTS ) { nIndex = 0; nOffset ++; }
					agent.getCurrentValue()->stop();
					++agent;
				}

				//wait for all thread stop(10 sec)...
				for( int i = 0; i < nAgentTeam; i ++ )
				{
					int nWait = i == nAgentTeam - 1 ? nIndex : MAXIMUM_WAIT_OBJECTS;
					if( nWait == 0 ) break;
					if(WAIT_TIMEOUT == ::WaitForMultipleObjects(nWait, hAllWorkThread[i], TRUE, 10*1000)) {
						//TODO ...
						// Terminate all thread!
						int a=0;
					}
				}

				delete [] hAllWorkThread;
				setState(Ready);

				break;
			}
			break;
		case LCT_ArouseStartFail:
			{
				for( std::list<int>::iterator it = m_hUnStartAgents.begin(); it != m_hUnStartAgents.end(); ) {
					Agent* pAgent = m_pAgentManager->findAgent( *it );
					if( pAgent ) {
						if( pAgent->start() == true ) {
							m_hUnStartAgents.pop_front();
						}
						else {
							++ it;
						}
						Sleep( 100 );
					}
					else {
						m_hUnStartAgents.pop_front();
					}
				}
				if( m_hUnStartAgents.size() > 0 ) pushLordCommand( LCT_ArouseStartFail );
			}
			break;

		default:
			break;
		}

	};

	return 0;
}
Esempio n. 2
0
bool Agent::operator==(Agent &other) const
{
	return (this->ID == other.getID());
}