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

#![cfg(feature = "hashbrown")]

use crate::{std_impls::hashmap::estimate_hashmap_size, Context, SizeOf};
use hashbrown::{HashMap, HashSet};

impl<K, S> SizeOf for HashSet<K, S>
where
    K: SizeOf,
    S: SizeOf,
{
    fn size_of_children(&self, context: &mut Context) {
        if self.capacity() != 0 {
            let (total_bytes, used_bytes) =
                estimate_hashmap_size::<K, ()>(self.len(), self.capacity());

            context
                .add(used_bytes)
                .add_excess(total_bytes - used_bytes)
                .add_distinct_allocation();

            self.iter().for_each(|key| key.size_of_children(context));
        }

        self.hasher().size_of_children(context);
    }
}

impl<K, V, S> SizeOf for HashMap<K, V, S>
where
    K: SizeOf,
    V: SizeOf,
    S: SizeOf,
{
    fn size_of_children(&self, context: &mut Context) {
        if self.capacity() != 0 {
            let (total_bytes, used_bytes) =
                estimate_hashmap_size::<K, V>(self.len(), self.capacity());

            context
                .add(used_bytes)
                .add_excess(total_bytes - used_bytes)
                .add_distinct_allocation();

            self.iter().for_each(|(key, value)| {
                key.size_of_children(context);
                value.size_of_children(context);
            });
        }

        self.hasher().size_of_children(context);
    }
}