any_message_input.h

Go to the documentation of this file.

// Copyright (c) Space-ng, inc. All rights reserved.
#pragma once
#include "sol3/core/any_message_view.h"
#include "sol3/core/message_port.h"
#include "sol3/core/shmem_buffer.h"
#include "sol3/core/stream_writer.h"
 
#include <flatbuffers/flatbuffers.h>
#include <spdlog/spdlog.h>
 
#include <chrono>
#include <memory>
#include <optional>
 
namespace sol3::core {
 
class AnyMessageInput {
 public:
  explicit AnyMessageInput(
      IBufferExchange& exchange,
      IBufferExchange::Endpoint endpoint,
      MessagePort port,
      MessageSchema expected_schema = MessageSchema());
 
  explicit AnyMessageInput(
      IBufferExchange& exchange,
      MessagePort port,
      MessageSchema expected_schema);
 
  AnyMessageInput(AnyMessageInput&& rhs) = delete;
  AnyMessageInput& operator=(AnyMessageInput&& rhs) = delete;
  AnyMessageInput(AnyMessageInput const&) = delete;
  AnyMessageInput& operator=(AnyMessageInput const& rhs) = delete;
  ~AnyMessageInput() = default;
 
  bool pollForBufferUpdate();
 
  size_t slotCount() const;
 
  size_t slotSize() const;
 
  void reset();
 
  AnyMessageView read(
      uint64_t seq, std::chrono::steady_clock::duration time_out = {});
 
  size_t readInto(
      uint64_t seq,
      flatbuffers::span<uint8_t> dest,
      std::chrono::steady_clock::duration time_out = {});
 
  size_t readInto(
      uint64_t seq,
      IStreamWriter& stream,
      std::chrono::steady_clock::duration time_out = {});
 
  AnyMessageView readHead(std::chrono::steady_clock::duration time_out = {});
 
  uint64_t head() const;
 
  MessagePort port() const;
 
  IBufferExchange::Endpoint const& endpoint() const;
 
  MessageSchema const& schema() const;
 
  void updateSchema(MessageSchema const& schema);
 
 private:
  std::shared_ptr<IBufferConst> swapBuffer(
      std::shared_ptr<IBufferConst> buffer);
 
  IBufferExchange& exchange_;
 
  std::optional<IBufferExchange::Endpoint> endpoint_;
  MessagePort port_;
 
  SeqRingReader reader_;
  std::vector<AnyMessageView> messages_;
 
  std::shared_ptr<IBufferConst> buffer_;
 
  MessageSchema schema_;
 
  uint64_t uuid_ = 0;
};
 
}  // namespace sol3::core