kyu-delta 0.3.0

Conflict-free idempotent delta batches for KyuGraph ingestion
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119

/// Statistics returned after processing a `DeltaBatch`.
#[derive(Clone, Debug, Default, PartialEq, Eq)]
pub struct DeltaStats {
    pub nodes_created: u64,
    pub nodes_updated: u64,
    pub nodes_deleted: u64,
    pub edges_created: u64,
    pub edges_updated: u64,
    pub edges_deleted: u64,
    pub total_deltas: u64,
    pub elapsed_micros: u64,
}

impl DeltaStats {
    pub fn total_mutations(&self) -> u64 {
        self.nodes_created
            + self.nodes_updated
            + self.nodes_deleted
            + self.edges_created
            + self.edges_updated
            + self.edges_deleted
    }

    /// Merge stats from another batch (additive).
    pub fn merge(&mut self, other: &DeltaStats) {
        self.nodes_created += other.nodes_created;
        self.nodes_updated += other.nodes_updated;
        self.nodes_deleted += other.nodes_deleted;
        self.edges_created += other.edges_created;
        self.edges_updated += other.edges_updated;
        self.edges_deleted += other.edges_deleted;
        self.total_deltas += other.total_deltas;
        self.elapsed_micros += other.elapsed_micros;
    }
}

impl std::fmt::Display for DeltaStats {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(
            f,
            "nodes: +{}/~{}/{}, edges: +{}/~{}/{}, total: {}, {}us",
            self.nodes_created,
            self.nodes_updated,
            self.nodes_deleted,
            self.edges_created,
            self.edges_updated,
            self.edges_deleted,
            self.total_deltas,
            self.elapsed_micros,
        )
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn default_is_zero() {
        let s = DeltaStats::default();
        assert_eq!(s.total_mutations(), 0);
        assert_eq!(s.total_deltas, 0);
    }

    #[test]
    fn total_mutations() {
        let s = DeltaStats {
            nodes_created: 3,
            nodes_updated: 2,
            nodes_deleted: 1,
            edges_created: 5,
            edges_updated: 0,
            edges_deleted: 1,
            total_deltas: 12,
            elapsed_micros: 100,
        };
        assert_eq!(s.total_mutations(), 12);
    }

    #[test]
    fn merge_additive() {
        let mut a = DeltaStats {
            nodes_created: 3,
            ..Default::default()
        };
        let b = DeltaStats {
            nodes_created: 2,
            edges_created: 5,
            ..Default::default()
        };
        a.merge(&b);
        assert_eq!(a.nodes_created, 5);
        assert_eq!(a.edges_created, 5);
    }

    #[test]
    fn display_format() {
        let s = DeltaStats {
            nodes_created: 1,
            edges_created: 2,
            total_deltas: 3,
            elapsed_micros: 50,
            ..Default::default()
        };
        let d = format!("{s}");
        assert!(d.contains("nodes: +1"));
        assert!(d.contains("edges: +2"));
        assert!(d.contains("50us"));
    }

    #[test]
    fn total_deltas_counted() {
        let s = DeltaStats {
            total_deltas: 42,
            ..Default::default()
        };
        assert_eq!(s.total_deltas, 42);
    }
}