transfer_metadata.hpp

Go to the documentation of this file.

 
#ifndef SNBMODULES_INCLUDE_SNBMODULES_TRANSFER_METADATA_HPP_
#define SNBMODULES_INCLUDE_SNBMODULES_TRANSFER_METADATA_HPP_
 
#include "snbmodules/common/protocols_enum.hpp"
#include "snbmodules/common/status_enum.hpp"
#include "snbmodules/ip_format.hpp"
#include "snbmodules/metadata_abstract.hpp"
 
#include <chrono>
#include <ctime>
#include <filesystem>
#include <iostream>
#include <map>
#include <stdexcept>
#include <string>
#include <utility>
 
namespace dunedaq::snbmodules {
class TransferMetadata : public MetadataAbstract
{
 
public:
  static const std::string m_file_extension;
 
  virtual std::string export_to_string_partial(bool force_all);
 
  // Overriden methods
  std::string export_to_string() override { return export_to_string_partial(true); }
  void from_string(const std::string&) override;
 
  // Used to export the metadata to a json file
  void generate_metadata_file(std::filesystem::path dest) override;
  void load_metadata_from_meta_file(std::filesystem::path src) override;
 
  bool operator==(MetadataAbstract const& other) const override
  {
    auto o = dynamic_cast<const TransferMetadata&>(other);
    return m_file_path == o.m_file_path && m_src == o.m_src && m_dest == o.m_dest && m_group_id == o.m_group_id;
  }
  bool operator==(TransferMetadata const& o) const
  {
    return m_file_path == o.m_file_path && m_src == o.m_src && m_dest == o.m_dest && m_group_id == o.m_group_id;
  }
 
  bool operator<(MetadataAbstract const& other) const override
  {
    auto o = dynamic_cast<const TransferMetadata&>(other);
    return m_file_path.string().compare(o.m_file_path.string());
  }
 
  TransferMetadata(const std::filesystem::path& file_path,
                   uint64_t bytes_size,
                   const IPFormat& src,
                   const std::string& hash = "",
                   const IPFormat& dest = IPFormat(),
                   const std::string& group_id = "",
                   uint64_t bytes_transferred = 0,
                   status_type::e_status status = status_type::e_status::WAITING)
    : m_hash(hash)
    , m_bytes_size(bytes_size)
    , m_bytes_transferred(bytes_transferred)
    , m_status(status)
    , m_src(src)
    , m_dest(dest)
    , m_group_id(group_id)
    , m_modified_fields({ { "hash", true },
                          { "size", true },
                          { "bytes_transferred", true },
                          { "transmission_speed", true },
                          { "status", true },
                          { "magnet_link", true },
                          { "error_code", true },
                          { "start_time", true },
                          { "end_time", true },
                          { "duration", true } })
  {
    // remove all occurences of ./ in the file path
    std::string file_path_str = file_path.string();
    std::string x = "./";
 
    size_t pos = 0;
    while (true) {
      pos = file_path_str.find(x, pos);
      if (pos == std::string::npos) {
        break;
      }
 
      file_path_str.replace(pos, x.length(), "");
    }
 
    m_file_path = std::filesystem::absolute(file_path_str);
  }
 
  explicit TransferMetadata(const std::filesystem::path& src, bool is_path = true)
  {
    if (is_path) {
      load_metadata_from_meta_file(src);
    } else {
      from_string(src.string());
    }
  }
 
  virtual ~TransferMetadata() {}
 
  // Setters
  inline void set_file_path(const std::filesystem::path& file_path)
  {
    // remove all occurences of ./ in the file path
    std::string file_path_str = file_path.string();
    std::string x = "./";
 
    size_t pos = 0;
    while (true) {
      pos = file_path_str.find(x, pos);
      if (pos == std::string::npos) {
        break;
      }
 
      file_path_str.replace(pos, x.length(), "");
    }
    m_file_path = std::filesystem::absolute(file_path_str);
  }
  inline void set_group_id(std::string group_id) { m_group_id = std::move(group_id); }
  void set_src(const IPFormat& source)
  {
    // Check if the source is not equal to the dest
    if (!(source == m_dest)) {
      m_src = source;
    } else {
      throw std::invalid_argument("The source cannot be equal to the destination");
    }
  }
 
  void set_dest(const IPFormat& dest)
  {
    // Check if the dest is not equal to the source
    if (!(dest == m_src)) {
      m_dest = dest;
    } else {
      throw std::invalid_argument("The destination cannot be equal to the source");
    }
  }
 
  inline void set_hash(std::string hash)
  {
    m_hash = std::move(hash);
    m_modified_fields["hash"] = true;
  }
 
  inline void set_size(uint64_t size)
  {
    m_bytes_size = size;
    m_modified_fields["size"] = true;
  }
 
  inline void set_bytes_transferred(uint64_t bytes_transferred)
  {
    m_bytes_transferred = bytes_transferred;
    m_modified_fields["bytes_transferred"] = true;
  }
 
  void set_progress(int purcent)
  {
    if (purcent < 0 || purcent > 100) {
      throw std::invalid_argument("The progress must be between 0 and 100");
    }
    set_bytes_transferred((purcent * m_bytes_size) / 100);
  }
 
  void set_status(status_type::e_status status)
  {
    m_status = status;
    if (status == status_type::e_status::DOWNLOADING || status == status_type::e_status::UPLOADING) {
      m_start_time =
        std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::system_clock::now().time_since_epoch())
          .count();
    }
    if (status == status_type::e_status::FINISHED || status == status_type::e_status::ERROR) {
      m_end_time =
        std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::system_clock::now().time_since_epoch())
          .count();
      m_duration += m_end_time - m_start_time;
    }
    if (status == status_type::e_status::PAUSED) {
      m_duration +=
        std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::system_clock::now().time_since_epoch())
          .count() -
        m_start_time;
    }
 
    m_modified_fields["status"] = true;
  }
 
  inline void set_magnet_link(std::string magnet_link)
  {
    m_magnet_link = std::move(magnet_link);
    m_modified_fields["magnet_link"] = true;
  }
 
  inline void set_error_code(std::string error_code)
  {
    m_error_code = std::move(error_code);
    m_modified_fields["error_code"] = true;
  }
 
  inline void set_transmission_speed(int32_t transmission_speed)
  {
    m_transmission_speed = transmission_speed;
    m_modified_fields["transmission_speed"] = true;
  }
 
  inline void set_duration(int64_t duration)
  {
    m_duration = duration;
    m_modified_fields["duration"] = true;
  }
 
  inline void set_start_time(int64_t start_time)
  {
    m_start_time = start_time;
    m_modified_fields["start_time"] = true;
  }
 
  inline void set_end_time(int64_t end_time)
  {
    m_end_time = end_time;
    m_modified_fields["end_time"] = true;
  }
 
  // Getters
  inline std::filesystem::path get_file_path() const { return m_file_path; }
  inline std::string get_file_name() const { return m_file_path.filename().string(); }
  inline std::string get_hash() const { return m_hash; }
  inline IPFormat get_src() const { return m_src; }
  inline IPFormat get_dest() const { return m_dest; }
  inline uint64_t get_size() const { return m_bytes_size; }
  inline uint64_t get_bytes_transferred() const { return m_bytes_transferred; }
  inline status_type::e_status get_status() const { return m_status; }
  inline std::string get_magnet_link() const { return m_magnet_link; }
  inline std::string get_group_id() const { return m_group_id; }
  inline int get_progress() const
  {
    return m_bytes_size == 0 ? 0 : static_cast<int>(m_bytes_transferred * 100 / m_bytes_size);
  }
  inline std::string get_error_code() const { return m_error_code; }
  inline int32_t get_transmission_speed() const { return m_transmission_speed; }
  int64_t get_total_duration_ms() const
  {
    if (m_start_time == 0) {
      return 0;
    } else if (m_end_time == 0) {
      return m_duration +
             (std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::system_clock::now().time_since_epoch())
                .count() -
              m_start_time);
    } else {
      return m_end_time - m_start_time;
    }
  }
  inline int64_t get_start_time() const { return m_start_time; }
  inline int64_t get_end_time() const { return m_end_time; }
  std::string get_start_time_str() const
  {
    if (m_start_time == 0) {
      return "N/A";
    }
    std::time_t t = m_start_time / 1000;
    std::tm tm = *std::localtime(&t);
    std::stringstream ss;
    ss << std::put_time(&tm, "%d-%m-%Y %H:%M:%S");
    return ss.str();
  }
  std::string get_end_time_str() const
  {
    if (m_end_time == 0) {
      return "N/A";
    }
    std::time_t t = m_end_time / 1000;
    std::tm tm = *std::localtime(&t);
    std::stringstream ss;
    ss << std::put_time(&tm, "%d-%m-%Y %H:%M:%S");
    return ss.str();
  }
 
private:
  std::filesystem::path m_file_path = "";
 
  std::string m_hash = "";
 
  uint64_t m_bytes_size = 0;
 
  uint64_t m_bytes_transferred = 0;
 
  int32_t m_transmission_speed = 0;
 
  status_type::e_status m_status = status_type::e_status::WAITING;
 
  IPFormat m_src = IPFormat();
 
  IPFormat m_dest = IPFormat();
 
  std::string m_magnet_link = "";
 
  std::string m_group_id = "";
 
  std::string m_error_code = "";
 
  int64_t m_start_time = 0;
 
  int64_t m_end_time = 0;
 
  int64_t m_duration = 0;
 
  std::map<std::string, bool> m_modified_fields = {
    { "file_path", false },         { "hash", false },       { "bytes_size", false },
    { "bytes_transferred", false }, { "status", false },     { "magnet_link", false },
    { "group_id", false },          { "error_code", false }, { "transmission_speed", false },
    { "start_time", false },        { "end_time", false },   { "duration", false }
  };
};
 
} // namespace dunedaq::snbmodules
#endif // SNBMODULES_INCLUDE_SNBMODULES_TRANSFER_METADATA_HPP_