CachePatriciaTree.h

Go to the documentation of this file.

 
#pragma once
#include "catapult/cache_db/CacheDatabase.h"
#include "catapult/cache_db/PatriciaTreeRdbDataSource.h"
#include <memory>
 
namespace catapult { namespace cache {
 
    template<typename TTree>
    class CachePatriciaTree {
    public:
        CachePatriciaTree(bool enable, CacheDatabase& database, size_t columnId)
                : m_pImpl(enable ? std::make_unique<Impl>(database, columnId) : nullptr)
        {}
 
    public:
        const auto* get() const {
            return m_pImpl ? &m_pImpl->tree() : nullptr;
        }
 
    public:
        auto rebase() {
            return m_pImpl ? m_pImpl->tree().rebase() : nullptr;
        }
 
        auto rebaseDetached() const {
            return m_pImpl ? m_pImpl->tree().rebaseDetached() : nullptr;
        }
 
        void commit() {
            if (m_pImpl)
                m_pImpl->commit();
        }
 
    private:
        class Impl {
        public:
            Impl(CacheDatabase& database, size_t columnId)
                    : m_container(database, columnId)
                    , m_dataSource(m_container)
                    , m_pTree(std::make_unique<TTree>(m_dataSource)) {
                Hash256 rootHash;
                if (!m_container.prop("root", rootHash))
                    return;
 
                if (Hash256() == rootHash)
                    return;
 
                m_pTree = std::make_unique<TTree>(m_dataSource, rootHash);
            }
 
        public:
            const auto& tree() const {
                return *m_pTree;
            }
 
            auto& tree() {
                return *m_pTree;
            }
 
        public:
            void commit() {
                m_pTree->commit();
 
                // skip setProp if hash did not change
                Hash256 rootHash;
                if (m_container.prop("root", rootHash) && rootHash == m_pTree->root())
                    return;
 
                m_container.setProp("root", m_pTree->root());
            }
 
        private:
            PatriciaTreeContainer m_container;
            PatriciaTreeRdbDataSource m_dataSource;
            std::unique_ptr<TTree> m_pTree;
        };
 
        std::unique_ptr<Impl> m_pImpl;
    };
}}