Struct Registry 

pub struct Registry { /* private fields */ }

Available on crate feature serve only.

Implementations

impl Registry

pub fn new(max_queued: usize) -> Self

pub fn try_reserve(&self) -> bool

Reserve a queue slot. Returns true if a slot was reserved, or false if the queue is full. Atomic against concurrent submits via a CAS loop (AtomicUsize::try_update, stabilized in Rust 1.95).

pub fn release_reservation(&self)

Release a slot reserved by try_reserve that will not be spawned (idempotency replay/conflict).

pub fn register(&self, run_id: String, token: CancellationToken)

pub fn mark_running(&self)

Queued → running: the run acquired its execution permit.

pub fn mark_running_unqueued(&self)

Running transition for a run that never occupied a local queue slot (the cluster claim path: submit writes Pending + releases its reservation, so no queued slot exists to consume). Only bumps in_flight.

pub fn mark_finished(&self, run_id: &str)

Running → terminal: drop the token and wake any shutdown drain waiter.

pub fn mark_queued_cancelled(&self, run_id: &str)

Queued → terminal: a run was cancelled (or the server shut down) while still waiting for an execution permit, so it never became in-flight. Release its queue slot (not in_flight), drop the token, and wake any drain waiter. Distinct from Self::mark_finished, which decrements in_flight (#146 R: a cancel on a queued run now takes effect at once).

pub fn cancel(&self, run_id: &str) -> bool

Cancel a live run. Returns true if a live token existed.

pub fn register_shard( &self, run_id: &str, shard_id: &str, token: CancellationToken, )

Register a shard’s cancel token under a per-shard key ({run_id}::{shard_id}). Separate from a run’s token so a sharded run’s shards each get their own cooperative-cancel signal (Mode B, #230 / F10).

pub fn deregister_shard(&self, run_id: &str, shard_id: &str)

Drop a token by key without touching the queue/in-flight counters. Used to remove a finished shard’s token (shard accounting is separate from the parent run’s in_flight, so Self::mark_finished is not appropriate).

pub fn cancel_run_shards(&self, run_id: &str) -> usize

Fire every registered shard token whose key belongs to run_id (key prefix {run_id}::). Returns how many tokens were fired. Drives a cross-instance cancel of a sharded run: the claim loop calls this for each run id returned by pending_shard_cancellations (F10).

pub fn queued(&self) -> usize

pub fn in_flight(&self) -> usize

pub fn is_full(&self) -> bool

pub async fn wait_drained(&self)

Resolve once no run is queued or in flight. Arms the notification before re-checking so a transition can’t be missed.