| 436 | } |
| 437 | |
| 438 | int64_t StreamReader::SkipRows(int64_t num_rows_to_skip) { |
| 439 | if (0 != column_index_) { |
| 440 | throw ParquetException("Must finish reading current row before skipping rows."); |
| 441 | } |
| 442 | int64_t num_rows_remaining_to_skip = num_rows_to_skip; |
| 443 | |
| 444 | while (!eof_ && (num_rows_remaining_to_skip > 0)) { |
| 445 | int64_t num_rows_in_row_group = row_group_reader_->metadata()->num_rows(); |
| 446 | int64_t num_rows_remaining_in_row_group = |
| 447 | num_rows_in_row_group - (current_row_ - row_group_row_offset_); |
| 448 | |
| 449 | if (num_rows_remaining_in_row_group > num_rows_remaining_to_skip) { |
| 450 | for (auto reader : column_readers_) { |
| 451 | SkipRowsInColumn(reader.get(), num_rows_remaining_to_skip); |
| 452 | } |
| 453 | current_row_ += num_rows_remaining_to_skip; |
| 454 | num_rows_remaining_to_skip = 0; |
| 455 | } else { |
| 456 | num_rows_remaining_to_skip -= num_rows_remaining_in_row_group; |
| 457 | current_row_ += num_rows_remaining_in_row_group; |
| 458 | NextRowGroup(); |
| 459 | } |
| 460 | } |
| 461 | return num_rows_to_skip - num_rows_remaining_to_skip; |
| 462 | } |
| 463 | |
| 464 | int64_t StreamReader::SkipColumns(int64_t num_columns_to_skip) { |
| 465 | int64_t num_columns_skipped = 0; |