dictionary_cache.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <functional>
#include <map>
#include <optional>
 
#include <sparrow/record_batch.hpp>
 
#include "sparrow_ipc/config/config.hpp"
 
namespace sparrow_ipc
{
    class SPARROW_IPC_API dictionary_cache
    {
    public:
        void store_dictionary(int64_t id, sparrow::record_batch batch, bool is_delta);

        std::optional<std::reference_wrapper<const sparrow::record_batch>> get_dictionary(int64_t id) const;

        [[nodiscard]] bool contains(int64_t id) const noexcept;

        void clear() noexcept;

        [[nodiscard]] size_t size() const noexcept ;
 
    private:
        std::map<int64_t, sparrow::record_batch> m_dictionaries;
    };
}