| 548 | } |
| 549 | |
| 550 | Status Close() override { |
| 551 | // reading from generator until end is reached. |
| 552 | std::shared_ptr<RecordBatch> batch; |
| 553 | RETURN_NOT_OK(ReadNext(&batch)); |
| 554 | while (batch != nullptr) { |
| 555 | RETURN_NOT_OK(ReadNext(&batch)); |
| 556 | } |
| 557 | return Status::OK(); |
| 558 | } |
| 559 | |
| 560 | MemoryPool* pool_; |
| 561 | std::shared_ptr<Schema> schema_; |