Struct VnodeRegistry 

pub struct VnodeRegistry { /* private fields */ }

Runtime registry of vnode topology and assignment.

Implementations

impl VnodeRegistry

pub fn new(vnode_count: u32) -> Self

Create a registry sized for vnode_count vnodes, all marked as NodeId::UNASSIGNED. The assignment version starts at 1.

Panics

Panics if vnode_count == 0.

pub fn new_unassigned(vnode_count: u32) -> Self

Create a registry with every vnode unassigned at version 0, so the first stored assignment snapshot (version >= 1) adopts through the standard rotation path.

Panics

Panics if vnode_count == 0.

pub fn single_owner(vnode_count: u32, owner: NodeId) -> Self

Create a registry where every vnode is owned by the same node.

Used by single-instance / embedded deployments.

Panics

Panics if vnode_count == 0.

pub fn vnode_count(&self) -> u32

Number of vnodes.

pub fn assignment_version(&self) -> u64

Current monotonic assignment version.

pub fn owner(&self, vnode: u32) -> NodeId

Owner of a given vnode. Returns NodeId::UNASSIGNED if the vnode is out of range or unassigned.

pub fn snapshot(&self) -> Arc<[NodeId]>

Snapshot the current assignment vector. Cheap — internally an Arc::clone.

pub fn versioned_snapshot(&self) -> VnodeAssignmentSnapshot

Consistent ownership, version, and source-handoff publication.

pub fn read_assignment(&self) -> VnodeAssignmentReadGuard<'_>

Pin the current publication for a short, non-blocking source capture.

pub fn set_assignment(&self, new_assignment: Arc<[NodeId]>)

Replace the full assignment and bump the version.

Panics

Panics if new_assignment.len() != self.vnode_count.

pub fn set_assignment_and_version( &self, new_assignment: Arc<[NodeId]>, version: u64, )

Replace the full assignment and set the version to version atomically. For recovery paths that must restore the registry to a persisted fence generation, not a fresh bump.

Panics

Panics on length mismatch, or if version is less than the current one (assignment versions are monotonic).

pub fn set_assignment_and_version_with_source_handoff( &self, new_assignment: Arc<[NodeId]>, version: u64, source_handoff: Arc<CommittedSourceHandoff>, )

Atomically publish ownership, its monotonic version, and the sealed connector cursors used by sources acquiring partitions in that version.

Panics

Panics on length mismatch or version regression.

pub fn set_assignment_and_version_carrying_source_handoff( &self, new_assignment: Arc<[NodeId]>, version: u64, )

Publish a version that does not acquire local ownership while retaining the prior version-bound handoff until the source has reconciled it.

pub fn vnode_for_key(&self, key: &[u8]) -> u32

Map a primary key to a vnode.

pub fn mark_restoring(&self, vnodes: &[u32])

Mark vnodes as Restoring.

Called during a rebalance for the vnodes a node newly acquires, before their committed state has been applied. Out-of-range ids are ignored.

pub fn mark_active(&self, vnodes: &[u32])

Mark vnodes as Active.

Called once a newly-acquired vnode’s state has been applied (or immediately for vnodes that had no durable state to restore). Out-of-range ids are ignored.

pub fn is_restoring(&self, vnode: u32) -> bool

Whether vnode is currently Restoring. Out-of-range ids are reported as not restoring.

pub fn any_restoring(&self) -> bool

Whether any vnode is currently restoring. Cheap pre-check the emission gate uses to skip per-row work in the common case.

pub fn restoring_vnodes(&self) -> Vec<u32>

Vnodes currently Restoring, ascending.

Trait Implementations

impl Debug for VnodeRegistry

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.