From cd790a94286cdc307d399b7f7a43e305e90de5bf Mon Sep 17 00:00:00 2001
From: Emil Fresk <emil.fresk@gmail.com>
Date: Mon, 9 Jan 2023 21:02:53 +0100
Subject: More work on new spawn/executor

---
 src/export/executor.rs | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

(limited to 'src/export')

diff --git a/src/export/executor.rs b/src/export/executor.rs
index 874ee19..2f88eff 100644
--- a/src/export/executor.rs
+++ b/src/export/executor.rs
@@ -1,9 +1,9 @@
+use super::atomic::{AtomicBool, Ordering};
 use core::{
     cell::UnsafeCell,
     future::Future,
     mem::{self, MaybeUninit},
     pin::Pin,
-    sync::atomic::{AtomicBool, Ordering},
     task::{Context, Poll, RawWaker, RawWakerVTable, Waker},
 };
 
@@ -53,9 +53,11 @@ impl<F: Future> AsyncTaskExecutor<F> {
         self.running.load(Ordering::Relaxed)
     }
 
-    /// Checks if a waker has pended the executor.
-    pub fn is_pending(&self) -> bool {
-        self.pending.load(Ordering::Relaxed)
+    /// Checks if a waker has pended the executor and simultaneously clears the flag.
+    pub fn check_and_clear_pending(&self) -> bool {
+        self.pending
+            .compare_exchange(true, false, Ordering::Relaxed, Ordering::Relaxed)
+            .is_ok()
     }
 
     // Used by wakers to indicate that the executor needs to run.
@@ -80,6 +82,7 @@ impl<F: Future> AsyncTaskExecutor<F> {
         debug_assert!(self.running.load(Ordering::Relaxed));
 
         self.task.get().write(MaybeUninit::new(future));
+        self.set_pending();
     }
 
     /// Poll the future in the executor.
-- 
cgit v1.2.3