fix TempFdArrayIterator next panic

tikv · Aug 3, 2024 · 13b8ac4 · 13b8ac4
1 parent 4939f73
commit 13b8ac4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/collector.rs b/src/collector.rs
@@ -221,6 +221,9 @@ impl<'a, T> Iterator for TempFdArrayIterator<'a, T> {
             Some(&self.buffer[self.index - 1])
         } else {
             let length = self.file_vec.len() / std::mem::size_of::<T>();
+            if length == 0 {
+                return None;
+            }
             let ts =
                 unsafe { std::slice::from_raw_parts(self.file_vec.as_ptr() as *const T, length) };
             if self.index - self.buffer.len() < ts.len() {