trie_serializer_impl.hpp

Go to the documentation of this file.

#ifndef KAGOME_STORAGE_TRIE_SERIALIZER_IMPL
#define KAGOME_STORAGE_TRIE_SERIALIZER_IMPL

#include "storage/trie/serialization/trie_serializer.hpp"

#include "storage/buffer_map_types.hpp"

namespace kagome::storage::trie {
  class Codec;
  class PolkadotTrieFactory;
  class TrieStorageBackend;
  struct BranchNode;
  struct TrieNode;
}  // namespace kagome::storage::trie

namespace kagome::storage::trie {

  class TrieSerializerImpl : public TrieSerializer {
   public:
    TrieSerializerImpl(std::shared_ptr<PolkadotTrieFactory> factory,
                       std::shared_ptr<Codec> codec,
                       std::shared_ptr<TrieStorageBackend> backend);
    ~TrieSerializerImpl() override = default;

    RootHash getEmptyRootHash() const override;

    outcome::result<RootHash> storeTrie(PolkadotTrie &trie) override;

    outcome::result<std::shared_ptr<PolkadotTrie>> retrieveTrie(
        const common::Buffer &db_key) const override;

   private:
    outcome::result<RootHash> storeRootNode(TrieNode &node);
    outcome::result<PolkadotTrie::NodePtr> retrieveNode(
        const common::Buffer &db_key) const;
    outcome::result<PolkadotTrie::NodePtr> retrieveNode(
        const std::shared_ptr<OpaqueTrieNode> &node) const;

    std::shared_ptr<PolkadotTrieFactory> trie_factory_;
    std::shared_ptr<Codec> codec_;
    std::shared_ptr<TrieStorageBackend> backend_;
  };
}  // namespace kagome::storage::trie

#endif  // KAGOME_STORAGE_TRIE_SERIALIZER_IMPL