chore(pool): fix endless loop during shutdown

fleek-network · Nov 15, 2024 · cc94b0a · cc94b0a
1 parent cf8e0d0
commit cc94b0a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/core/pool/src/event.rs b/core/pool/src/event.rs
@@ -418,7 +418,10 @@ where
                     task.abort();
                 }
                 self.ongoing_async_tasks.clear();
-                while !matches!(self.event_queue.try_recv(), Err(TryRecvError::Empty)) {}
+                while !matches!(
+                    self.event_queue.try_recv(),
+                    Err(TryRecvError::Empty | TryRecvError::Disconnected)
+                ) {}
             },
             "POOL: spawn"
         );