propmaster 0.1.0

Properties and configurations for rust
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
120
121
122
123
124
125
126
127
128
129

use std::collections::HashMap;

use crate::property::Property;
pub struct PropertyMap {
    stored_num: usize,
    properties: HashMap<String, Property>,
}

impl PropertyMap {
    pub fn new() -> Self {
        PropertyMap {
            stored_num: 0,
            properties: HashMap::new(),
        }
    }

    pub fn stored_num(&self) -> usize {
        self.stored_num
    }

    pub fn get(&self, name: &str) -> Option<&Property> {
        self.properties.get(name)
    }

    pub fn set(&mut self, name: &str, property: Property) -> bool {
        if self.properties.contains_key(name) {
            false
        } else {
            self.stored_num += 1;
            self.properties.insert(name.to_string(), property);
            true
        }
    }

    /**
     * Returns the old property if it existed, and a boolean indicating whether it existed.
     * and was upserted.
     */
    pub fn set_or_upsert(
        &mut self,
        name: &str,
        property: Property,
    ) -> (Option<Property>, bool) {
        if self.properties.contains_key(name) {
            (self.properties.insert(name.to_string(), property), true)
        } else {
            self.stored_num += 1;
            (self.properties.insert(name.to_string(), property), false)
        }
    }

    pub fn remove(&mut self, name: &str) -> bool {
        if self.properties.contains_key(name) {
            self.stored_num -= 1;
            self.properties.remove(name);
            true
        } else {
            false
        }
    }

    pub fn clear(&mut self) {
        self.properties.clear();
        self.stored_num = 0;
    }

    pub fn iter(&self) -> std::collections::hash_map::Iter<String, Property> {
        self.properties.iter()
    }

    pub fn iter_mut(&mut self) -> std::collections::hash_map::IterMut<String, Property> {
        self.properties.iter_mut()
    }
}

impl<'a> Default for PropertyMap {
    fn default() -> Self {
        Self::new()
    }
}

#[cfg(test)]
mod test {
    use crate::property::PropertyValue;

    use super::*;

    #[test]
    fn propmap_insert() {
        let mut propmap = PropertyMap::new();
        let prop = Property::new(PropertyValue::Integer(1));
        assert!(propmap.set("test", Property::new(PropertyValue::Integer(1))));
        assert_eq!(propmap.stored_num(), 1);
        assert!(!propmap.set("test", Property::new(PropertyValue::Integer(1))));
        assert_eq!(propmap.stored_num(), 1);
    }

    #[test]
    fn propmap_upsert() {
        let mut propmap = PropertyMap::new();
        let prop = Property::new(PropertyValue::Integer(1));
        assert!(!propmap.set_or_upsert("test", Property::new(PropertyValue::Integer(1))).1);
        assert_eq!(propmap.stored_num(), 1);
        assert!(propmap.set_or_upsert("test", Property::new(PropertyValue::Integer(1))).1);
        assert_eq!(propmap.stored_num(), 1);
    }

    #[test]
    fn propmap_remove() {
        let mut propmap = PropertyMap::new();
        let prop = Property::new(PropertyValue::Integer(1));
        assert!(propmap.set("test", prop));
        assert_eq!(propmap.stored_num(), 1);
        assert!(propmap.remove("test"));
        assert_eq!(propmap.stored_num(), 0);
        assert!(!propmap.remove("test"));
        assert_eq!(propmap.stored_num(), 0);
    }

    #[test]
    fn propmap_clear() {
        let mut propmap = PropertyMap::new();
        let prop = Property::new(PropertyValue::Integer(1));
        assert!(propmap.set("test", prop));
        assert_eq!(propmap.stored_num(), 1);
        propmap.clear();
        assert_eq!(propmap.stored_num(), 0);
    }
}