Struct NodeRegistry 

pub struct NodeRegistry { /* private fields */ }

Node Registry manages cluster topology

Implementations

impl NodeRegistry

pub fn new(partition_count: u32) -> Self

Create new node registry

Arguments

• partition_count: Total fixed partitions (32 for single-node, 1024+ for cluster)

pub fn register_node(&self, node: Node)

Register a node in the cluster

Automatically rebalances partitions across healthy nodes.

pub fn unregister_node(&self, node_id: u32)

Unregister a node from the cluster

Triggers automatic rebalancing to remaining nodes.

pub fn set_node_health(&self, node_id: u32, healthy: bool)

Mark node as healthy or unhealthy

Unhealthy nodes are excluded from partition assignment.

pub fn node_for_partition(&self, partition_id: u32) -> Option<u32>

Find node responsible for a partition

Returns None if no healthy node is assigned to the partition.

pub fn get_node(&self, node_id: u32) -> Option<Node>

Get node by ID

pub fn all_nodes(&self) -> Vec<Node>

Get all nodes

pub fn healthy_nodes(&self) -> Vec<Node>

Get healthy nodes

pub fn partition_distribution(&self) -> HashMap<u32, Vec<u32>>

Get partition distribution statistics

pub fn is_cluster_healthy(&self) -> bool

Check if cluster is healthy

Returns true if all partitions have at least one healthy node assigned.

pub fn node_count(&self) -> usize

Get node count

pub fn healthy_node_count(&self) -> usize

Get healthy node count