CacheStorageAdapter.h

Go to the documentation of this file.

 
#pragma once
#include "CacheStorage.h"
#include "CatapultCacheView.h"
#include "ChunkedDataLoader.h"
#include "catapult/exceptions.h"
 
namespace catapult { namespace cache {
 
    template<typename TCache, typename TStorageTraits>
    class CacheStorageAdapter : public CacheStorage {
    public:
        explicit CacheStorageAdapter(TCache& cache)
                : m_cache(cache)
                , m_name(TCache::Name)
        {}
 
    public:
        const std::string& name() const override {
            return m_name;
        }
 
    public:
        void saveAll(const CatapultCacheView& cacheView, io::OutputStream& output) const override {
            const auto& view = cacheView.sub<TCache>();
            io::Write64(output, view.size());
 
            auto pIterableView = view.tryMakeIterableView();
            for (const auto& element : *pIterableView)
                SaveValue(element, output);
 
            output.flush();
        }
 
        void saveSummary(const CatapultCacheDelta&, io::OutputStream&) const override {
            CATAPULT_THROW_INVALID_ARGUMENT("CacheStorageAdapter does not support saveSummary");
        }
 
        void loadAll(io::InputStream& input, size_t batchSize) override {
            auto delta = m_cache.createDelta();
 
            ChunkedDataLoader<TStorageTraits> loader(input);
            while (loader.hasNext()) {
                loader.next(batchSize, *delta);
                m_cache.commit();
            }
        }
 
    private:
        // assume pair indicates maps and only forward value to save
 
        template<typename T>
        static void SaveValue(const T& value, io::OutputStream& output) {
            TStorageTraits::Save(value, output);
        }
 
        template<typename T1, typename T2>
        static void SaveValue(const std::pair<T1, T2>& pair, io::OutputStream& output) {
            TStorageTraits::Save(pair.second, output);
        }
 
    private:
        TCache& m_cache;
        std::string m_name;
    };
}}