CacheDatabaseMixin.h

Go to the documentation of this file.

 
#pragma once
#include "CacheConfiguration.h"
#include "catapult/cache_db/CacheDatabase.h"
#include "catapult/cache_db/UpdateSet.h"
#include "catapult/deltaset/ConditionalContainer.h"
 
namespace catapult { namespace cache {
 
    class CacheDatabaseMixin {
    protected:
        CacheDatabaseMixin(
                const CacheConfiguration& config,
                const std::vector<std::string>& columnFamilyNames,
                FilterPruningMode pruningMode = FilterPruningMode::Disabled)
                : m_pDatabase(config.ShouldUseCacheDatabase
                        ? std::make_unique<CacheDatabase>(CacheDatabaseSettings(
                                config.CacheDatabaseDirectory,
                                GetAdjustedColumnFamilyNames(config, columnFamilyNames),
                                config.MaxCacheDatabaseWriteBatchSize,
                                pruningMode))
                        : std::make_unique<CacheDatabase>())
                , m_containerMode(GetContainerMode(config))
                , m_hasPatriciaTreeSupport(config.ShouldStorePatriciaTrees)
        {}
 
    protected:
        bool hasPatriciaTreeSupport() const {
            return m_hasPatriciaTreeSupport;
        }
 
        CacheDatabase& database() {
            return *m_pDatabase;
        }
 
    protected:
        static deltaset::ConditionalContainerMode GetContainerMode(const CacheConfiguration& config) {
            return config.ShouldUseCacheDatabase
                    ? deltaset::ConditionalContainerMode::Storage
                    : deltaset::ConditionalContainerMode::Memory;
        }
 
        void flush() {
            if (deltaset::ConditionalContainerMode::Storage == m_containerMode)
                database().flush();
        }
 
    private:
        static std::vector<std::string> GetAdjustedColumnFamilyNames(
                const CacheConfiguration& config,
                const std::vector<std::string>& columnFamilyNames) {
            auto adjustedColumnFamilyNames = columnFamilyNames;
            if (config.ShouldStorePatriciaTrees)
                adjustedColumnFamilyNames.push_back("patricia_tree");
 
            return adjustedColumnFamilyNames;
        }
 
    private:
        std::unique_ptr<CacheDatabase> m_pDatabase;
        const deltaset::ConditionalContainerMode m_containerMode;
        const bool m_hasPatriciaTreeSupport;
    };
}}