Struct TaskRegistry 

pub struct TaskRegistry { /* private fields */ }

A clone-safe registry of spawned tasks.

TaskRegistry: Clone + Send + Sync. Cloning is cheap (Arc bump) and all clones share the same underlying task list.

Implementations

impl TaskRegistry

pub fn new(shutdown: ShutdownToken) -> Self

Construct an empty registry linked to the given shutdown token.

pub fn spawn<F>( &self, name: &'static str, kind: TaskKind, fut: F, ) -> AbortHandle

where F: Future<Output = Result<()>> + Send + 'static,

Spawn a tracked task.

The closure receives no arguments — if it needs the shutdown token or a sub-registry, capture them via move. Use shutdown to obtain a clone of the token.

Returns an AbortHandle so the caller can force-abort the task if needed (e.g., a per-peer handler when the peer disconnects). The real JoinHandle is held by the registry and is awaited in join_all at shutdown. Callers that want to wait for a single task to complete must await by their own means (a tokio::sync::oneshot or similar inside fut) — the registry is optimised for the shutdown-join case, not per-task completion.

pub fn shutdown(&self) -> &ShutdownToken

Borrow the shutdown token this registry is linked to.

pub async fn join_all(&self, deadline: Duration) -> Result<()>

Await all tracked tasks up to deadline; abort laggards.

Returns Ok(()) if every task exited inside the deadline. Returns ServiceError::ShutdownDeadlineExceeded otherwise, with the count of aborted tasks.

pub fn snapshot(&self) -> Vec<TaskSummary>

Snapshot of currently-tracked tasks (live only; completed tasks are not garbage-collected until join_all).

pub fn len(&self) -> usize

Current live-task count.

pub fn is_empty(&self) -> bool

Whether no tasks are currently registered.

Trait Implementations

impl Clone for TaskRegistry

fn clone(&self) -> TaskRegistry

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.