any_message_log_writer.h

Go to the documentation of this file.

// Copyright (c) Space-ng, inc. All rights reserved.
 
#pragma once
 
#include "sol3/core/message_port.h"
#include "sol3/core/message_schema.h"
#include "sol3/core/msg/message_log_config.h"
#include "sol3/core/msg/message_log_header.h"
#include "sol3/core/size_delimited_log_writer.h"
#include "sol3/core/stream_writer.h"
 
#include <flatbuffers/flatbuffers.h>
 
#include <optional>
#include <string>
namespace sol3::core {
class AnyMessageLogWriter : public IStreamWriter {
 public:
  explicit AnyMessageLogWriter(
      msg::MessageLogConfigT const& cfg,
      MessageSchema const& schema,
      std::chrono::system_clock::time_point start_stamp);
 
  ~AnyMessageLogWriter();
 
  void logMessage(
      flatbuffers::span<uint8_t const> header_span,
      flatbuffers::span<uint8_t const> payload_span);
 
  void commit(size_t message_count);
 
  void rollback();
 
  void write(char const* data, size_t size) override;
 
  cpp::fs::path closeLog(std::chrono::system_clock::time_point end_stamp);
 
  cpp::fs::path rotateLog(std::chrono::system_clock::time_point stamp);
 
  cpp::fs::path const& currentLogPath() const { return log_path_; }
 
 private:
  void writeFileHeader(std::chrono::system_clock::time_point start_stamp);
  void closeAndUpdateFileHeader(
      std::chrono::system_clock::time_point end_stamp);
 
  msg::MessageLogConfigT log_config_;
  MessagePort port_;
  std::string host_id_;
  std::string run_id_;
  cpp::fs::path log_dir_;
  cpp::fs::path log_path_;
  std::chrono::system_clock::time_point log_start_stamp_;
  std::optional<size_t> log_header_offset_;
  uint64_t message_count_ = 0;
  std::optional<SizeDelimitedLogWriter> writer_;
 
  MessageSchema payload_schema_;
  msg::MessageLogHeaderT log_header_;
  flatbuffers::FlatBufferBuilder log_header_builder_;
};
}  // namespace sol3::core