ReadThroughMemoryDataSource.h

Go to the documentation of this file.

 
#pragma once
#include "MemoryDataSource.h"
 
namespace catapult { namespace tree {
 
    template<typename TBackingDataSource>
    class ReadThroughMemoryDataSource {
    public:
        explicit ReadThroughMemoryDataSource(
                const TBackingDataSource& backingDataSource,
                DataSourceVerbosity verbosity = DataSourceVerbosity::Off)
                : m_backingDataSource(backingDataSource)
                , m_memoryDataSource(verbosity)
        {}
 
    public:
        size_t size() const {
            return m_memoryDataSource.size();
        }
 
    public:
        std::unique_ptr<const TreeNode> get(const Hash256& hash) const {
            auto pNode = m_memoryDataSource.get(hash);
            return pNode ? std::move(pNode) : m_backingDataSource.get(hash);
        }
 
        void forEach(const consumer<const TreeNode&>& consumer) const {
            m_memoryDataSource.forEach(consumer);
        }
 
    public:
        void set(const LeafTreeNode& node) {
            m_memoryDataSource.set(node);
        }
 
        void set(const BranchTreeNode& node) {
            m_memoryDataSource.set(node);
        }
 
        void clear() {
            m_memoryDataSource.clear();
        }
 
    private:
        const TBackingDataSource& m_backingDataSource;
        MemoryDataSource m_memoryDataSource;
    };
}}