mab122
/
radicle-heartwood-lfs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

crdt: Implement `LWWMap` in terms of `GMap` 

Signed-off-by: Alexis Sellier <alexis@radicle.xyz>
This commit is contained in:
Alexis Sellier 2022-12-01 14:27:21 +01:00
parent bcef784ad4
commit ff79347eb0
No known key found for this signature in database
2 changed files with 12 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
radicle-crdt/src/lwwmap.rs
View File

@ -1,6 +1,4 @@
use std::collections::btree_map::Entry; use crate::gmap::GMap;
use std::collections::BTreeMap;
use crate::lwwreg::LWWReg; use crate::lwwreg::LWWReg;
use crate::{clock, Semilattice}; use crate::{clock, Semilattice};
@ -10,13 +8,13 @@ use crate::{clock, Semilattice};
/// the "add" takes precedence over the "remove". /// the "add" takes precedence over the "remove".
#[derive(Debug, Clone, PartialEq, Eq)] #[derive(Debug, Clone, PartialEq, Eq)]
pub struct LWWMap<K, V, C = clock::Lamport> { pub struct LWWMap<K, V, C = clock::Lamport> {
inner: BTreeMap<K, LWWReg<Option<V>, C>>, inner: GMap<K, LWWReg<Option<V>, C>>,
} }
impl<K: Ord, V: Semilattice, C: PartialOrd + Ord> LWWMap<K, V, C> { impl<K: Ord, V: Semilattice, C: PartialOrd + Ord> LWWMap<K, V, C> {
pub fn singleton(key: K, value: V, clock: C) -> Self { pub fn singleton(key: K, value: V, clock: C) -> Self {
Self { Self {
inner: BTreeMap::from_iter([(key, LWWReg::new(Some(value), clock))]), inner: GMap::singleton(key, LWWReg::new(Some(value), clock)),
} }
} }
@ -29,25 +27,11 @@ impl<K: Ord, V: Semilattice, C: PartialOrd + Ord> LWWMap<K, V, C> {
} }
pub fn insert(&mut self, key: K, value: V, clock: C) { pub fn insert(&mut self, key: K, value: V, clock: C) {
match self.inner.entry(key) { self.inner.insert(key, LWWReg::new(Some(value), clock));
Entry::Occupied(mut e) => {
e.get_mut().set(Some(value), clock);
}
Entry::Vacant(e) => {
e.insert(LWWReg::new(Some(value), clock));
}
}
} }
pub fn remove(&mut self, key: K, clock: C) { pub fn remove(&mut self, key: K, clock: C) {
match self.inner.entry(key) { self.inner.insert(key, LWWReg::new(None, clock));
Entry::Occupied(mut e) => {
e.get_mut().set(None, clock);
}
Entry::Vacant(e) => {
e.insert(LWWReg::new(None, clock));
}
}
} }
pub fn contains_key(&self, key: &K) -> bool { pub fn contains_key(&self, key: &K) -> bool {
@ -76,12 +60,12 @@ impl<K: Ord, V: Semilattice, C: PartialOrd + Ord> LWWMap<K, V, C> {
impl<K, V, C> Default for LWWMap<K, V, C> { impl<K, V, C> Default for LWWMap<K, V, C> {
fn default() -> Self { fn default() -> Self {
Self { Self {
inner: BTreeMap::default(), inner: GMap::default(),
} }
} }
} }
impl<K: Ord, V: Semilattice + PartialOrd + Eq, C: Ord> FromIterator<(K, V, C)> for LWWMap<K, V, C> { impl<K: Ord, V: Semilattice, C: Ord> FromIterator<(K, V, C)> for LWWMap<K, V, C> {
fn from_iter<I: IntoIterator<Item = (K, V, C)>>(iter: I) -> Self { fn from_iter<I: IntoIterator<Item = (K, V, C)>>(iter: I) -> Self {
let mut map = LWWMap::default(); let mut map = LWWMap::default();
for (k, v, c) in iter.into_iter() { for (k, v, c) in iter.into_iter() {
@ -91,7 +75,7 @@ impl<K: Ord, V: Semilattice + PartialOrd + Eq, C: Ord> FromIterator<(K, V, C)> f
} }
} }
impl<K: Ord, V: Semilattice + PartialOrd + Eq, C: Ord> Extend<(K, V, C)> for LWWMap<K, V, C> { impl<K: Ord, V: Semilattice, C: Ord> Extend<(K, V, C)> for LWWMap<K, V, C> {
fn extend<I: IntoIterator<Item = (K, V, C)>>(&mut self, iter: I) { fn extend<I: IntoIterator<Item = (K, V, C)>>(&mut self, iter: I) {
for (k, v, c) in iter.into_iter() { for (k, v, c) in iter.into_iter() {
self.insert(k, v, c); self.insert(k, v, c);
@ -102,20 +86,11 @@ impl<K: Ord, V: Semilattice + PartialOrd + Eq, C: Ord> Extend<(K, V, C)> for LWW
impl<K, V, C> Semilattice for LWWMap<K, V, C> impl<K, V, C> Semilattice for LWWMap<K, V, C>
where where
K: Ord, K: Ord,
V: Semilattice + PartialOrd + Eq, V: Semilattice,
C: Ord + Default, C: Ord,
{ {
fn merge(&mut self, other: Self) { fn merge(&mut self, other: Self) {
for (k, v) in other.inner.into_iter() { self.inner.merge(other.inner);
match self.inner.entry(k) {
Entry::Occupied(mut e) => {
e.get_mut().merge(v);
}
Entry::Vacant(e) => {
e.insert(v);
}
}
}
} }
} }

2
radicle-crdt/src/lwwreg.rs
View File

@ -66,7 +66,7 @@ impl<T: Default, C: Default + Bounded> Default for LWWReg<T, C> {
impl<T, C> Semilattice for LWWReg<T, C> impl<T, C> Semilattice for LWWReg<T, C>
where where
T: PartialOrd + Semilattice, T: Semilattice,
C: PartialOrd, C: PartialOrd,
{ {
fn merge(&mut self, other: Self) { fn merge(&mut self, other: Self) {