| 340 | } |
| 341 | |
| 342 | static RecordBatchGenerator GeneratorFromReader( |
| 343 | const Future<std::shared_ptr<csv::StreamingReader>>& reader, int64_t batch_size) { |
| 344 | auto gen_fut = reader.Then( |
| 345 | [batch_size]( |
| 346 | const std::shared_ptr<csv::StreamingReader>& reader) -> RecordBatchGenerator { |
| 347 | auto batch_gen = [reader]() { return reader->ReadNextAsync(); }; |
| 348 | return MakeChunkedBatchGenerator(std::move(batch_gen), batch_size); |
| 349 | }); |
| 350 | return MakeFromFuture(std::move(gen_fut)); |
| 351 | } |
| 352 | |
| 353 | } // namespace |
| 354 |
no test coverage detected