Skip to content

Commit

Permalink
Fix StoreRatio fields visibility and improve documentation
Browse files Browse the repository at this point in the history
  • Loading branch information
ia0 committed Oct 29, 2020
1 parent 8e22fbd commit 5ce9194
Show file tree
Hide file tree
Showing 2 changed files with 29 additions and 11 deletions.
22 changes: 13 additions & 9 deletions libraries/persistent_store/src/model.rs
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ use std::collections::{HashMap, HashSet};
#[derive(Clone, Debug)]
pub struct StoreModel {
/// Represents the content of the store.
map: HashMap<usize, Box<[u8]>>,
content: HashMap<usize, Box<[u8]>>,

/// The modeled storage configuration.
format: Format,
Expand Down Expand Up @@ -53,13 +53,13 @@ pub enum StoreOperation {
impl StoreModel {
/// Creates an empty model for a given storage configuration.
pub fn new(format: Format) -> StoreModel {
let map = HashMap::new();
StoreModel { map, format }
let content = HashMap::new();
StoreModel { content, format }
}

/// Returns the modeled content.
pub fn map(&self) -> &HashMap<usize, Box<[u8]>> {
&self.map
pub fn content(&self) -> &HashMap<usize, Box<[u8]>> {
&self.content
}

/// Returns the storage configuration.
Expand All @@ -79,7 +79,7 @@ impl StoreModel {
/// Returns the capacity according to the model.
pub fn capacity(&self) -> StoreRatio {
let total = self.format.total_capacity();
let used: usize = self.map.values().map(|x| self.entry_size(x)).sum();
let used: usize = self.content.values().map(|x| self.entry_size(x)).sum();
StoreRatio { used, total }
}

Expand All @@ -100,11 +100,15 @@ impl StoreModel {
}
// Fail if there is not enough capacity.
let capacity = match updates.len() {
// An empty transaction doesn't consume anything.
0 => 0,
// Transactions with a single update are optimized by avoiding a marker entry.
1 => match &updates[0] {
StoreUpdate::Insert { value, .. } => self.entry_size(value),
// Transactions with a single update which is a removal don't consume anything.
StoreUpdate::Remove { .. } => 0,
},
// A transaction consumes one word for the marker entry in addition to its updates.
_ => 1 + updates.iter().map(|x| self.update_size(x)).sum::<usize>(),
};
if self.capacity().remaining() < capacity {
Expand All @@ -114,10 +118,10 @@ impl StoreModel {
for update in updates {
match update {
StoreUpdate::Insert { key, value } => {
self.map.insert(key, value.into_boxed_slice());
self.content.insert(key, value.into_boxed_slice());
}
StoreUpdate::Remove { key } => {
self.map.remove(&key);
self.content.remove(&key);
}
}
}
Expand All @@ -129,7 +133,7 @@ impl StoreModel {
if min_key > self.format.max_key() {
return Err(StoreError::InvalidArgument);
}
self.map.retain(|&k, _| k < min_key);
self.content.retain(|&k, _| k < min_key);
Ok(())
}

Expand Down
18 changes: 16 additions & 2 deletions libraries/persistent_store/src/store.rs
Original file line number Diff line number Diff line change
Expand Up @@ -69,18 +69,32 @@ pub type StoreResult<T> = Result<T, StoreError>;
///
/// This is used for the [capacity] and [lifetime] metrics. Those metrics are measured in words.
///
/// # Invariant
///
/// - The used value does not exceed the total: `used <= total`.
///
/// [capacity]: struct.Store.html#method.capacity
/// [lifetime]: struct.Store.html#method.lifetime
#[derive(Copy, Clone, PartialEq, Eq)]
pub struct StoreRatio {
/// How much of the metric is used.
pub used: usize,
pub(crate) used: usize,

/// How much of the metric can be used at most.
pub total: usize,
pub(crate) total: usize,
}

impl StoreRatio {
/// How much of the metric is used.
pub fn used(self) -> usize {
self.used
}

/// How much of the metric can be used at most.
pub fn total(self) -> usize {
self.total
}

/// How much of the metric is remaining.
pub fn remaining(self) -> usize {
self.total - self.used
Expand Down

0 comments on commit 5ce9194

Please sign in to comment.