SocketMgr< SocketType > Class Template Reference

#include "SocketMgr.h"

Public Member Functions
virtual	~SocketMgr ()
virtual bool	StartNetwork (Acore::Asio::IoContext &ioContext, std::string const &bindIp, uint16 port, int threadCount)
virtual void	StopNetwork ()
void	Wait ()
virtual void	OnSocketOpen (tcp::socket &&sock, uint32 threadIndex)
int32	GetNetworkThreadCount () const
uint32	SelectThreadWithMinConnections () const
std::pair< tcp::socket *, uint32 >	GetSocketForAccept ()

Protected Member Functions
	SocketMgr ()=default
virtual NetworkThread< SocketType > *	CreateThreads () const =0

Protected Attributes
std::unique_ptr< AsyncAcceptor >	_acceptor
std::unique_ptr< NetworkThread< SocketType >[]>	_threads
int32	_threadCount {}

Detailed Description

template<class SocketType>
class SocketMgr< SocketType >

Constructor & Destructor Documentation

~SocketMgr()

template<class SocketType >

virtual SocketMgr< SocketType >::~SocketMgr ( )

inlinevirtual

    {
        ASSERT(!_threads && !_acceptor && !_threadCount, "StopNetwork must be called prior to SocketMgr destruction");
    }

References SocketMgr< SocketType >::_acceptor, SocketMgr< SocketType >::_threadCount, SocketMgr< SocketType >::_threads, and ASSERT.

SocketMgr()

template<class SocketType >

SocketMgr< SocketType >::SocketMgr ( )

protecteddefault

Member Function Documentation

CreateThreads()

template<class SocketType >

virtual NetworkThread< SocketType > * SocketMgr< SocketType >::CreateThreads ( ) const

protectedpure virtual

Implemented in AuthSocketMgr, and WorldSocketMgr.

Referenced by SocketMgr< SocketType >::StartNetwork().

GetNetworkThreadCount()

template<class SocketType >

int32 SocketMgr< SocketType >::GetNetworkThreadCount ( ) const

inline

{ return _threadCount; }

References SocketMgr< SocketType >::_threadCount.

GetSocketForAccept()

template<class SocketType >

std::pair< tcp::socket *, uint32 > SocketMgr< SocketType >::GetSocketForAccept ( )

inline

    {
        uint32 threadIndex = SelectThreadWithMinConnections();
        return { _threads[threadIndex].GetSocketForAccept(), threadIndex };
    }

References SocketMgr< SocketType >::_threads, and SocketMgr< SocketType >::SelectThreadWithMinConnections().

Referenced by SocketMgr< SocketType >::StartNetwork().

OnSocketOpen()

template<class SocketType >

virtual void SocketMgr< SocketType >::OnSocketOpen ( tcp::socket &&  sock, uint32  threadIndex  )

inlinevirtual

Reimplemented in WorldSocketMgr.

    {
        try
        {
            std::shared_ptr<SocketType> newSocket = std::make_shared<SocketType>(std::move(sock));
            _threads[threadIndex].AddSocket(newSocket);
        }
        catch (boost::system::system_error const& err)
        {
            LOG_WARN("network", "Failed to retrieve client's remote address {}", err.what());
        }
    }

References SocketMgr< SocketType >::_threads, and LOG_WARN.

Referenced by AuthSocketMgr::OnSocketAccept().

SelectThreadWithMinConnections()

template<class SocketType >

uint32 SocketMgr< SocketType >::SelectThreadWithMinConnections ( ) const

inline

    {
        uint32 min = 0;
 
        for (int32 i = 1; i < _threadCount; ++i)
            if (_threads[i].GetConnectionCount() < _threads[min].GetConnectionCount())
                min = i;
 
        return min;
    }

References SocketMgr< SocketType >::_threadCount, and SocketMgr< SocketType >::_threads.

Referenced by SocketMgr< SocketType >::GetSocketForAccept().

StartNetwork()

template<class SocketType >

virtual bool SocketMgr< SocketType >::StartNetwork ( Acore::Asio::IoContext &  ioContext, std::string const &  bindIp, uint16  port, int  threadCount  )

inlinevirtual

Reimplemented in AuthSocketMgr.

    {
        ASSERT(threadCount > 0);
 
        std::unique_ptr<AsyncAcceptor> acceptor;
        try
        {
            acceptor = std::make_unique<AsyncAcceptor>(ioContext, bindIp, port);
        }
        catch (boost::system::system_error const& err)
        {
            LOG_ERROR("network", "Exception caught in SocketMgr.StartNetwork ({}:{}): {}", bindIp, port, err.what());
            return false;
        }
 
        if (!acceptor->Bind())
        {
            LOG_ERROR("network", "StartNetwork failed to bind socket acceptor");
            return false;
        }
 
        _acceptor = std::move(acceptor);
        _threadCount = threadCount;
        _threads = std::unique_ptr<NetworkThread<SocketType>[]>(CreateThreads());
 
        ASSERT(_threads);
 
        for (int32 i = 0; i < _threadCount; ++i)
            _threads[i].Start();
 
        _acceptor->SetSocketFactory([this]() { return GetSocketForAccept(); });
        return true;
    }

References SocketMgr< SocketType >::_acceptor, SocketMgr< SocketType >::_threadCount, SocketMgr< SocketType >::_threads, ASSERT, SocketMgr< SocketType >::CreateThreads(), SocketMgr< SocketType >::GetSocketForAccept(), and LOG_ERROR.

StopNetwork()

template<class SocketType >

virtual void SocketMgr< SocketType >::StopNetwork ( )

inlinevirtual

Reimplemented in WorldSocketMgr.

    {
        _acceptor->Close();
 
        for (int32 i = 0; i < _threadCount; ++i)
            _threads[i].Stop();
 
        Wait();
 
        _acceptor.reset();
        _threads.reset();
        _threadCount = 0;
    }

References SocketMgr< SocketType >::_acceptor, SocketMgr< SocketType >::_threadCount, SocketMgr< SocketType >::_threads, and SocketMgr< SocketType >::Wait().

Wait()

template<class SocketType >

void SocketMgr< SocketType >::Wait ( )

inline

    {
        for (int32 i = 0; i < _threadCount; ++i)
            _threads[i].Wait();
    }

References SocketMgr< SocketType >::_threadCount, SocketMgr< SocketType >::_threads, and SocketMgr< SocketType >::Wait().

Referenced by SocketMgr< SocketType >::StopNetwork(), and SocketMgr< SocketType >::Wait().

Member Data Documentation

_acceptor

template<class SocketType >

std::unique_ptr<AsyncAcceptor> SocketMgr< SocketType >::_acceptor

protected

Referenced by SocketMgr< SocketType >::StartNetwork(), SocketMgr< SocketType >::StopNetwork(), and SocketMgr< SocketType >::~SocketMgr().

_threadCount

template<class SocketType >

int32 SocketMgr< SocketType >::_threadCount {}

protected

Referenced by SocketMgr< SocketType >::GetNetworkThreadCount(), SocketMgr< SocketType >::SelectThreadWithMinConnections(), SocketMgr< SocketType >::StartNetwork(), SocketMgr< SocketType >::StopNetwork(), SocketMgr< SocketType >::Wait(), and SocketMgr< SocketType >::~SocketMgr().

_threads

template<class SocketType >

std::unique_ptr<NetworkThread<SocketType>[]> SocketMgr< SocketType >::_threads

protected

Referenced by SocketMgr< SocketType >::GetSocketForAccept(), SocketMgr< SocketType >::OnSocketOpen(), SocketMgr< SocketType >::SelectThreadWithMinConnections(), SocketMgr< SocketType >::StartNetwork(), SocketMgr< SocketType >::StopNetwork(), SocketMgr< SocketType >::Wait(), and SocketMgr< SocketType >::~SocketMgr().