reset-current-batch-size-with-each-items-reset

pragmaxim-com · Jun 11, 2024 · 0b170f6 · 0b170f6
1 parent 0320f5b
commit 0b170f6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -138,6 +138,7 @@ where
                     let last = if me.items.is_empty() {
                         None
                     } else {
+                        *me.current_batch_size = 0;
                         Some(std::mem::take(me.items))
                     };
 
@@ -147,6 +148,7 @@ where
         }
 
         if !me.items.is_empty() {
+            *me.current_batch_size = 0;
             return Poll::Ready(Some(std::mem::take(me.items)));
         }