Create and implement socket class

CMakeLists.txt

@@ -51,6 +51,7 @@ add_executable(blah2
   src/process/tracker/Tracker.cpp
   src/process/spectrum/SpectrumAnalyser.cpp
   src/process/meta/HammingNumber.cpp
+  src/process/meta/Socket.cpp
   src/data/IqData.cpp
   src/data/Map.cpp
   src/data/Detection.cpp

src/blah2.cpp

@@ -15,6 +15,7 @@
 #include "process/detection/Interpolate.h"
 #include "process/spectrum/SpectrumAnalyser.h"
 #include "process/tracker/Tracker.h"
+#include "process/meta/Socket.h"
 #include "data/meta/Constants.h"
 
 #include <ryml/ryml.hpp>
@@ -118,40 +119,12 @@ int main(int argc, char **argv)
   tree["network"]["ports"]["timing"] >> port_timing;
   tree["network"]["ports"]["iqdata"] >> port_iqdata;
   tree["network"]["ip"] >> ip;
-  asio::io_service io_service;
-  asio::ip::tcp::socket socket_map(io_service);
-  asio::ip::tcp::socket socket_detection(io_service);
-  asio::ip::tcp::socket socket_track(io_service);
-  asio::ip::tcp::socket socket_timestamp(io_service);
-  asio::ip::tcp::socket socket_timing(io_service);
-  asio::ip::tcp::socket socket_iqdata(io_service);
-  asio::ip::tcp::endpoint endpoint_map;
-  asio::ip::tcp::endpoint endpoint_detection;
-  asio::ip::tcp::endpoint endpoint_track;
-  asio::ip::tcp::endpoint endpoint_timestamp;
-  asio::ip::tcp::endpoint endpoint_timing;
-  asio::ip::tcp::endpoint endpoint_iqdata;
-  endpoint_map = asio::ip::tcp::endpoint(
-    asio::ip::address::from_string(ip), port_map);
-  endpoint_detection = asio::ip::tcp::endpoint(
-    asio::ip::address::from_string(ip), port_detection);
-  endpoint_track = asio::ip::tcp::endpoint(
-    asio::ip::address::from_string(ip), port_track);
-  endpoint_timestamp = asio::ip::tcp::endpoint(
-    asio::ip::address::from_string(ip), port_timestamp);
-  endpoint_timing = asio::ip::tcp::endpoint(
-    asio::ip::address::from_string(ip), port_timing);
-  endpoint_iqdata = asio::ip::tcp::endpoint(
-    asio::ip::address::from_string(ip), port_iqdata);
-  socket_map.connect(endpoint_map);
-  socket_detection.connect(endpoint_detection);
-  socket_track.connect(endpoint_track);
-  socket_timestamp.connect(endpoint_timestamp);
-  socket_timing.connect(endpoint_timing);
-  socket_iqdata.connect(endpoint_iqdata);
-  asio::error_code err;
-  std::string subdata;
-  uint32_t MTU = 1024;
+  Socket socket_map(ip, port_map);
+  Socket socket_detection(ip, port_detection);
+  Socket socket_track(ip, port_track);
+  Socket socket_timestamp(ip, port_timestamp);
+  Socket socket_timing(ip, port_timing);
+  Socket socket_iqdata(ip, port_iqdata);
 
   // setup process ambiguity
   int32_t delayMin, delayMax;
@@ -278,11 +251,7 @@ int main(int argc, char **argv)
 
           // output IqData meta data
           jsonIqData = x->to_json(time[0]/1000);
-          for (int i = 0; i < (jsonIqData.size() + MTU - 1) / MTU; i++)
-          {
-            subdata = jsonIqData.substr(i * MTU, MTU);
-            socket_iqdata.write_some(asio::buffer(subdata, subdata.size()), err);
-          }
+          socket_iqdata.sendData(jsonIqData);
 
           // output map data
           mapJson = map->to_json(time[0]/1000);
@@ -291,30 +260,19 @@ int main(int argc, char **argv)
           {
             map->save(mapJson, saveMapPath);
           }
-          for (int i = 0; i < (mapJson.size() + MTU - 1) / MTU; i++)
-          {
-            subdata = mapJson.substr(i * MTU, MTU);
-            socket_map.write_some(asio::buffer(subdata, subdata.size()), err);
-          }
+          socket_map.sendData(mapJson);
+
           // output detection data
           detectionJson = detection->to_json(time[0]/1000);
           detectionJson = detection->delay_bin_to_km(detectionJson, fs);
-          for (int i = 0; i < (detectionJson.size() + MTU - 1) / MTU; i++)
-          {
-            subdata = detectionJson.substr(i * MTU, MTU);
-            socket_detection.write_some(asio::buffer(subdata, subdata.size()), err);
-          }
+          socket_detection.sendData(detectionJson);
           delete detection;
           delete detection1;
           delete detection2;
 
           // output tracker data
           jsonTracker = track->to_json(time[0]/1000);
-          for (int i = 0; i < (jsonTracker.size() + MTU - 1) / MTU; i++)
-          {
-            subdata = jsonTracker.substr(i * MTU, MTU);
-            socket_track.write_some(asio::buffer(subdata, subdata.size()), err);
-          }
+          socket_track.sendData(jsonTracker);
 
           // output radar data timer
           timing_helper(timing_name, timing_time, time, "output_radar_data");
@@ -329,13 +287,13 @@ int main(int argc, char **argv)
           // output timing data
           timing->update(time[0]/1000, timing_time, timing_name);
           jsonTiming = timing->to_json();
-          socket_timing.write_some(asio::buffer(jsonTiming, 1500), err);
+          socket_timing.sendData(jsonTiming);
           timing_time.clear();
           timing_name.clear();
 
           // output CPI timestamp for updating data
           std::string t0_string = std::to_string(time[0]/1000);
-          socket_timestamp.write_some(asio::buffer(t0_string, 100), err);
+          socket_timestamp.sendData(t0_string);
           time.clear();
         }
       }

src/process/meta/Socket.cpp

@@ -0,0 +1,27 @@
+#include "Socket.h"
+#include <iostream>
+
+asio::io_context Socket::io_context;
+const uint32_t Socket::MTU = 1024;
+
+Socket::Socket(const std::string& ip, uint16_t port)
+    : endpoint(asio::ip::address::from_string(ip), port), socket(io_context) {
+    socket.connect(endpoint);
+}
+
+Socket::~Socket()
+{
+}
+
+void Socket::sendData(const std::string& data) {
+    asio::error_code err;
+
+    for (std::size_t i = 0; i < (data.size() + MTU - 1) / MTU; ++i) {
+        std::string subdata = data.substr(i * MTU, MTU);
+        socket.write_some(asio::buffer(subdata, subdata.size()), err);
+
+        if (err) {
+            std::cerr << "Error sending data: " << err.message() << std::endl;
+        }
+    }
+}

src/process/meta/Socket.h

@@ -0,0 +1,47 @@
+/// @file Socket.h
+/// @class Socket
+/// @brief A class to implement network socket functionality.
+/// @details Used to pass radar data from app to the API.
+/// @author 30hours
+
+#ifndef SOCKET_H
+#define SOCKET_H
+
+#include <string>
+#include <cstdint>
+#include <asio.hpp>
+
+class Socket 
+{
+private:
+    /// @brief Common io_context for all socket objects.
+    static asio::io_context io_context;
+
+    /// @brief Common MTU size for all socket objects.
+    static const uint32_t MTU;
+
+    /// @brief The ASIO socket.
+    asio::ip::tcp::socket socket;
+
+    /// @brief The ASIO endpoint.
+    asio::ip::tcp::endpoint endpoint;
+
+public:
+    /// @brief Constructor for Socket.
+    /// @param ip IP address of data destination.
+    /// @param port Port of data destination.
+    /// @return The object.
+    Socket(const std::string& ip, uint16_t port);
+
+    /// @brief Destructor.
+    /// @return Void.
+    ~Socket();
+
+    /// @brief Helper function to send data in chunks.
+    /// @param data String of complete data to send.
+    /// @return Void.
+    void sendData(const std::string& data);
+
+};
+
+#endif