arrow_io_source.hpp

Go to the documentation of this file.

/*
 * Copyright (c) 2023-2024, NVIDIA CORPORATION.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#pragma once
 
#include "datasource.hpp"
 
#include <cudf/utilities/export.hpp>
 
#include <arrow/filesystem/filesystem.h>
#include <arrow/io/interfaces.h>
 
#include <memory>
#include <string>
#include <utility>
 
namespace CUDF_EXPORT cudf {
namespace io {
class arrow_io_source : public datasource {
 public:
  explicit arrow_io_source(std::string const& arrow_uri);
 
  explicit arrow_io_source(std::shared_ptr<arrow::io::RandomAccessFile> file)
    : arrow_file(std::move(file))
  {
  }
 
  std::unique_ptr<buffer> host_read(size_t offset, size_t size) override;
 
  size_t host_read(size_t offset, size_t size, uint8_t* dst) override;
  [[nodiscard]] size_t size() const override;
 
 private:
  std::shared_ptr<arrow::fs::FileSystem> filesystem;
  std::shared_ptr<arrow::io::RandomAccessFile> arrow_file;
};
  // end of group
}  // namespace io
}  // namespace CUDF_EXPORT cudf