Struct ShardRegistry 

pub struct ShardRegistry { /* private fields */ }

Registry of active shards in the cluster.

The registry tracks all shards, their current status, and health metrics. It is used by the coordinator to manage the cluster topology and route requests to healthy shards.

Implementations

impl ShardRegistry

pub fn new() -> Self

Create a new empty shard registry.

pub fn with_heartbeat_timeout(timeout_ms: u64) -> Self

Create a registry with custom heartbeat timeout.

pub fn register(&mut self, info: ShardInfo) -> ShardId

Register a new shard and return its assigned ID.

The shard will be assigned a unique ID and added to the registry. Its initial status will be set to Online.

pub fn register_with_id(&mut self, info: ShardInfo, id: ShardId) -> ShardId

Register a shard with a specific ID.

This is useful when restoring state or in deterministic testing. The next_id counter will be updated if necessary.

pub fn get(&self, id: &ShardId) -> Option<&ShardInfo>

Get shard info by ID.

pub fn get_registered(&self, id: &ShardId) -> Option<&RegisteredShard>

Get registered shard by ID (includes status).

pub fn get_registered_mut( &mut self, id: &ShardId, ) -> Option<&mut RegisteredShard>

Get mutable registered shard by ID.

pub fn remove(&mut self, id: &ShardId) -> Option<ShardInfo>

Remove a shard from the registry.

pub fn all(&self) -> Vec<ShardInfo>

Get all shard infos.

pub fn all_ids(&self) -> Vec<ShardId>

Get all shard IDs.

pub fn count(&self) -> usize

Get the number of registered shards.

pub fn contains(&self, id: &ShardId) -> bool

Check if a shard exists.

pub fn heartbeat(&mut self, id: &ShardId)

Update heartbeat timestamp for a shard.

pub fn heartbeat_with_timestamp(&mut self, id: &ShardId, timestamp: u64)

Update heartbeat with explicit timestamp (for testing or remote sync).

pub fn set_status(&mut self, id: &ShardId, status: ShardStatus)

Update shard status.

pub fn get_status(&self, id: &ShardId) -> Option<ShardStatus>

Get the status of a shard.

pub fn update_metrics( &mut self, id: &ShardId, document_count: usize, memory_bytes: u64, )

Update shard metrics.

pub fn online_shards(&self) -> Vec<ShardInfo>

Get all online shards.

pub fn shards_with_status(&self, status: ShardStatus) -> Vec<ShardInfo>

Get all shards with a specific status.

pub fn check_dead_shards(&mut self) -> Vec<ShardId>

Check for and mark dead shards based on heartbeat timeout.

Returns the IDs of shards that were marked as offline.

pub fn total_documents(&self) -> u64

Get total document count across all shards.

pub fn total_memory(&self) -> u64

Get total memory usage across all shards.

pub fn least_loaded_shard(&self) -> Option<ShardId>

Get the shard with the least documents (for load balancing).

Trait Implementations

impl Default for ShardRegistry

fn default() -> Self

Returns the “default value” for a type.