FileBlockStorage.h

Go to the documentation of this file.

 
#pragma once
#include "BlockStorage.h"
#include "IndexFile.h"
#include "RawFile.h"
#include <string>
 
namespace catapult { namespace io {
 
    enum class FileBlockStorageMode {
        Hash_Index,
 
        None
    };
 
    class FileBlockStorage final : public PrunableBlockStorage {
    public:
        explicit FileBlockStorage(const std::string& dataDirectory, FileBlockStorageMode mode = FileBlockStorageMode::Hash_Index);
 
    public:
        // LightBlockStorage
        Height chainHeight() const override;
        model::HashRange loadHashesFrom(Height height, size_t maxHashes) const override;
        void saveBlock(const model::BlockElement& blockElement) override;
        void dropBlocksAfter(Height height) override;
 
        // BlockStorage
        std::shared_ptr<const model::Block> loadBlock(Height height) const override;
        std::shared_ptr<const model::BlockElement> loadBlockElement(Height height) const override;
        std::pair<std::vector<uint8_t>, bool> loadBlockStatementData(Height height) const override;
 
        // PrunableBlockStorage
        void purge() override;
 
    private:
        void requireHeight(Height height, const char* description) const;
 
    private:
        class HashFile final {
        public:
            explicit HashFile(const std::string& dataDirectory);
 
            model::HashRange loadHashesFrom(Height height, size_t numHashes) const;
            void save(Height height, const Hash256& hash);
            void reset();
 
        private:
            const std::string& m_dataDirectory;
 
            // used for caching inside save()
            uint64_t m_cachedDirectoryId;
            std::unique_ptr<RawFile> m_pCachedHashFile;
        };
 
        std::string m_dataDirectory;
        FileBlockStorageMode m_mode;
 
        HashFile m_hashFile;
        IndexFile m_indexFile;
    };
}}