Method register_task

nodedb/src/control/shutdown/bus.rs:220–246

Register a task for the given drain phase. Returns a `DrainGuard` the task must hold until its cleanup is complete. `abort_handle`: if `Some`, the task will be aborted if it misses the budget. Pass `None` for blocking threads.

(
        &self,
        drain_at: ShutdownPhase,
        name: &'static str,
        abort_handle: Option<tokio::task::AbortHandle>,
    )

    /// `abort_handle`: if `Some`, the task will be aborted if it misses
    /// the budget. Pass `None` for blocking threads.
    pub fn register_task(
        &self,
        drain_at: ShutdownPhase,
        name: &'static str,
        abort_handle: Option<tokio::task::AbortHandle>,
    ) -> DrainGuard {
        let mut guard = lock_bus(&self.state);
        let id = guard.alloc_id();
        guard.tasks.insert(
            id,
            TaskEntry {
                name,
                phase: drain_at,
                drained: false,
                abort_handle,
            },
        );
        let phase_rx = self.phase_tx.subscribe();
        DrainGuard {
            task_id: id,
            phase: drain_at,
            state: Arc::clone(&self.state),
            phase_rx,
            reported: false,
            name,
        }
    }

    /// Initiate graceful shutdown. Idempotent — second call is a no-op.
    ///