deserialization_context.hpp

Go to the documentation of this file.

#pragma once
 
#include <optional>
#include <span>
#include <string>
#include <unordered_set>
#include <vector>
 
#include <sparrow/c_interface.hpp>
 
#include "Schema_generated.h"
#include "Message_generated.h"
 
#include "sparrow_ipc/dictionary_cache.hpp"
#include "sparrow_ipc/metadata.hpp"
 
namespace sparrow_ipc
{
    struct deserialization_context
    {
        const org::apache::arrow::flatbuf::RecordBatch& record_batch;
        std::span<const uint8_t> body;
        size_t& buffer_index;
        size_t& node_index;
        size_t& variadic_counts_idx;
 
        deserialization_context(
            const org::apache::arrow::flatbuf::RecordBatch& record_batch,
            std::span<const uint8_t> body,
            size_t& buffer_index,
            size_t& node_index,
            size_t& variadic_counts_idx)
            : record_batch(record_batch)
            , body(body)
            , buffer_index(buffer_index)
            , node_index(node_index)
            , variadic_counts_idx(variadic_counts_idx)
        {}
    };

    struct field_descriptor
    {
        int64_t length;
        std::string name;
        std::optional<std::vector<sparrow::metadata_pair>> metadata;
        std::optional<std::unordered_set<sparrow::ArrowFlag>> flags;
        bool decode_dictionary_indices;
        const org::apache::arrow::flatbuf::Field& field;
        const dictionary_cache* dictionaries = nullptr;
 
        field_descriptor(
            int64_t length,
            std::string name,
            std::optional<std::vector<sparrow::metadata_pair>> metadata,
            std::optional<std::unordered_set<sparrow::ArrowFlag>> flags,
            bool decode_dictionary_indices,
            const org::apache::arrow::flatbuf::Field& field,
            const dictionary_cache* dictionaries)
            : length(length)
            , name(std::move(name))
            , metadata(std::move(metadata))
            , flags(std::move(flags))
            , decode_dictionary_indices(decode_dictionary_indices)
            , field(field)
            , dictionaries(dictionaries)
        {}
    };
} // namespace sparrow_ipc