bitcoin-mem 0.1.19

traits for tracking dynamic and recursive memory usage
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

// ---------------- [ File: bitcoin-mem/src/incremental_dynamic_usage.rs ]
crate::ix!();

/// Trait for computing the additional heap allocation that **one more**
/// element would cause inside a container (mirrors the C++ memusage logic).
pub trait IncrementalDynamicUsage {
    fn incremental_dynamic_usage(&self) -> usize;
}

impl<K, S> IncrementalDynamicUsage for HashSet<K, S>
where
    S: BuildHasher,
{
    #[inline]
    fn incremental_dynamic_usage(&self) -> usize {
        malloc_usage(core::mem::size_of::<StlTreeNode<K>>())
    }
}

impl<K, V, S> IncrementalDynamicUsage for HashMap<K, V, S>
where
    S: BuildHasher,
{
    #[inline]
    fn incremental_dynamic_usage(&self) -> usize {
        malloc_usage(core::mem::size_of::<StlTreeNode<(K, V)>>())
    }
}

impl<T> IncrementalDynamicUsage for Arc<T>
where
    T: IncrementalDynamicUsage + ?Sized,
{
    #[inline]
    fn incremental_dynamic_usage(&self) -> usize {
        (**self).incremental_dynamic_usage()
    }
}