PatriciaTreeRdbDataSource.h

Go to the documentation of this file.

 
#pragma once
#include "PatriciaTreeContainer.h"
#include "catapult/types.h"
 
namespace catapult { namespace cache {
 
    class PatriciaTreeRdbDataSource {
    public:
        explicit PatriciaTreeRdbDataSource(PatriciaTreeContainer& container) : m_container(container)
        {}
 
    public:
        size_t size() {
            return m_container.size();
        }
 
        std::unique_ptr<const tree::TreeNode> get(const Hash256& hash) const {
            auto iter = m_container.find(hash);
            if (m_container.cend() == iter)
                return nullptr;
 
            const auto& pair = *iter;
            return std::make_unique<const tree::TreeNode>(pair.second.copy());
        }
 
    public:
        void set(const tree::LeafTreeNode& node) {
            set(tree::TreeNode(node));
        }
 
        void set(const tree::BranchTreeNode& node) {
            set(tree::TreeNode(node));
        }
 
    private:
        void set(const tree::TreeNode& node) {
            m_container.insert(std::make_pair(node.hash(), node.copy()));
        }
 
    private:
        PatriciaTreeContainer& m_container;
    };
}}