block_storage_impl.hpp

Go to the documentation of this file.

#ifndef KAGOME_BLOCKCHAIN_BLOCKSTORAGEIMPL
#define KAGOME_BLOCKCHAIN_BLOCKSTORAGEIMPL

#include "blockchain/block_storage.hpp"

#include "blockchain/impl/common.hpp"
#include "crypto/hasher.hpp"
#include "log/logger.hpp"
#include "storage/predefined_keys.hpp"

namespace kagome::blockchain {

  class BlockStorageImpl : public BlockStorage {
   public:
    ~BlockStorageImpl() override = default;

    static outcome::result<std::shared_ptr<BlockStorageImpl>> create(
        storage::trie::RootHash state_root,
        const std::shared_ptr<storage::BufferStorage> &storage,
        const std::shared_ptr<crypto::Hasher> &hasher);

    outcome::result<std::vector<primitives::BlockHash>> getBlockTreeLeaves()
        const override;
    outcome::result<void> setBlockTreeLeaves(
        std::vector<primitives::BlockHash> leaves) override;

    outcome::result<bool> hasBlockHeader(
        const primitives::BlockId &id) const override;

    outcome::result<std::optional<primitives::BlockHeader>> getBlockHeader(
        const primitives::BlockId &id) const override;
    outcome::result<std::optional<primitives::BlockBody>> getBlockBody(
        const primitives::BlockId &id) const override;
    outcome::result<std::optional<primitives::BlockData>> getBlockData(
        const primitives::BlockId &id) const override;
    outcome::result<std::optional<primitives::Justification>> getJustification(
        const primitives::BlockId &block) const override;

    outcome::result<void> putNumberToIndexKey(
        const primitives::BlockInfo &block) override;

    outcome::result<primitives::BlockHash> putBlockHeader(
        const primitives::BlockHeader &header) override;
    outcome::result<void> putBlockData(
        primitives::BlockNumber block_number,
        const primitives::BlockData &block_data) override;
    outcome::result<void> removeBlockData(
        primitives::BlockNumber block_number,
        const primitives::BlockDataFlags &remove_flags) override;
    outcome::result<primitives::BlockHash> putBlock(
        const primitives::Block &block) override;

    outcome::result<void> putJustification(
        const primitives::Justification &j,
        const primitives::BlockHash &hash,
        primitives::BlockNumber number) override;
    outcome::result<void> removeJustification(
        const primitives::BlockHash &hash,
        primitives::BlockNumber number) override;

    outcome::result<void> removeBlock(
        const primitives::BlockInfo &block) override;

    outcome::result<primitives::BlockInfo> getLastFinalized() const override;

   private:
    BlockStorageImpl(std::shared_ptr<storage::BufferStorage> storage,
                     std::shared_ptr<crypto::Hasher> hasher);

    std::shared_ptr<storage::BufferStorage> storage_;
    std::shared_ptr<crypto::Hasher> hasher_;
    log::Logger logger_;

    mutable std::optional<std::vector<primitives::BlockHash>>
        block_tree_leaves_;
  };
}  // namespace kagome::blockchain

#endif  // KAGOME_BLOCKCHAIN_BLOCKSTORAGEIMPL