mars-shard · topology

#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Serialize, Deserialize)]
pub enum ShardStatus {
    /// Shard is operational and accepting reads/writes.
    Active,
    /// Shard is in the process of splitting into two.
    Splitting,
    /// Shard is being merged with another shard.
    Merging,
    /// Shard data is being migrated to another cluster.
    Migrating,
    /// Shard is not available.
    Offline,
}

#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct ShardInfo {
/// Unique shard identifier.

pub shard_id: String,
/// Cluster this shard belongs to.

pub cluster_id: String,
/// Region where this shard is deployed.

pub region: String,
/// Current operational status.

pub status: ShardStatus,
/// Capacity utilization as a fraction (0.0 to 1.0).

pub capacity_pct: f64,
/// Number of entries stored in this shard.

pub entry_count: u64,
/// IDs of shadow (replica) shards.

#[serde(default, skip_serializing_if = "Vec::is_empty")]
pub shadow_shard_ids: Vec<String>,
/// Primary node responsible for this shard.

#[serde(skip_serializing_if = "Option::is_none")]
pub primary_node: Option<String>,
/// When this shard was created.

pub created_at: chrono::DateTime<chrono::Utc>
}

pub struct ShardTopology {

}

pub fn new() -> Self;

pub fn add_shard(&mut self, info: ShardInfo) -> Result<(), ShardError>;

pub fn remove_shard(&mut self, shard_id: &str) -> Result<(), ShardError>;

pub fn get_shard(&self, shard_id: &str) -> Option<&ShardInfo>;

pub fn get_shard_mut(&mut self, shard_id: &str) -> Option<&mut ShardInfo>;

pub fn shards_in_cluster(&self, cluster_id: &str) -> Vec<&ShardInfo>;

pub fn shards_in_region(&self, region: &str) -> Vec<&ShardInfo>;

pub fn active_shards(&self) -> Vec<&ShardInfo>;

pub fn overloaded_shards(&self, threshold: f64) -> Vec<&ShardInfo>;

pub fn underloaded_shards(&self, threshold: f64) -> Vec<&ShardInfo>;

pub fn total_entries(&self) -> u64;

pub fn shard_count(&self) -> usize;

pub fn cluster_count(&self) -> usize;

pub fn region_count(&self) -> usize;