rnc 0.1.1

arbitrary TCP and UDP connections and listens
Documentation 
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

use std::future::Future;

use tokio_util::sync::CancellationToken;

/// Create a cancellable task.
///
/// # Arguments
///
/// * `token` - The token that will be cancelled when the task is cancelled.
/// * `task` - The task to spawn.
#[macro_export]
macro_rules! cancellable_task {
    ( $token:expr, $task:expr ) => {{
        let token = $token.clone();

        async move {
            tokio::select! {
                _ = token.cancelled() => {}
                _ = $task => {}
            }
        }
    }};
}

/// Spawn a task that will be cancelled when the token is cancelled.
///
/// # Arguments
///
/// * `token` - The token that will be cancelled when the task is cancelled.
/// * `task` - The task to spawn.
///
/// # Returns
///
/// The join handle of the spawned task.
pub fn spawn_cancellable_task<T>(token: &CancellationToken, task: T) -> tokio::task::JoinHandle<()>
where
    T: Future + Send + 'static,
{
    tokio::spawn(cancellable_task!(token, task))
}