task-motel 0.1.0

Opinionated (Tokio) task manager with nested task groups and stoppable tasks
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
41
42
43
44

use futures::{future::BoxFuture, Future, FutureExt, Stream, StreamExt};
use tokio::time::error::Elapsed;

use crate::{StopListener, TmResult};

pub async fn quickpoll<T, F: Future<Output = T>>(f: F) -> Result<T, Elapsed> {
    tokio::time::timeout(tokio::time::Duration::from_millis(10), f).await
}

pub async fn not_ready<T>(f: impl Future<Output = T>) -> bool {
    quickpoll(f).await.is_err()
}

pub async fn ready<T>(f: impl Future<Output = T>) -> bool {
    quickpoll(f).await.is_ok()
}

pub async fn ok_fut(f: impl Future<Output = ()>) -> TmResult {
    f.await;
    Ok(())
}

pub fn blocker(info: &str, stop_rx: StopListener) -> BoxFuture<'static, String> {
    let info = info.to_string();
    async move {
        stop_rx.await;
        tracing::info!("stopped: {}", info);
        info
    }
    .boxed()
}

pub fn fused<'a>(
    info: &str,
    mut stream: impl Stream<Item = ()> + Unpin + Send + 'a,
) -> BoxFuture<'a, String> {
    let info = info.to_string();
    async move {
        stream.next().await;
        tracing::info!("stopped: {}", info);
        info
    }
    .boxed()
}