tosub 1.0.0

A subsystem lifecycle management library for tokio applications
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

use miette::IntoDiagnostic;
use std::{io, time::Duration};
use tokio::{select, time::sleep};
use tracing::level_filters::LevelFilter;
use tracing_subscriber::{EnvFilter, Layer, fmt, layer::SubscriberExt, util::SubscriberInitExt};

#[tokio::main]
async fn main() -> miette::Result<()> {
    tracing_subscriber::registry()
        .with(
            fmt::Layer::new().with_writer(io::stderr).with_filter(
                EnvFilter::builder()
                    .with_default_directive(LevelFilter::INFO.into())
                    .from_env_lossy(),
            ),
        )
        .init();

    tosub::build_root("root")
        .catch_signals()
        .with_timeout(Duration::from_secs(5))
        .start(|s| async move {
            s.spawn("branch", |s| async move {
                let tick = s.spawn("tick", |s| async move {
                    for i in 0..10 {
                        select! {
                            _ = sleep(Duration::from_millis(950)) => println!("tick {i}"),
                            _ = s.shutdown_requested() => break,
                        }
                    }
                    Ok::<(), miette::Error>(())
                });

                let tock = s.spawn("tock", |s| async move {
                    for i in 0..10 {
                        if i == 4 {
                            return Err(miette::miette!("Oh noes!"));
                        }
                        select! {
                            _ = sleep(Duration::from_millis(1000)) => println!("tock {i}"),
                            _ = s.shutdown_requested() => break,
                        }
                    }
                    Ok::<(), miette::Error>(())
                });

                tick.join().await;
                tock.join().await;
                s.request_global_shutdown();

                Ok::<(), miette::Error>(())
            })
            .join()
            .await;

            Ok::<(), miette::Error>(())
        })
        .await
        .into_diagnostic()?;

    Ok(())
}