Spaces:
Build error
Build error
File size: 1,325 Bytes
84d2a97 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
use super::*;
/// # Cancel safety
///
/// This function is cancel safe.
///
/// If cancelled, the cancellation token provided to the `task` will be triggered automatically.
pub async fn spawn_cancel_on_drop<Out, Task>(task: Task) -> Result<Out, Error>
where
Task: FnOnce(CancellationToken) -> Out + Send + 'static,
Out: Send + 'static,
{
let cancel = CancellationToken::new();
let task = {
let cancel = cancel.child_token();
move || task(cancel)
};
let guard = cancel.drop_guard();
let output = tokio::task::spawn_blocking(task).await?;
guard.disarm();
Ok(output)
}
/// # Cancel safety
///
/// This function is cancel safe.
///
/// If cancelled without triggering the cancellation token, the `task` will still run to completion.
///
/// This function *will* return early, and the `task` *may* return early by triggering the
/// cancellation token.
pub async fn spawn_cancel_on_token<Out, Task>(
cancel: CancellationToken,
task: Task,
) -> Result<Out, Error>
where
Task: FnOnce(CancellationToken) -> Out + Send + 'static,
Out: Send + 'static,
{
let task = {
let cancel = cancel.child_token();
move || task(cancel)
};
let output = future::cancel_on_token(cancel, tokio::task::spawn_blocking(task)).await??;
Ok(output)
}
|