Acceptor.cpp

Go to the documentation of this file.

/**
* @file Acceptor.cpp.
* @brief The Acceptor Class Implementation.
* @author Spices & Muduo.
*/
 
#include "Pchheader.h"
#include "Acceptor.h"
#include "EventLoop.h"
 
namespace Spices {
 
namespace Net {
 
    Acceptor::Acceptor(const InetAddress& listenAddress, bool reusePort)
        : m_IsListening(false)
    {
        SPICES_PROFILE_ZONE;
 
        /**
        * @brief Create Server Socket.
        */
        m_AcceptSocket.Create();
        m_AcceptSocket.SetReuseAddress(true);
        m_AcceptSocket.BindAddress(listenAddress);
 
        m_AcceptChannel = std::make_shared<Channel>(m_AcceptSocket.Fd(), EventLoopThreadWrapper::GetInst());
        m_AcceptChannel->SetReadCallback([=]() { HandleRead(); });
    }
 
    Acceptor::~Acceptor()
    {
        SPICES_PROFILE_ZONE;
 
        m_AcceptChannel->DisableAll();
        m_AcceptChannel->Remove();
    }
 
    void Acceptor::Listen()
    {
        SPICES_PROFILE_ZONE;
 
        m_IsListening = true;
        m_AcceptSocket.Listen();
        m_AcceptChannel->EnableReading();
    }
 
    void Acceptor::HandleRead() const
    {
        InetAddress peerAddress;
        const SOCKET connectFd = m_AcceptSocket.Accept(&peerAddress);
 
        if (m_ConnectionCallback)
        {
            m_ConnectionCallback(connectFd, peerAddress);
        }
        else
        {
            ::closesocket(connectFd);
        }
    }
 
}
 
}