object-rainbow-apply 0.0.0-a.1

asynchronous map trait for object-rainbow-history
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

use object_rainbow::{InlineOutput, ReflessObject, Traversible};
use object_rainbow_trie::{TrieMap, TrieSet};

use crate::Apply;

impl<K: ReflessObject, V: 'static + Send + Sync + Clone> Apply<(Option<V>, K)> for TrieMap<K, V>
where
    Option<V>: Traversible + InlineOutput,
{
    type Output = Option<(V, K)>;

    async fn apply(
        &mut self,
        (value, key): (Option<V>, K),
    ) -> object_rainbow::Result<Self::Output> {
        if let Some(value) = value {
            self.insert(&key, value).await
        } else {
            self.remove(&key).await
        }
        .map(|o| o.map(|value| (value, key)))
    }
}

impl<K: ReflessObject, V: 'static + Send + Sync + Clone> Apply<(V, K)> for TrieMap<K, V>
where
    Option<V>: Traversible + InlineOutput,
{
    type Output = Option<(V, K)>;

    async fn apply(&mut self, (value, key): (V, K)) -> object_rainbow::Result<Self::Output> {
        self.insert(&key, value)
            .await
            .map(|o| o.map(|value| (value, key)))
    }
}

/// `true` represents removal, `false` represents insertion to keep layout equivalence.
impl<T: ReflessObject> Apply<(bool, T)> for TrieSet<T> {
    type Output = bool;

    async fn apply(&mut self, (remove, value): (bool, T)) -> object_rainbow::Result<Self::Output> {
        Ok(if remove {
            !self.remove(&value).await?
        } else {
            self.insert(&value).await?
        })
    }
}

impl<T: ReflessObject> Apply<(Option<()>, T)> for TrieSet<T> {
    type Output = Option<T>;

    async fn apply(
        &mut self,
        (target, value): (Option<()>, T),
    ) -> object_rainbow::Result<Self::Output> {
        Ok(if target.is_some() {
            !self.insert(&value).await?
        } else {
            self.remove(&value).await?
        }
        .then_some(value))
    }
}

impl<T: ReflessObject> Apply<T> for TrieSet<T> {
    type Output = Option<T>;

    async fn apply(&mut self, value: T) -> object_rainbow::Result<Self::Output> {
        Ok((!self.insert(&value).await?).then_some(value))
    }
}

impl<T: ReflessObject> Apply<((), T)> for TrieSet<T> {
    type Output = Option<T>;

    async fn apply(&mut self, ((), value): ((), T)) -> object_rainbow::Result<Self::Output> {
        self.apply(value).await
    }
}