summaryrefslogtreecommitdiff
path: root/cli/util/sync.rs
diff options
context:
space:
mode:
Diffstat (limited to 'cli/util/sync.rs')
-rw-r--r--cli/util/sync.rs18
1 files changed, 18 insertions, 0 deletions
diff --git a/cli/util/sync.rs b/cli/util/sync.rs
index dddb5991c..3986c883e 100644
--- a/cli/util/sync.rs
+++ b/cli/util/sync.rs
@@ -8,6 +8,7 @@ use std::sync::Arc;
use deno_core::futures::task::AtomicWaker;
use deno_core::futures::Future;
use deno_core::parking_lot::Mutex;
+use tokio_util::sync::CancellationToken;
/// Simplifies the use of an atomic boolean as a flag.
#[derive(Debug, Default)]
@@ -160,6 +161,23 @@ impl<'a> Future for TaskQueuePermitAcquireFuture<'a> {
}
}
+#[derive(Debug, Default, Clone)]
+pub struct AsyncFlag(CancellationToken);
+
+impl AsyncFlag {
+ pub fn raise(&self) {
+ self.0.cancel();
+ }
+
+ pub fn is_raised(&self) -> bool {
+ self.0.is_cancelled()
+ }
+
+ pub fn wait_raised(&self) -> impl std::future::Future<Output = ()> + '_ {
+ self.0.cancelled()
+ }
+}
+
#[cfg(test)]
mod test {
use deno_core::futures;