# tokio-task-supervisor
A wrapper around `tokio_util::task::TaskTracker` that adds coordinated shutdown via a shared cancellation token.
## What it does
- Tracks running tasks (same as `TaskTracker`)
- Provides a shared `CancellationToken` for shutdown
- Spawns tasks that automatically get cancellation tokens
- Handles graceful shutdown (close tracker + cancel token + wait)
## Usage
```rust
use tokio_task_supervisor::TaskSupervisor;
use tokio::time::{sleep, Duration};
#[tokio::main]
async fn main() {
let supervisor = TaskSupervisor::new();
// Spawn tasks that can be cancelled
let handle = supervisor.spawn_with_token(|token| async move {
loop {
tokio::select! {
_ = token.cancelled() => break,
_ = sleep(Duration::from_millis(100)) => {}
}
}
});
// Shutdown all tasks
supervisor.shutdown().await;
handle.await.expect("task finished");
}
```
## API
- `spawn_with_token()` - spawn task with cancellation token
- `spawn_with_cancel()` - spawn task that races against cancellation
- `shutdown()` - cancel all tasks and wait for completion
## License
MIT