hyphae 0.3.0

Reactive cells and runtime primitives for rship
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
82
83
84
85
86
87
88

use std::sync::{
    Arc,
    atomic::{AtomicUsize, Ordering},
};

use super::{CellValue, Watchable};
use crate::{
    cell::{Cell, CellImmutable, CellMutable},
    signal::Signal,
};

pub trait SkipExt<T>: Watchable<T> {
    #[track_caller]
    fn skip(&self, count: usize) -> Cell<T, CellImmutable>
    where
        T: CellValue,
        Self: Clone + Send + Sync + 'static,
    {
        let cell = Cell::<T, CellMutable>::new(self.get());
        let cell = if let Some(name) = self.name() {
            cell.with_name(format!("{}::skip", name))
        } else {
            cell
        };

        let to_skip = Arc::new(AtomicUsize::new(count));
        let weak = cell.downgrade();
        let guard = self.subscribe(move |signal| {
            if let Some(c) = weak.upgrade() {
                match signal {
                    Signal::Value(_) => {
                        let prev = to_skip.fetch_update(Ordering::SeqCst, Ordering::SeqCst, |n| {
                            if n > 0 { Some(n - 1) } else { None }
                        });
                        if prev.is_err() {
                            c.notify(signal.clone()); // Arc clone, no deep copy
                        }
                    }
                    Signal::Complete => c.notify(Signal::Complete),
                    Signal::Error(e) => c.notify(Signal::Error(e.clone())),
                }
            }
        });
        cell.own(guard);

        cell.lock()
    }
}

impl<T, W: Watchable<T>> SkipExt<T> for W {}

#[cfg(test)]
mod tests {
    use std::sync::{
        Arc,
        atomic::{AtomicU64, Ordering as AtomicOrdering},
    };

    use super::*;
    use crate::Mutable;

    #[test]
    fn test_skip_ignores_first_n() {
        let source = Cell::new(0u64);
        let skipped = source.skip(2);
        let count = Arc::new(AtomicU64::new(0));

        let c = count.clone();
        let _guard = skipped.subscribe(move |_| {
            c.fetch_add(1, AtomicOrdering::SeqCst);
        });

        // Watch always fires once immediately with current value
        assert_eq!(count.load(AtomicOrdering::SeqCst), 1);

        // First set - skipped (to_skip goes 2 -> 1 from initial watch, then 1 -> 0 here)
        source.set(1);
        assert_eq!(count.load(AtomicOrdering::SeqCst), 1);

        // Second set - now emits (to_skip is 0)
        source.set(2);
        assert_eq!(count.load(AtomicOrdering::SeqCst), 2);

        // Third set - emits
        source.set(3);
        assert_eq!(count.load(AtomicOrdering::SeqCst), 3);
    }
}