avoid-micro-batching

pragmaxim-com · Jun 11, 2024 · 5fb2c23 · 5fb2c23
1 parent cc86a28
commit 5fb2c23
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -121,7 +121,9 @@ where
         let mut me = self.as_mut().project();
         loop {
             match me.stream.as_mut().poll_next(cx) {
-                Poll::Pending => break,
+                Poll::Pending => {
+                    return Poll::Pending;
+                }
                 Poll::Ready(Some(item)) => {
                     if me.items.is_empty() {
                         me.items.reserve(*me.min_batch_size);
@@ -141,18 +143,10 @@ where
                         *me.current_batch_size = 0;
                         Some(std::mem::take(me.items))
                     };
-
                     return Poll::Ready(last);
                 }
             }
         }
-
-        if !me.items.is_empty() {
-            *me.current_batch_size = 0;
-            return Poll::Ready(Some(std::mem::take(me.items)));
-        }
-
-        Poll::Pending
     }
 }