Fix missed items in thread queue

lune-org · Feb 13, 2024 · 71026f9 · 71026f9
1 parent 7c59d0c
commit 71026f9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/queue.rs b/lib/queue.rs
@@ -60,7 +60,12 @@ impl ThreadQueue {
     #[inline]
     pub async fn wait_for_item(&self) {
         if self.queue.is_empty() {
-            self.event.listen().await;
+            let listener = self.event.listen();
+            // NOTE: Need to check again, we could have gotten
+            // new queued items while creating our listener
+            if self.queue.is_empty() {
+                listener.await;
+            }
         }
     }