Repo created

libs/tracking/connection.cpp

@@ -0,0 +1,68 @@
+#include "tracking/connection.hpp"
+
+#include "tracking/protocol.hpp"
+
+#include "platform/platform.hpp"
+#include "platform/socket.hpp"
+
+namespace
+{
+uint32_t constexpr kSocketTimeoutMs = 10000;
+}  // namespace
+
+namespace tracking
+{
+Connection::Connection(std::unique_ptr<platform::Socket> socket, std::string const & host, uint16_t port,
+                       bool isHistorical)
+  : m_socket(std::move(socket))
+  , m_host(host)
+  , m_port(port)
+{
+  if (!m_socket)
+    return;
+
+  m_socket->SetTimeout(kSocketTimeoutMs);
+}
+
+// TODO: implement handshake
+bool Connection::Reconnect()
+{
+  if (!m_socket)
+    return false;
+
+  m_socket->Close();
+
+  if (!m_socket->Open(m_host, m_port))
+    return false;
+
+  // TODO: generate unique client id here.
+  std::string clientId = "TODO";
+  auto packet = Protocol::CreateAuthPacket(clientId);
+  if (!m_socket->Write(packet.data(), static_cast<uint32_t>(packet.size())))
+    return false;
+
+  std::string check(std::begin(Protocol::kFail), std::end(Protocol::kFail));
+  bool const isSuccess = m_socket->Read(reinterpret_cast<uint8_t *>(&check[0]), static_cast<uint32_t>(check.size()));
+  if (!isSuccess || check != std::string(std::begin(Protocol::kOk), std::end(Protocol::kOk)))
+    return false;
+
+  return true;
+}
+
+void Connection::Shutdown()
+{
+  if (!m_socket)
+    return;
+
+  m_socket->Close();
+}
+
+bool Connection::Send(boost::circular_buffer<DataPoint> const & points)
+{
+  if (!m_socket)
+    return false;
+
+  auto packet = Protocol::CreateDataPacket(points, tracking::Protocol::PacketType::CurrentData);
+  return m_socket->Write(packet.data(), static_cast<uint32_t>(packet.size()));
+}
+}  // namespace tracking