| 134 | class RecordBatchReaderHead : public arrow::RecordBatchReader { |
| 135 | public: |
| 136 | RecordBatchReaderHead(std::shared_ptr<arrow::RecordBatchReader> reader, |
| 137 | int64_t num_rows) |
| 138 | : done_(false), schema_(reader->schema()), reader_(reader), num_rows_(num_rows) {} |
| 139 | |
| 140 | std::shared_ptr<arrow::Schema> schema() const override { return schema_; } |
| 141 |