ic-stable-memory 0.4.4

Internet Computer's stable memory collections and tools
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

use crate::collections::hash_map::SHashMap;
use crate::encoding::AsFixedSizeBytes;
use crate::primitive::s_ref::SRef;
use crate::primitive::StableType;
use std::hash::Hash;

pub struct SHashMapIter<
    'a,
    K: StableType + AsFixedSizeBytes + Hash + Eq,
    V: StableType + AsFixedSizeBytes,
> {
    map: &'a SHashMap<K, V>,
    i: usize,
}

impl<'a, K: StableType + AsFixedSizeBytes + Hash + Eq, V: StableType + AsFixedSizeBytes>
    SHashMapIter<'a, K, V>
{
    pub fn new(map: &'a SHashMap<K, V>) -> Self {
        Self { map, i: 0 }
    }
}

impl<'a, K: StableType + AsFixedSizeBytes + Eq + Hash, V: StableType + AsFixedSizeBytes> Iterator
    for SHashMapIter<'a, K, V>
{
    type Item = (SRef<'a, K>, SRef<'a, V>);

    fn next(&mut self) -> Option<Self::Item> {
        if self.map.is_empty() {
            return None;
        }

        loop {
            if self.i == self.map.capacity() {
                break None;
            }

            if let Some(k) = self.map.get_key(self.i) {
                let v = self.map.get_val(self.i);

                self.i += 1;

                return Some((k, v));
            }

            self.i += 1;
        }
    }
}