arrow_export.hpp

Go to the documentation of this file.

#pragma once
 
 
#ifdef ITCH_WITH_ARROW
 
#include <cstdint>
#include <memory>
#include <string>
 
#include "itch/messages.hpp"
 
namespace arrow {
class Array;
}  // namespace arrow
 
namespace itch::io {
 
class ArrowExporter {
   public:
    ArrowExporter();
 
    ~ArrowExporter();
 
    ArrowExporter(const ArrowExporter&) = delete;
 
    auto operator=(const ArrowExporter&) -> ArrowExporter& = delete;
 
    ArrowExporter(ArrowExporter&&) noexcept = default;
 
    auto operator=(ArrowExporter&&) noexcept -> ArrowExporter& = default;
 
    auto append(const Message& message) -> void;
 
    [[nodiscard]] auto rows() const noexcept -> std::uint64_t;
 
    [[nodiscard]] auto write_parquet(const std::string& path) -> bool;
 
    [[nodiscard]] auto error() const -> const std::string&;
 
   private:
    struct Impl;
    std::unique_ptr<Impl> m_impl;
};
 
}  // namespace itch::io
 
#endif  // ITCH_WITH_ARROW