fix(row selection): off-by-one error

sdd · sdd · commit 40e9bff0aece · 2024-08-17T00:20:42.000+01:00
diff --git a/crates/iceberg/src/arrow/reader.rs b/crates/iceberg/src/arrow/reader.rs
@@ -441,7 +441,7 @@ impl ArrowReader {
                 // skip row groups that aren't present in selected_row_groups
                 if idx == selected_row_groups[selected_row_groups_idx] {
                     selected_row_groups_idx += 1;
-                    if selected_row_groups_idx == selected_row_groups.len() {
+                    if selected_row_groups_idx == selected_row_groups.len() + 1 {
                         break;
                     }
                 } else {

Original file line number	Diff line number	Diff line change
`@@ -441,7 +441,7 @@ impl ArrowReader {`
`441`	`441`	`// skip row groups that aren't present in selected_row_groups`
`442`	`442`	`if idx == selected_row_groups[selected_row_groups_idx] {`
`443`	`443`	`selected_row_groups_idx += 1;`
`444`		`- if selected_row_groups_idx == selected_row_groups.len() {`
	`444`	`+ if selected_row_groups_idx == selected_row_groups.len() + 1 {`
`445`	`445`	`break;`
`446`	`446`	`}`
`447`	`447`	`} else {`