(&mut self)
| 1283 | type Item = Bytes; |
| 1284 | |
| 1285 | fn next(&mut self) -> Option<Self::Item> { |
| 1286 | // Return `batch_size` rows per batch: |
| 1287 | let mut buffer = Vec::new(); |
| 1288 | for _ in 0..self.batch_size { |
| 1289 | if self.offset >= self.line_count { |
| 1290 | break; |
| 1291 | } |
| 1292 | buffer.extend_from_slice(&csv_line(self.offset)); |
| 1293 | self.offset += 1; |
| 1294 | } |
| 1295 | |
| 1296 | (!buffer.is_empty()).then(|| buffer.into()) |
| 1297 | } |
| 1298 | } |
| 1299 | |
| 1300 | fn csv_expected_batch(schema: SchemaRef, line_count: usize) -> Result<RecordBatch> { |