perf: improve AtomicWaker::wake performance (#2983)

2026-01-25 03:26:14 +00:00 · 2026-01-21 21:58:28 +08:00
parent 2bb9a288a9
commit 11719721e9
1 changed files with 2 additions and 1 deletions
--- a/futures-core/src/task/__internal/atomic_waker.rs
+++ b/futures-core/src/task/__internal/atomic_waker.rs
@@ -385,7 +385,8 @@ impl AtomicWaker {
                let waker = unsafe { (*self.waker.get()).take() };

                // Release the lock
-                self.state.fetch_and(!WAKING, Release);
+                let old_state = self.state.swap(WAITING, Release);
+                debug_assert!(old_state == WAKING);

                waker
            }