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

use crate::{Context, DeepSizeOf};

/// For use on types defined in external crates
/// with known heap sizes.
///
/// Stolen from `heapsize` crate
#[macro_export]
macro_rules! known_deep_size(
    ($size:expr, $($type:ty),+) => (
        $(
            impl $crate::DeepSizeOf for $type {
                #[inline(always)]
                fn deep_size_of_children(&self, _: &mut $crate::Context) -> usize {
                    $size
                }
            }
        )+
    );
    ($size:expr, $($type:ident<$($gen:ident),+>),+) => (
        $(
            impl<$($gen: $crate::HeapSizeOf),+> $crate::DeepSizeOf for $type<$($gen),+> {
                #[inline(always)]
                fn deep_size_of_children(&self, _: &mut $crate::Context) -> usize {
                    $size
                }
            }
        )+
    );
);

known_deep_size!(0, bool, char, str);
known_deep_size!(0, u8, u16, u32, u64, usize);
known_deep_size!(0, i8, i16, i32, i64, isize);
known_deep_size!(0, f32, f64);
known_deep_size!(0, ());
known_deep_size!(
    0,
    std::sync::atomic::AtomicBool,
    std::sync::atomic::AtomicIsize,
    std::sync::atomic::AtomicUsize
);

impl<T: ?Sized> DeepSizeOf for std::marker::PhantomData<T> {
    fn deep_size_of_children(&self, _: &mut Context) -> usize {
        0
    }
}

impl DeepSizeOf for String {
    fn deep_size_of_children(&self, context: &mut Context) -> usize {
        self.as_str().deep_size_of_children(context)
    }
}