Struct RebalanceManager 

pub struct RebalanceManager { /* private fields */ }

Manager for automatic rebalancing

Implementations

impl RebalanceManager

pub fn new(config: RebalanceConfig) -> Self

Create a new rebalance manager

pub fn register_node(&self, load: NodeLoad)

Register a node with its load information

pub fn unregister_node(&self, node_id: &str)

Remove a node from tracking

pub fn update_node_load( &self, node_id: &str, update: impl FnOnce(&mut NodeLoad), )

Update node load information

pub fn register_shard(&self, shard_id: &str, node_id: &str)

Register a shard assignment

pub fn unregister_shard(&self, shard_id: &str)

Remove a shard assignment

pub fn calculate_imbalance(&self) -> f64

Calculate cluster imbalance score (0.0 = perfectly balanced, 1.0 = maximally imbalanced)

pub fn needs_rebalance(&self) -> bool

Check if rebalance is needed

pub fn can_rebalance(&self) -> bool

Check if enough time has passed since last rebalance

pub fn trigger_rebalance( &self, trigger: RebalanceTrigger, ) -> Result<RebalancePlan>

Trigger a rebalance operation

pub fn create_move_plan(&self) -> Result<Vec<ShardMove>>

Plan the shard movements

pub fn start_execution(&self) -> Result<()>

Start executing the rebalance plan

pub fn start_move(&self, move_id: &str) -> Result<()>

Start a specific move

pub fn update_move_progress( &self, move_id: &str, bytes_transferred: u64, total_bytes: u64, ) -> Result<()>

Update move progress

pub fn advance_to_verify(&self, move_id: &str) -> Result<()>

Advance move to verification stage

pub fn advance_to_routing(&self, move_id: &str) -> Result<()>

Advance move to routing update stage

pub fn complete_move(&self, move_id: &str) -> Result<()>

Complete a move

pub fn fail_move(&self, move_id: &str, error: &str) -> Result<()>

Fail a move

pub fn cancel(&self) -> Result<()>

Cancel the current rebalance

pub fn is_cancelled(&self) -> bool

Check if rebalance was cancelled

pub fn get_plan(&self) -> Option<RebalancePlan>

Get the current rebalance plan

pub fn get_queued_moves(&self) -> Vec<String>

Get moves that are ready to start

pub fn get_active_moves(&self) -> Vec<ShardMove>

Get currently executing moves

pub fn get_stats(&self) -> RebalanceStats

Get rebalance statistics

pub fn get_node_load(&self, node_id: &str) -> Option<NodeLoad>

Get node load information

pub fn get_all_node_loads(&self) -> Vec<NodeLoad>

Get all node loads

pub fn on_node_joined( &self, node_id: &str, capacity: u64, ) -> Result<Option<RebalancePlan>>

Handle node join event

pub fn on_node_left(&self, node_id: &str) -> Result<Option<RebalancePlan>>

Handle node leave event