tcp_acceptor.cpp

Go to the documentation of this file.

/* Diagnostic Client library
 * Copyright (C) 2024  Avijit Dey
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */
 
// includes
#include "boost-support/server/tcp/tcp_acceptor.h"
 
#include <boost/asio.hpp>
 
#include "boost-support/common/logger.h"
#include "boost-support/socket/tcp/tcp_socket.h"
 
namespace boost_support {
namespace server {
namespace tcp {
namespace {
using Tcp = boost::asio::ip::tcp;
 
using TcpIpAddress = boost::asio::ip::address;
 
std::string CreateServerName(std::string_view server_name, std::uint16_t server_count) {
  std::string final_server_name{server_name};
  final_server_name.append(std::to_string(server_count));
  return final_server_name;
}
}  // namespace
 
class TcpAcceptor::TcpAcceptorImpl final {
 public:
  using TcpSocket = socket::tcp::TcpSocket;
 
 public:
  TcpAcceptorImpl(std::string_view acceptor_name, std::string_view local_ip_address,
                  std::uint16_t local_port_num, std::uint8_t maximum_connection) noexcept
      : io_context_{},
        server_count_{0u},
        acceptor_name_{acceptor_name},
        acceptor_{io_context_,
                  Tcp::endpoint(TcpIpAddress::from_string(std::string{local_ip_address}.c_str()),
                                local_port_num)} {
    acceptor_.listen(maximum_connection);
  }
 
  std::optional<TcpServer> GetTcpServer() noexcept {
    using TcpErrorCodeType = boost::system::error_code;
    std::optional<TcpServer> tcp_server{};
    TcpErrorCodeType ec{};
    Tcp::endpoint endpoint{};
 
    // blocking accept
    TcpSocket::Socket accepted_socket{acceptor_.accept(endpoint, ec)};
    if (ec.value() == boost::system::errc::success) {
      tcp_server.emplace(CreateServerName(acceptor_name_, server_count_),
                         TcpSocket{std::move(accepted_socket)});
      common::logger::LibBoostLogger::GetLibBoostLogger().GetLogger().LogDebug(
          FILE_NAME, __LINE__, __func__, [&endpoint](std::stringstream &msg) {
            msg << "Tcp socket connection received from client "
                << "<" << endpoint.address().to_string() << "," << endpoint.port() << ">";
          });
      // increment the server count
      server_count_++;
    } else {
      common::logger::LibBoostLogger::GetLibBoostLogger().GetLogger().LogError(
          FILE_NAME, __LINE__, __func__, [ec](std::stringstream &msg) {
            msg << "Tcp socket accept failed with error: " << ec.message();
          });
    }
    return tcp_server;
  }
 
 private:
  using Acceptor = boost::asio::ip::tcp::acceptor;
 
  boost::asio::io_context io_context_;
 
  std::uint16_t server_count_;
 
  std::string acceptor_name_;
 
  Acceptor acceptor_;
};
 
TcpAcceptor::TcpAcceptor(std::string_view acceptor_name, std::string_view local_ip_address,
                         std::uint16_t local_port_num, std::uint8_t maximum_connection) noexcept
    : tcp_acceptor_impl_{std::make_unique<TcpAcceptorImpl>(acceptor_name, local_ip_address,
                                                           local_port_num, maximum_connection)} {}
 
TcpAcceptor::~TcpAcceptor() noexcept = default;
 
std::optional<TcpServer> TcpAcceptor::GetTcpServer() noexcept {
  return tcp_acceptor_impl_->GetTcpServer();
}
 
}  // namespace tcp
}  // namespace server
}  // namespace boost_support