52 lines
962 B
C++
Executable File
52 lines
962 B
C++
Executable File
#include "precompiled.h"
|
|
|
|
#include <Network/Session.h>
|
|
#include <Network/Network.h>
|
|
#include <CLogger.h>
|
|
|
|
using namespace std;
|
|
|
|
void CSessionManager::Register(CNetSession *pSession)
|
|
{
|
|
CScopeLock scopeLock(m_Mutex);
|
|
|
|
m_Sessions[pSession]=pSession;
|
|
}
|
|
|
|
bool CSessionManager::Deregister(CNetSession *pSession)
|
|
{
|
|
CScopeLock scopeLock(m_Mutex);
|
|
|
|
SessionMap::iterator it=m_Sessions.find(pSession);
|
|
if (it == m_Sessions.end())
|
|
return false;
|
|
else
|
|
{
|
|
m_Sessions.erase(it);
|
|
return true;
|
|
}
|
|
}
|
|
|
|
void CSessionManager::Poll()
|
|
{
|
|
m_Mutex.Lock();
|
|
SessionMap::iterator it=m_Sessions.begin();
|
|
while (it != m_Sessions.end())
|
|
{
|
|
CNetMessage *pMsg;
|
|
if ((pMsg = it->second->TryPop()) != NULL)
|
|
{
|
|
CNetSession *pSess=it->second;
|
|
m_Mutex.Unlock();
|
|
if (!pSess->HandleMessage(pMsg))
|
|
{
|
|
LOG(WARNING, "CSessionManager::Poll(): Unhandled message %s.\n", pMsg->GetString().c_str());
|
|
delete pMsg;
|
|
}
|
|
m_Mutex.Lock();
|
|
}
|
|
++it;
|
|
}
|
|
m_Mutex.Unlock();
|
|
}
|