| 537 | } |
| 538 | |
| 539 | Status WriteRecordBatch(const RecordBatch& batch) override { |
| 540 | RecordBatchIterator iterator = RecordBatchSliceIterator(batch, options_.batch_size); |
| 541 | for (auto maybe_slice : iterator) { |
| 542 | ARROW_ASSIGN_OR_RAISE(std::shared_ptr<RecordBatch> slice, maybe_slice); |
| 543 | RETURN_NOT_OK(TranslateMinimalBatch(*slice)); |
| 544 | RETURN_NOT_OK(FlushToSink()); |
| 545 | stats_.num_record_batches++; |
| 546 | } |
| 547 | return Status::OK(); |
| 548 | } |
| 549 | |
| 550 | Status WriteTable(const Table& table, int64_t max_chunksize) override { |
| 551 | TableBatchReader reader(table); |