githubmirror/omim

tracking/connection.cpp
68 строк2 KB
Olga Khlopkova
[map][tracking] Archieve tracker - bicycles, pedestrians, cars.
16 дек 2019, 12:26
16 дек 2019, 12:2615ff0c2
#include "tracking/connection.hpp"

#include "tracking/protocol.hpp"

#include "platform/platform.hpp"
#include "platform/socket.hpp"

#include "3party/Alohalytics/src/alohalytics.h"

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(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;

  alohalytics::Stats::Instance().LogEvent("TrafficTrack_reconnect");
  m_socket->Close();

  if (!m_socket->Open(m_host, m_port))
    return false;

  auto packet = Protocol::CreateAuthPacket(GetPlatform().UniqueClientId());
  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