use super::RegisterCmd;
use crate::types::{Error, Result};
use serde::{Deserialize, Serialize};
use std::collections::BTreeMap;
use xor_name::XorName;
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct DataExchange {
pub chunk_data: ChunkDataExchange,
pub reg_data: RegisterDataExchange,
}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct ChunkDataExchange {
pub adult_levels: BTreeMap<XorName, StorageLevel>,
}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct RegisterDataExchange(pub BTreeMap<XorName, Vec<RegisterCmd>>);
#[derive(Copy, Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Serialize, Deserialize)]
pub struct StorageLevel(u8);
impl StorageLevel {
pub const MAX: u8 = 10;
pub fn zero() -> Self {
Self(0)
}
pub fn from(value: u8) -> Result<Self> {
if value > Self::MAX {
Err(Error::OutOfRange)
} else {
Ok(Self(value))
}
}
pub fn next(&self) -> Result<StorageLevel> {
StorageLevel::from(self.0 + 1)
}
pub fn previous(&self) -> Result<StorageLevel> {
if self.0 == 0 {
Err(Error::OutOfRange)
} else {
StorageLevel::from(self.0 - 1)
}
}
pub fn value(&self) -> u8 {
self.0
}
}