trie_storage_backend_batch.hpp

Go to the documentation of this file.

#ifndef KAGOME_STORAGE_TRIE_IMPL_TRIE_STORAGE_BACKEND_BATCH
#define KAGOME_STORAGE_TRIE_IMPL_TRIE_STORAGE_BACKEND_BATCH

#include "common/buffer.hpp"
#include "storage/face/write_batch.hpp"

namespace kagome::storage::trie {

  class TrieStorageBackendBatch
      : public face::WriteBatch<common::BufferView, common::Buffer> {
   public:
    TrieStorageBackendBatch(
        std::unique_ptr<face::WriteBatch<common::BufferView, common::Buffer>>
            storage_batch,
        common::Buffer node_prefix);
    ~TrieStorageBackendBatch() override = default;

    outcome::result<void> commit() override;

    outcome::result<void> put(const common::BufferView &key,
                              const common::Buffer &value) override;

    outcome::result<void> put(const common::BufferView &key,
                              common::Buffer &&value) override;

    outcome::result<void> remove(const common::BufferView &key) override;
    void clear() override;

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

    std::unique_ptr<face::WriteBatch<common::BufferView, common::Buffer>>
        storage_batch_;
    common::Buffer node_prefix_;
  };

}  // namespace kagome::storage::trie

#endif  // KAGOME_STORAGE_TRIE_IMPL_TRIE_STORAGE_BACKEND_BATCH