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
use std::hash::Hash;
use map::{Map, BloomMap};
use Arena;
pub struct Set<'arena, I: 'arena> {
map: Map<'arena, I, ()>,
}
impl<'arena, I> Set<'arena, I>
where
I: Eq + Hash + Copy,
{
#[inline]
pub fn new() -> Self {
Set {
map: Map::new(),
}
}
#[inline]
pub fn insert(&self, arena: &'arena Arena, item: I) {
self.map.insert(arena, item, ());
}
#[inline]
pub fn contains(&self, item: I) -> bool {
self.map.contains_key(item)
}
#[inline]
pub fn is_empty(&self) -> bool {
self.map.is_empty()
}
#[inline]
pub fn clear(&self) {
self.map.clear()
}
}
pub struct BloomSet<'arena, I: 'arena> {
map: BloomMap<'arena, I, ()>,
}
impl<'arena, I> BloomSet<'arena, I>
where
I: Eq + Hash + Copy + AsRef<[u8]>,
{
#[inline]
pub fn new() -> Self {
BloomSet {
map: BloomMap::new(),
}
}
#[inline]
pub fn insert(&self, arena: &'arena Arena, item: I) {
self.map.insert(arena, item, ());
}
#[inline]
pub fn contains(&self, item: I) -> bool {
self.map.contains_key(item)
}
#[inline]
pub fn is_empty(&self) -> bool {
self.map.is_empty()
}
#[inline]
pub fn clear(&self) {
self.map.clear()
}
}
#[cfg(test)]
mod test {
use super::*;
#[test]
fn set() {
let arena = Arena::new();
let set = Set::new();
set.insert(&arena, "foo");
set.insert(&arena, "bar");
set.insert(&arena, "doge");
assert_eq!(set.contains("foo"), true);
assert_eq!(set.contains("bar"), true);
assert_eq!(set.contains("doge"), true);
assert_eq!(set.contains("moon"), false);
}
#[test]
fn bloom_set() {
let arena = Arena::new();
let set = BloomSet::new();
set.insert(&arena, "foo");
set.insert(&arena, "bar");
set.insert(&arena, "doge");
assert_eq!(set.contains("foo"), true);
assert_eq!(set.contains("bar"), true);
assert_eq!(set.contains("doge"), true);
assert_eq!(set.contains("moon"), false);
}
}