reset woken of outer future after polled

tokio-rs · Oct 7, 2021 · bb720b1 · bb720b1
1 parent d047584
commit bb720b1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tokio/src/runtime/basic_scheduler.rs b/tokio/src/runtime/basic_scheduler.rs
@@ -224,7 +224,7 @@ impl<P: Park> Inner<P> {
             pin!(future);
 
             'outer: loop {
-                if scheduler.spawner.was_woken() || !polled {
+                if scheduler.spawner.reset_woken() || !polled {
                     polled = true;
                     scheduler.stats.incr_poll_count();
                     if let Ready(v) = crate::coop::budget(|| future.as_mut().poll(&mut cx)) {
@@ -423,8 +423,9 @@ impl Spawner {
         waker_ref(&self.shared)
     }
 
-    fn was_woken(&self) -> bool {
-        self.shared.woken.load(Acquire)
+    // reset woken to false and return original value
+    fn reset_woken(&self) -> bool {
+        self.shared.woken.swap(false, AcqRel)
     }
 }