PatriciaTreeCacheMixins.h

Go to the documentation of this file.

 
#pragma once
#include "PatriciaTreeUtils.h"
#include "catapult/utils/HexFormatter.h"
#include "catapult/exceptions.h"
 
namespace catapult { namespace cache {
 
    template<typename TTree>
    class PatriciaTreeMixin {
    public:
        explicit PatriciaTreeMixin(const TTree* pTree) : m_pTree(pTree)
        {}
 
    public:
        bool supportsMerkleRoot() const {
            return !!m_pTree;
        }
 
        std::pair<Hash256, bool> tryGetMerkleRoot() const {
            return m_pTree
                    ? std::make_pair(m_pTree->root(), true)
                    : std::make_pair(Hash256(), false);
        }
 
        std::pair<Hash256, bool> tryLookup(const typename TTree::KeyType& key, std::vector<tree::TreeNode>& nodePath) const {
            return m_pTree
                    ? m_pTree->lookup(key, nodePath)
                    : std::make_pair(Hash256(), false);
        }
 
    private:
        const TTree* m_pTree;
    };
 
    template<typename TSet, typename TTree>
    class PatriciaTreeDeltaMixin {
    public:
        PatriciaTreeDeltaMixin(TSet& set, const std::shared_ptr<TTree>& pTree)
                : m_set(set)
                , m_pTree(pTree)
                , m_nextGenerationId(m_set.generationId())
        {}
 
    public:
        bool supportsMerkleRoot() const {
            return !!m_pTree;
        }
 
        std::pair<Hash256, bool> tryGetMerkleRoot() const {
            return m_pTree
                    ? std::make_pair(m_pTree->root(), true)
                    : std::make_pair(Hash256(), false);
        }
 
        void updateMerkleRoot(Height height) {
            if (!m_pTree)
                return;
 
            ApplyDeltasToTree(*m_pTree, m_set, m_nextGenerationId, height);
            setApplyCheckpoint();
        }
 
        void setMerkleRoot(const Hash256& merkleRoot) {
            if (!m_pTree)
                CATAPULT_THROW_RUNTIME_ERROR_1("cannot set merkle root", merkleRoot);
 
            m_pTree->reset(merkleRoot);
            setApplyCheckpoint();
        }
 
    private:
        void setApplyCheckpoint() {
            m_set.incrementGenerationId();
            m_nextGenerationId = m_set.generationId();
            m_pTree->setCheckpoint();
        }
 
    private:
        TSet& m_set;
        std::shared_ptr<TTree> m_pTree;
        uint32_t m_nextGenerationId;
    };
}}