trie_storage_backend_impl.hpp

Go to the documentation of this file.

#ifndef KAGOME_STORAGE_TRIE_IMPL_TRIE_STORAGE_BACKEND
#define KAGOME_STORAGE_TRIE_IMPL_TRIE_STORAGE_BACKEND

#include "common/buffer.hpp"
#include "outcome/outcome.hpp"
#include "storage/trie/trie_storage_backend.hpp"

namespace kagome::storage::trie {

  class TrieStorageBackendImpl : public TrieStorageBackend {
   public:
    TrieStorageBackendImpl(std::shared_ptr<BufferStorage> storage,
                           common::Buffer node_prefix);

    ~TrieStorageBackendImpl() override = default;

    std::unique_ptr<Cursor> cursor() override;
    std::unique_ptr<face::WriteBatch<BufferView, Buffer>> batch() override;

    outcome::result<Buffer> load(const BufferView &key) const override;
    outcome::result<std::optional<Buffer>> tryLoad(
        const BufferView &key) const override;
    outcome::result<bool> contains(const BufferView &key) const override;
    bool empty() const 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 common::BufferView &key) override;

    size_t size() const override;

   private:
    common::Buffer prefixKey(const common::BufferView &key) const;

    std::shared_ptr<BufferStorage> storage_;
    common::Buffer node_prefix_;
  };

}  // namespace kagome::storage::trie

#endif  // KAGOME_STORAGE_TRIE_IMPL_TRIE_STORAGE_BACKEND