udp_connection.h

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/.
 */
#ifndef DIAG_CLIENT_LIB_LIB_BOOST_SUPPORT_SRC_BOOST_SUPPORT_CONNECTION_UDP_UDP_CONNECTION_H_
#define DIAG_CLIENT_LIB_LIB_BOOST_SUPPORT_SRC_BOOST_SUPPORT_CONNECTION_UDP_UDP_CONNECTION_H_
 
#include <functional>
 
#include "boost-support/error_domain/boost_support_error_domain.h"
#include "core/include/result.h"
 
namespace boost_support {
namespace connection {
namespace udp {
 
template<typename Socket>
class UdpConnection final {
 public:
  using UdpMessage = typename Socket::UdpMessage;
 
  using UdpMessagePtr = typename Socket::UdpMessagePtr;
 
  using UdpMessageConstPtr = typename Socket::UdpMessageConstPtr;
 
  using HandlerRead = std::function<void(UdpMessagePtr)>;
 
 public:
  explicit UdpConnection(Socket socket) noexcept : socket_{std::move(socket)} {}
 
  UdpConnection(const UdpConnection &other) noexcept = delete;
  UdpConnection &operator=(const UdpConnection &other) &noexcept = delete;
 
  UdpConnection(UdpConnection &&other) noexcept = default;
  UdpConnection &operator=(UdpConnection &&other) &noexcept = default;
 
  ~UdpConnection() noexcept = default;
 
  void SetReadHandler(HandlerRead read_handler) { socket_.SetReadHandler(std::move(read_handler)); }
 
  void Initialize() noexcept {
    // Open socket
    socket_.Open();
  }
 
  void DeInitialize() noexcept { socket_.Close(); }
 
  core_type::Result<void> Transmit(UdpMessageConstPtr message) noexcept {
    return socket_.Transmit(std::move(message))
        .AndThen([]() { return core_type::Result<void>::FromValue(); })
        .MapError([](typename Socket::SocketError const &) {
          return error_domain::MakeErrorCode(error_domain::BoostSupportErrorErrc::kSocketError);
        });
  }
 
 private:
  Socket socket_;
};
 
}  // namespace udp
}  // namespace connection
}  // namespace boost_support
#endif  // DIAG_CLIENT_LIB_LIB_BOOST_SUPPORT_SRC_BOOST_SUPPORT_CONNECTION_UDP_UDP_CONNECTION_H_