buffer_stream.hpp

Go to the documentation of this file.

#ifndef KAGOME_CORE_STORAGE_TRIE_BUFFER_STREAM
#define KAGOME_CORE_STORAGE_TRIE_BUFFER_STREAM

#include <gsl/span>

#include "common/buffer.hpp"

namespace kagome::storage::trie {

  class BufferStream {
    using index_type = gsl::span<const uint8_t>::index_type;

   public:
    explicit BufferStream(gsl::span<const uint8_t> buf) : data_{buf} {}

    bool hasMore(index_type num_bytes) const {
      return data_.size() >= num_bytes;
    }

    uint8_t next() {
      auto byte = data_.at(0);
      data_ = data_.last(data_.size() - 1);
      return byte;
    }

    gsl::span<const uint8_t> leftBytes() const {
      return data_;
    }

   private:
    gsl::span<const uint8_t> data_;
  };
}  // namespace kagome::storage::trie

#endif  // KAGOME_CORE_STORAGE_TRIE_BUFFER_STREAM