any_message_output.h

Go to the documentation of this file.

// Copyright (c) Space-ng, inc. All rights reserved.
#pragma once
 
#include "sol3/core/message_schema.h"
#include "sol3/core/msg/buffer.h"
#include "sol3/core/seq_ring.h"
#include "sol3/core/shmem_buffer.h"
#include "sol3/log/except.h"
 
namespace sol3::core {
 
class AnyMessageOutput {
 public:
  AnyMessageOutput(
      IBufferExchange& exchange,
      MessagePort port,
      MessageSchema schema,
      size_t slot_count,
      size_t slot_size);
 
  ~AnyMessageOutput();
 
  AnyMessageOutput(AnyMessageOutput&& rhs) = delete;
  AnyMessageOutput& operator=(AnyMessageOutput&& rhs) = delete;
 
  AnyMessageOutput(AnyMessageOutput const&) = delete;
  AnyMessageOutput& operator=(AnyMessageOutput const& rhs) = delete;
 
  MessagePort const& port() const { return port_; }
 
  flatbuffers::span<uint8_t const> headerSpan() const;
 
  flatbuffers::span<uint8_t const> payloadSpan() const;
 
  uint64_t write(
      flatbuffers::span<uint8_t const> header_span,
      flatbuffers::span<uint8_t const> payload_span);
 
  msg::MessageHeader const* headerRoot() const;
 
  template <typename TTableType>
  TTableType const* payloadRoot() const {
    if (payloadSpan().empty()) {
      return nullptr;
    }
    return flatbuffers::GetSizePrefixedRoot<TTableType>(payloadSpan().data());
  }
 
  MessageSchema schema() const { return schema_; }
 
 private:
  uint64_t dataSize() const {
    return SeqRingWriter::minDataSize(writer_.slotCount(), writer_.slotSize());
  }
  uint64_t trailerSize() { return output_buffer_->size() - dataSize(); }
  uint8_t* trailerData();
 
  IBufferExchange& exchange_;
  MessagePort port_;
  MessageSchema schema_;
  std::unique_ptr<IBufferMutable> output_buffer_;
  SeqRingWriter writer_;
  uint8_t const* last_header_data_ = nullptr;
  uint8_t const* last_payload_data_ = nullptr;
  size_t last_header_size_ = 0;
  size_t last_payload_size_ = 0;
};
 
}  // namespace sol3::core