Cancel fix

2026-03-16 22:24:39 -07:00
parent 053459f340
commit 1c1c47f5b2
1 changed files with 5 additions and 3 deletions
--- a/crates/pile-toolbox/src/cancelabletask.rs
+++ b/crates/pile-toolbox/src/cancelabletask.rs
@@ -80,10 +80,12 @@ impl CancelFlag {

 	#[inline]
 	pub async fn await_cancel(&self) {
-		if self.is_cancelled() {
-			return;
+		let notified = self.notify.notified();
+		tokio::pin!(notified);
+		notified.as_mut().enable();
+		if !self.is_cancelled() {
+			notified.await;
 		}
-		self.notify.notified().await;
 	}

 	#[inline]