channel: early break when fetch nothing (#13516)

close #13394 Otherwise it will waste CPU on loop. Signed-off-by: Jay Lee <BusyJayLee@gmail.com> Co-authored-by: Ti Chi Robot <ti-community-prow-bot@tidb.io>
tikv · Oct 12, 2022 · 066a422 · 066a422
1 parent c286b8a
commit 066a422
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/components/tikv_util/src/mpsc/future.rs b/components/tikv_util/src/mpsc/future.rs
@@ -213,6 +213,8 @@ where
         for _ in 1..ctx.max_batch_size {
             if let Poll::Ready(Some(m)) = ctx.rx.poll_next_unpin(cx) {
                 (ctx.collector)(&mut collector, m);
+            } else {
+                break;
             }
         }
         Poll::Ready(Some(collector))