| 194 | } |
| 195 | |
| 196 | void ScheduleBatch(std::shared_ptr<RecordBatch> batch) { |
| 197 | file_tasks_->AddSimpleTask( |
| 198 | [self = shared_from_this(), batch = std::move(batch)]() { |
| 199 | return self->WriteNext(std::move(batch)); |
| 200 | }, |
| 201 | "DatasetWriter::WriteBatch"sv); |
| 202 | } |
| 203 | |
| 204 | Result<int64_t> PopAndDeliverStagedBatch() { |
| 205 | ARROW_ASSIGN_OR_RAISE(std::shared_ptr<RecordBatch> next_batch, PopStagedBatch()); |