deserialize_utils.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <optional>
#include <span>
#include <unordered_set>
#include <utility>
#include <variant>
 
#include <sparrow/buffer/buffer.hpp>
#include <sparrow/c_interface.hpp>
 
#include "Message_generated.h"
#include "Schema_generated.h"
 
namespace sparrow_ipc::utils
{
    [[nodiscard]] std::pair<std::uint8_t*, int64_t>
    get_bitmap_pointer_and_null_count(std::span<const uint8_t> validity_buffer_span, const int64_t length);

    [[nodiscard]] std::span<const uint8_t> get_buffer(
        const org::apache::arrow::flatbuf::RecordBatch& record_batch,
        std::span<const uint8_t> body,
        size_t& buffer_index
    );

    [[nodiscard]] std::variant<sparrow::buffer<std::uint8_t>, std::span<const std::uint8_t>>
    get_decompressed_buffer(
        std::span<const uint8_t> buffer_span,
        const org::apache::arrow::flatbuf::BodyCompression* compression
    );

    [[nodiscard]] std::optional<std::unordered_set<sparrow::ArrowFlag>>
    get_sparrow_flags(const org::apache::arrow::flatbuf::Field& field);

    template <typename T>
    inline std::string get_fb_name(const T* obj, const std::string& default_val = "")
    {
        if (!obj)
        {
            return default_val;
        }
 
        auto name = obj->name();
        return name ? name->str() : default_val;
    }
}