Fix size hint on the vector iterator

scylladb · Aug 12, 2024 · de99af7 · de99af7
1 parent 4a295d6
commit de99af7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/scylla-cql/src/types/deserialize/value.rs b/scylla-cql/src/types/deserialize/value.rs
@@ -753,6 +753,7 @@ where
 pub struct VectorIterator<'frame, T> {
     coll_typ: &'frame ColumnType,
     elem_typ: &'frame ColumnType,
+    count: usize,
     raw_iter: VectorBytesSequenceIterator<'frame>,
     phantom_data: std::marker::PhantomData<T>,
 }
@@ -768,6 +769,7 @@ impl<'frame, T> VectorIterator<'frame, T> {
         Self {
             coll_typ,
             elem_typ,
+            count,
             raw_iter: VectorBytesSequenceIterator::new(count, elem_len, slice),
             phantom_data: std::marker::PhantomData,
         }
@@ -845,7 +847,7 @@ where
 
     #[inline]
     fn size_hint(&self) -> (usize, Option<usize>) {
-        self.raw_iter.size_hint()
+        (self.count, Some(self.count))
     }
 }