diff options
| author | megamage <none@none> | 2008-12-08 16:34:03 -0600 |
|---|---|---|
| committer | megamage <none@none> | 2008-12-08 16:34:03 -0600 |
| commit | ce643187f06cd0b3829ab90deff335f5c9ec5a84 (patch) | |
| tree | dc09eece1fc5dce2a942bdc27a4a045518ef9daa /src/game/WorldSocketMgr.cpp | |
| parent | 5321be4c0aa48daaf84ffd602d47e650e53403ef (diff) | |
*Update to Mangos 6888. Source: Mangos.
*Major change: Make corpse erase async, Avoid DB access on deleting guild member.
*Skip 6885 because of performance concern.
--HG--
branch : trunk
Diffstat (limited to 'src/game/WorldSocketMgr.cpp')
| -rw-r--r-- | src/game/WorldSocketMgr.cpp | 462 |
1 files changed, 235 insertions, 227 deletions
diff --git a/src/game/WorldSocketMgr.cpp b/src/game/WorldSocketMgr.cpp index a11960f33f5..baf8896f3f2 100644 --- a/src/game/WorldSocketMgr.cpp +++ b/src/game/WorldSocketMgr.cpp @@ -1,5 +1,5 @@ /* -* Copyright (C) 2005-2008 MaNGOS <http://www.mangosproject.org/> +* Copyright (C) 2005-2008,2007 MaNGOS <http://getmangos.com/> * * Copyright (C) 2008 Trinity <http://www.trinitycore.org/> * @@ -10,12 +10,12 @@ * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of -* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software -* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ /** \file WorldSocketMgr.cpp @@ -46,315 +46,323 @@ #include "Database/DatabaseEnv.h" #include "WorldSocket.h" -/** -* This is a helper class to WorldSocketMgr ,that manages +/** +* This is a helper class to WorldSocketMgr ,that manages * network threads, and assigning connections from acceptor thread * to other network threads */ class ReactorRunnable : protected ACE_Task_Base { -public: + public: - ReactorRunnable () : - m_ThreadId (-1), - m_Connections (0), - m_Reactor (0) - { - ACE_Reactor_Impl* imp = 0; + ReactorRunnable () : + m_ThreadId (-1), + m_Connections (0), + m_Reactor (0) + { + ACE_Reactor_Impl* imp = 0; -#if defined (ACE_HAS_EVENT_POLL) || defined (ACE_HAS_DEV_POLL) + #if defined (ACE_HAS_EVENT_POLL) || defined (ACE_HAS_DEV_POLL) - imp = new ACE_Dev_Poll_Reactor (); + imp = new ACE_Dev_Poll_Reactor (); - imp->max_notify_iterations (128); - imp->restart (1); + imp->max_notify_iterations (128); + imp->restart (1); -#else + #else - imp = new ACE_TP_Reactor (); - imp->max_notify_iterations (128); + imp = new ACE_TP_Reactor (); + imp->max_notify_iterations (128); -#endif + #endif - m_Reactor = new ACE_Reactor (imp, 1); - } + m_Reactor = new ACE_Reactor (imp, 1); + } - virtual ~ReactorRunnable () - { - this->Stop (); - this->Wait (); + virtual ~ReactorRunnable () + { + Stop (); + Wait (); - if (m_Reactor) - delete m_Reactor; - } + if (m_Reactor) + delete m_Reactor; + } - void Stop () - { - m_Reactor->end_reactor_event_loop (); - } + void Stop () + { + m_Reactor->end_reactor_event_loop (); + } - int Start () - { - if (m_ThreadId != -1) - return -1; + int Start () + { + if (m_ThreadId != -1) + return -1; - return (m_ThreadId = this->activate ()); - } + return (m_ThreadId = activate ()); + } - void Wait () - { - ACE_Task_Base::wait (); - } + void Wait () { ACE_Task_Base::wait (); } - long Connections () - { - return static_cast<long> (m_Connections.value ()); - } + long Connections () + { + return static_cast<long> (m_Connections.value ()); + } - int AddSocket (WorldSocket* sock) - { - ACE_GUARD_RETURN (ACE_Thread_Mutex, Guard, m_NewSockets_Lock, -1); + int AddSocket (WorldSocket* sock) + { + ACE_GUARD_RETURN (ACE_Thread_Mutex, Guard, m_NewSockets_Lock, -1); - ++m_Connections; - sock->AddReference(); - sock->reactor (m_Reactor); - m_NewSockets.insert (sock); + ++m_Connections; + sock->AddReference(); + sock->reactor (m_Reactor); + m_NewSockets.insert (sock); - return 0; - } + return 0; + } - ACE_Reactor* GetReactor () - { - return m_Reactor; - } + ACE_Reactor* GetReactor () + { + return m_Reactor; + } -protected: + protected: - void AddNewSockets () - { - ACE_GUARD (ACE_Thread_Mutex, Guard, m_NewSockets_Lock); + void AddNewSockets () + { + ACE_GUARD (ACE_Thread_Mutex, Guard, m_NewSockets_Lock); - if (m_NewSockets.empty ()) - return; + if (m_NewSockets.empty ()) + return; - for (SocketSet::iterator i = m_NewSockets.begin (); i != m_NewSockets.end (); ++i) - { - WorldSocket* sock = (*i); + for (SocketSet::iterator i = m_NewSockets.begin (); i != m_NewSockets.end (); ++i) + { + WorldSocket* sock = (*i); - if (sock->IsClosed ()) - { - sock->RemoveReference (); - --m_Connections; - } - else - m_Sockets.insert (sock); - } + if (sock->IsClosed ()) + { + sock->RemoveReference (); + --m_Connections; + } + else + m_Sockets.insert (sock); + } - m_NewSockets.clear (); - } + m_NewSockets.clear (); + } - virtual int svc (void) - { - DEBUG_LOG ("Network Thread Starting"); + virtual int svc () + { + DEBUG_LOG ("Network Thread Starting"); - WorldDatabase.ThreadStart (); + WorldDatabase.ThreadStart (); - ACE_ASSERT (m_Reactor); + ACE_ASSERT (m_Reactor); - SocketSet::iterator i, t; + SocketSet::iterator i, t; - while (!m_Reactor->reactor_event_loop_done ()) - { - // Don't be too smart to move this outside the loop - // the run_reactor_event_loop will modify interval - ACE_Time_Value interval (0, 10000); + while (!m_Reactor->reactor_event_loop_done ()) + { + // dont be too smart to move this outside the loop + // the run_reactor_event_loop will modify interval + ACE_Time_Value interval (0, 10000); - if (m_Reactor->run_reactor_event_loop (interval) == -1) - break; + if (m_Reactor->run_reactor_event_loop (interval) == -1) + break; - AddNewSockets (); + AddNewSockets (); - for (i = m_Sockets.begin (); i != m_Sockets.end ();) - { - if ((*i)->Update () == -1) - { - t = i; - i++; - (*t)->CloseSocket (); - (*t)->RemoveReference (); - --m_Connections; - m_Sockets.erase (t); - } - else - i++; - } - } - - WorldDatabase.ThreadEnd (); + for (i = m_Sockets.begin (); i != m_Sockets.end ();) + { + if ((*i)->Update () == -1) + { + t = i; + i++; + (*t)->CloseSocket (); + (*t)->RemoveReference (); + --m_Connections; + m_Sockets.erase (t); + } + else + i++; + } + } - DEBUG_LOG ("Network Thread Exitting"); - - return 0; - } + WorldDatabase.ThreadEnd (); -private: - typedef ACE_Atomic_Op<ACE_SYNCH_MUTEX, long> AtomicInt; - typedef std::set<WorldSocket*> SocketSet; + DEBUG_LOG ("Network Thread Exitting"); - ACE_Reactor* m_Reactor; - AtomicInt m_Connections; - int m_ThreadId; + return 0; + } - SocketSet m_Sockets; + private: + typedef ACE_Atomic_Op<ACE_SYNCH_MUTEX, long> AtomicInt; + typedef std::set<WorldSocket*> SocketSet; - SocketSet m_NewSockets; - ACE_Thread_Mutex m_NewSockets_Lock; -}; + ACE_Reactor* m_Reactor; + AtomicInt m_Connections; + int m_ThreadId; + SocketSet m_Sockets; + SocketSet m_NewSockets; + ACE_Thread_Mutex m_NewSockets_Lock; +}; WorldSocketMgr::WorldSocketMgr () : -m_NetThreadsCount (0), -m_NetThreads (0), -m_SockOutKBuff (-1), -m_SockOutUBuff (65536), -m_UseNoDelay (true), -m_Acceptor (0) {} + m_NetThreadsCount (0), + m_NetThreads (0), + m_SockOutKBuff (-1), + m_SockOutUBuff (65536), + m_UseNoDelay (true), + m_Acceptor (0) +{ +} WorldSocketMgr::~WorldSocketMgr () { - if (m_NetThreads) - delete [] m_NetThreads; + if (m_NetThreads) + delete [] m_NetThreads; - if(m_Acceptor) - delete m_Acceptor; + if(m_Acceptor) + delete m_Acceptor; } -int WorldSocketMgr::StartReactiveIO (ACE_UINT16 port, const char* address) +int +WorldSocketMgr::StartReactiveIO (ACE_UINT16 port, const char* address) { - m_UseNoDelay = sConfig.GetBoolDefault ("Network.TcpNodelay", true); + m_UseNoDelay = sConfig.GetBoolDefault ("Network.TcpNodelay", true); - int num_threads = sConfig.GetIntDefault ("Network.Threads", 1); + int num_threads = sConfig.GetIntDefault ("Network.Threads", 1); - if (num_threads <= 0) - { - sLog.outError ("Network.Threads is wrong in your config file"); - return -1; - } + if (num_threads <= 0) + { + sLog.outError ("Network.Threads is wrong in your config file"); + return -1; + } - m_NetThreadsCount = static_cast<size_t> (num_threads + 1); + m_NetThreadsCount = static_cast<size_t> (num_threads + 1); - m_NetThreads = new ReactorRunnable[m_NetThreadsCount]; + m_NetThreads = new ReactorRunnable[m_NetThreadsCount]; - sLog.outBasic ("Max alowed socket connections %d",ACE::max_handles ()); + sLog.outBasic ("Max allowed socket connections %d",ACE::max_handles ()); - m_SockOutKBuff = sConfig.GetIntDefault ("Network.OutKBuff", -1); // -1 means use default + // -1 means use default + m_SockOutKBuff = sConfig.GetIntDefault ("Network.OutKBuff", -1); - m_SockOutUBuff = sConfig.GetIntDefault ("Network.OutUBuff", 65536); + m_SockOutUBuff = sConfig.GetIntDefault ("Network.OutUBuff", 65536); - if ( m_SockOutUBuff <= 0 ) - { - sLog.outError ("Network.OutUBuff is wrong in your config file"); - return -1; - } + if ( m_SockOutUBuff <= 0 ) + { + sLog.outError ("Network.OutUBuff is wrong in your config file"); + return -1; + } - WorldSocket::Acceptor *acc = new WorldSocket::Acceptor; - m_Acceptor = acc; + WorldSocket::Acceptor *acc = new WorldSocket::Acceptor; + m_Acceptor = acc; - ACE_INET_Addr listen_addr (port, address); + ACE_INET_Addr listen_addr (port, address); - if (acc->open (listen_addr, m_NetThreads[0].GetReactor (), ACE_NONBLOCK) == -1) - { - sLog.outError ("Failed to open acceptor ,check if the port is free"); - return -1; - } + if (acc->open (listen_addr, m_NetThreads[0].GetReactor (), ACE_NONBLOCK) == -1) + { + sLog.outError ("Failed to open acceptor ,check if the port is free"); + return -1; + } - for (size_t i = 0; i < m_NetThreadsCount; ++i) - m_NetThreads[i].Start (); + for (size_t i = 0; i < m_NetThreadsCount; ++i) + m_NetThreads[i].Start (); - return 0; + return 0; } -int WorldSocketMgr::StartNetwork (ACE_UINT16 port, const char* address) +int +WorldSocketMgr::StartNetwork (ACE_UINT16 port, const char* address) { - if (!sLog.IsOutDebug ()) - ACE_Log_Msg::instance ()->priority_mask (LM_ERROR, ACE_Log_Msg::PROCESS); + if (!sLog.IsOutDebug ()) + ACE_Log_Msg::instance ()->priority_mask (LM_ERROR, ACE_Log_Msg::PROCESS); - if (this->StartReactiveIO (port, address) == -1) - return -1; + if (StartReactiveIO (port, address) == -1) + return -1; - return 0; + return 0; } -void WorldSocketMgr::StopNetwork () +void +WorldSocketMgr::StopNetwork () { - if (m_Acceptor) - { - WorldSocket::Acceptor* acc = dynamic_cast<WorldSocket::Acceptor*> (m_Acceptor); + if (m_Acceptor) + { + WorldSocket::Acceptor* acc = dynamic_cast<WorldSocket::Acceptor*> (m_Acceptor); - if (acc) - acc->close (); - } + if (acc) + acc->close (); + } - if (m_NetThreadsCount != 0) - { - for (size_t i = 0; i < m_NetThreadsCount; ++i) - m_NetThreads[i].Stop (); - } + if (m_NetThreadsCount != 0) + { + for (size_t i = 0; i < m_NetThreadsCount; ++i) + m_NetThreads[i].Stop (); + } - this->Wait (); + Wait (); } -void WorldSocketMgr::Wait () +void +WorldSocketMgr::Wait () { - if (m_NetThreadsCount != 0) - { - for (size_t i = 0; i < m_NetThreadsCount; ++i) - m_NetThreads[i].Wait (); - } + if (m_NetThreadsCount != 0) + { + for (size_t i = 0; i < m_NetThreadsCount; ++i) + m_NetThreads[i].Wait (); + } } -int WorldSocketMgr::OnSocketOpen (WorldSocket* sock) +int +WorldSocketMgr::OnSocketOpen (WorldSocket* sock) { - // set some options here - if (m_SockOutKBuff >= 0) - if (sock->peer ().set_option (SOL_SOCKET, - SO_SNDBUF, - (void*) & m_SockOutKBuff, - sizeof (int)) == -1 && errno != ENOTSUP) - { - sLog.outError ("WorldSocketMgr::OnSocketOpen set_option SO_SNDBUF"); - return -1; - } - - static const int ndoption = 1; - - // Set TCP_NODELAY. - if (m_UseNoDelay) - if (sock->peer ().set_option (ACE_IPPROTO_TCP, - TCP_NODELAY, - (void*) & ndoption, - sizeof (int)) == -1) - { - sLog.outError ("WorldSocketMgr::OnSocketOpen: peer ().set_option TCP_NODELAY errno = %s", ACE_OS::strerror (errno)); - return -1; - } - - sock->m_OutBufferSize = static_cast<size_t> (m_SockOutUBuff); - - // we skip the Acceptor Thread - size_t min = 1; - - ACE_ASSERT (m_NetThreadsCount >= 1); - - for (size_t i = 1; i < m_NetThreadsCount; ++i) - if (m_NetThreads[i].Connections () < m_NetThreads[min].Connections ()) - min = i; - - return m_NetThreads[min].AddSocket (sock); + // set some options here + if (m_SockOutKBuff >= 0) + { + if (sock->peer ().set_option (SOL_SOCKET, + SO_SNDBUF, + (void*) & m_SockOutKBuff, + sizeof (int)) == -1 && errno != ENOTSUP) + { + sLog.outError ("WorldSocketMgr::OnSocketOpen set_option SO_SNDBUF"); + return -1; + } + } + + static const int ndoption = 1; + + // Set TCP_NODELAY. + if (m_UseNoDelay) + { + if (sock->peer ().set_option (ACE_IPPROTO_TCP, + TCP_NODELAY, + (void*)&ndoption, + sizeof (int)) == -1) + { + sLog.outError ("WorldSocketMgr::OnSocketOpen: peer ().set_option TCP_NODELAY errno = %s", ACE_OS::strerror (errno)); + return -1; + } + } + + sock->m_OutBufferSize = static_cast<size_t> (m_SockOutUBuff); + + // we skip the Acceptor Thread + size_t min = 1; + + ACE_ASSERT (m_NetThreadsCount >= 1); + + for (size_t i = 1; i < m_NetThreadsCount; ++i) + if (m_NetThreads[i].Connections () < m_NetThreads[min].Connections ()) + min = i; + + return m_NetThreads[min].AddSocket (sock); } -WorldSocketMgr* WorldSocketMgr::Instance () +WorldSocketMgr* +WorldSocketMgr::Instance () { - return ACE_Singleton<WorldSocketMgr,ACE_Thread_Mutex>::instance(); + return ACE_Singleton<WorldSocketMgr,ACE_Thread_Mutex>::instance(); } |
