kivis 0.2.2

Type-safe database schema generation for Rust with support for custom storage backends, automatic indexing, and layered cache architectures
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

// Example showing how to use the AtomicStorage trait
// This example requires the "atomic" feature to be enabled

#[cfg(feature = "atomic")]
fn atomic_storage_example() {
    use kivis::{AtomicStorage, Storage};
    use std::{cmp::Reverse, collections::BTreeMap, fmt::Display, ops::Range};

    // Define a custom error type
    #[derive(Debug, PartialEq, Eq)]
    struct MyError(String);

    impl Display for MyError {
        fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
            write!(f, "MyError: {}", self.0)
        }
    }

    // A storage implementation that supports atomic operations
    struct MyAtomicStorage {
        data: BTreeMap<Reverse<Vec<u8>>, Vec<u8>>,
    }

    impl MyAtomicStorage {
        fn new() -> Self {
            Self {
                data: BTreeMap::new(),
            }
        }
    }

    // First implement the basic Storage trait
    impl Storage for MyAtomicStorage {
        type StoreError = MyError;

        fn insert(&mut self, key: Vec<u8>, value: Vec<u8>) -> Result<(), Self::StoreError> {
            self.data.insert(Reverse(key), value);
            Ok(())
        }

        fn get(&self, key: Vec<u8>) -> Result<Option<Vec<u8>>, Self::StoreError> {
            Ok(self.data.get(&Reverse(key)).cloned())
        }

        fn remove(&mut self, key: Vec<u8>) -> Result<Option<Vec<u8>>, Self::StoreError> {
            Ok(self.data.remove(&Reverse(key)))
        }

        fn iter_keys(
            &self,
            range: Range<Vec<u8>>,
        ) -> Result<impl Iterator<Item = Result<Vec<u8>, Self::StoreError>>, Self::StoreError>
        {
            let reverse_range = Reverse(range.end)..Reverse(range.start);
            let iter = self.data.range(reverse_range);
            Ok(iter.map(|(k, _v)| Ok(k.0.clone())))
        }
    }

    // Then implement the AtomicStorage trait
    impl AtomicStorage for MyAtomicStorage {
        fn batch_mixed(
            &mut self,
            inserts: Vec<(Vec<u8>, Vec<u8>)>,
            removes: Vec<Vec<u8>>,
        ) -> Result<Vec<Option<Vec<u8>>>, Self::StoreError> {
            let mut removed = Vec::new();
            for (key, value) in inserts {
                self.data.insert(Reverse(key), value);
            }
            for key in removes {
                removed.push(self.data.remove(&Reverse(key)));
            }
            Ok(removed)
        }

        // batch_mixed has a default implementation that calls batch_remove then batch_insert
        // You can override it if you need different behavior
    }

    // Usage example
    let mut storage = MyAtomicStorage::new();

    // Perform atomic batch operations
    let operations = vec![
        (b"user1".to_vec(), b"Alice".to_vec()),
        (b"user2".to_vec(), b"Bob".to_vec()),
        (b"user3".to_vec(), b"Charlie".to_vec()),
    ];

    // All insertions happen atomically
    storage.batch_mixed(operations, Vec::new()).unwrap();

    // All removals happen atomically
    let keys_to_remove = vec![b"user1".to_vec(), b"user3".to_vec()];
    let removed = storage.batch_mixed(Vec::new(), keys_to_remove).unwrap();

    println!("Removed values: {:?}", removed);
}

fn main() {
    #[cfg(feature = "atomic")]
    atomic_storage_example();

    #[cfg(not(feature = "atomic"))]
    println!("This example requires the 'atomic' feature to be enabled. Run with: cargo run --example atomic_storage --features atomic");
}