use miette::miette;
use std::process::ExitCode;
use std::{io, time::Duration};
use tokio::time::sleep;
use tosub::Subsystem;
use tracing::level_filters::LevelFilter;
use tracing_subscriber::{EnvFilter, Layer, fmt, layer::SubscriberExt, util::SubscriberInitExt};
#[tokio::main]
async fn main() -> miette::Result<ExitCode> {
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("hello_world")
.catch_signals()
.with_timeout(Duration::from_secs(5))
.start(|root| async move {
root.spawn("child 1", |subsystem: Subsystem<()>| async move {
println!("Hello from {}", subsystem.name());
sleep(Duration::from_secs(1)).await;
Err(miette!("Oopsie whoopsie!"))
});
root.spawn("child 2", |subsystem| async move {
println!("Hello from {}", subsystem.name());
subsystem.shutdown_requested().await;
println!("{} needs a seconds to shut down ...", subsystem.name());
sleep(Duration::from_secs(1)).await;
Ok::<(), miette::ErrReport>(())
});
root.spawn("child 3", |subsystem| async move {
println!("Hello from {}", subsystem.name());
subsystem.shutdown_requested().await;
println!("{} needs two seconds to shut down ...", subsystem.name());
sleep(Duration::from_secs(2)).await;
Ok::<(), miette::ErrReport>(())
});
root.shutdown_requested().await;
Ok::<(), miette::ErrReport>(())
})
.await
}