skip iterator removed from primitive encoding

apache · Jun 8, 2024 · 4fec097 · 4fec097
1 parent 087f34b
commit 4fec097
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/arrow-row/src/fixed.rs b/arrow-row/src/fixed.rs
@@ -222,7 +222,9 @@ pub fn encode<T: FixedLengthEncoding, I: IntoIterator<Item = Option<T>>>(
     i: I,
     opts: SortOptions,
 ) {
-    for (offset, maybe_val) in offsets.iter_mut().skip(1).zip(i) {
+    let mut offset_idx = 1;
+    for maybe_val in i {
+        let offset = &mut offsets[offset_idx];
         let end_offset = *offset + T::ENCODED_LEN;
         if let Some(val) = maybe_val {
             let to_write = &mut data[*offset..end_offset];
@@ -237,6 +239,7 @@ pub fn encode<T: FixedLengthEncoding, I: IntoIterator<Item = Option<T>>>(
             data[*offset] = null_sentinel(opts);
         }
         *offset = end_offset;
+        offset_idx += 1;
     }
 }