UdpTransport.hpp

#pragma once
 
#include <array>
#include <asio.hpp>
#include <cstddef>
#include <cstdint>
#include <functional>
#include <vector>
 
#include "../config/NetworkConfig.hpp"
#include "../metrics/Metrics.hpp"
 
class UdpTransport {
public:
  UdpTransport(asio::io_context &context, std::uint16_t port,
               NetworkMetrics &metrics);
  ~UdpTransport() = default;
 
  void startReceive();
 
  void send(const asio::ip::udp::endpoint &destination,
            const std::vector<std::uint8_t> &data);
 
  void close();
 
  bool isOpen() const;
 
  void
  setReceiveCallback(std::function<void(const asio::ip::udp::endpoint &Endpoint,
                                        const std::vector<std::uint8_t> &)>
                         callback);
 
private:
  static constexpr std::size_t BUFFER_SIZE = 2048;
 
  asio::ip::udp::socket _socket;
  asio::ip::udp::endpoint _endpoint;
  NetworkMetrics &_metrics;
 
  std::array<std::uint8_t, BUFFER_SIZE> _receiveBuffer;
  std::function<void(const asio::ip::udp::endpoint &Endpoint,
                     const std::vector<std::uint8_t> &)>
      _receiveCallback;
 
  void execClose();
};