persistent_trie_batch_impl.hpp

Go to the documentation of this file.

#ifndef KAGOME_STORAGE_TRIE_IMPL_PERSISTENT_TRIE_BATCH
#define KAGOME_STORAGE_TRIE_IMPL_PERSISTENT_TRIE_BATCH

#include <memory>

#include "log/logger.hpp"
#include "primitives/event_types.hpp"
#include "storage/changes_trie/changes_tracker.hpp"
#include "storage/trie/codec.hpp"
#include "storage/trie/serialization/trie_serializer.hpp"
#include "storage/trie/trie_batches.hpp"

namespace kagome::storage::trie {

  class PersistentTrieBatchImpl final : public PersistentTrieBatch {
   public:
    enum class Error {
      NO_TRIE = 1,
    };

    static std::unique_ptr<PersistentTrieBatchImpl> create(
        std::shared_ptr<Codec> codec,
        std::shared_ptr<TrieSerializer> serializer,
        std::optional<std::shared_ptr<changes_trie::ChangesTracker>> changes,
        std::shared_ptr<PolkadotTrie> trie);
    ~PersistentTrieBatchImpl() override = default;

    outcome::result<RootHash> commit() override;
    std::unique_ptr<TopperTrieBatch> batchOnTop() override;

    outcome::result<BufferConstRef> get(const BufferView &key) const override;
    outcome::result<std::optional<BufferConstRef>> tryGet(
        const BufferView &key) const override;
    std::unique_ptr<PolkadotTrieCursor> trieCursor() override;
    outcome::result<bool> contains(const BufferView &key) const override;
    bool empty() const override;
    outcome::result<std::tuple<bool, uint32_t>> clearPrefix(
        const BufferView &prefix,
        std::optional<uint64_t> limit = std::nullopt) override;
    outcome::result<void> put(const BufferView &key,
                              const Buffer &value) override;
    outcome::result<void> put(const BufferView &key, Buffer &&value) override;
    outcome::result<void> remove(const BufferView &key) override;

   private:
    PersistentTrieBatchImpl(
        std::shared_ptr<Codec> codec,
        std::shared_ptr<TrieSerializer> serializer,
        std::optional<std::shared_ptr<changes_trie::ChangesTracker>> changes,
        std::shared_ptr<PolkadotTrie> trie);

    std::shared_ptr<Codec> codec_;
    std::shared_ptr<TrieSerializer> serializer_;
    std::optional<std::shared_ptr<changes_trie::ChangesTracker>> changes_;
    std::shared_ptr<PolkadotTrie> trie_;

    log::Logger logger_ = log::createLogger("PersistentTrieBatch", "storage");
  };

}  // namespace kagome::storage::trie

OUTCOME_HPP_DECLARE_ERROR(kagome::storage::trie,
                          PersistentTrieBatchImpl::Error);

#endif  // KAGOME_STORAGE_TRIE_IMPL_PERSISTENT_TRIE_BATCH