Optional implementation of DeepSizeOf for BoomHashMap

Cargo.toml

@@ -18,6 +18,7 @@ rayon = { version=">=1.0", optional = true }
 crossbeam-utils = { version=">=0.7.2, <0.9", optional = true }
 wyhash = ">=0.3, <=0.5"
 log = "0.4.*"
+deepsize = { version="0.2.0", optional = true }
 
 [dev-dependencies]
 quickcheck = "1.0.2"

src/bitvector.rs

@@ -30,6 +30,8 @@ use std::sync::atomic::{AtomicU64, Ordering};
 
 #[cfg(feature = "serde")]
 use serde::{self, Deserialize, Serialize};
+#[cfg(feature = "deepsize")]
+use deepsize::DeepSizeOf;
 
 #[cfg(feature = "parallel")]
 type Word = AtomicU64;
@@ -384,6 +386,13 @@ impl BitVector {
     }
 }
 
+#[cfg(feature = "deepsize")]
+impl DeepSizeOf for BitVector {
+    fn deep_size_of_children(&self, context: &mut deepsize::Context) -> usize {
+        self.bits.deep_size_of_children(context) + self.vector.deep_size_of_children(context)
+    }
+}
+
 /// Iterator for BitVector
 pub struct BitVectorIter<'a> {
     iter: ::std::slice::Iter<'a, Word>,

src/hashmap.rs

@@ -3,6 +3,9 @@
 #[cfg(feature = "serde")]
 use serde::{self, Deserialize, Serialize};
 
+#[cfg(feature = "deepsize")]
+use deepsize::DeepSizeOf;
+
 use crate::Mphf;
 use std::borrow::Borrow;
 use std::fmt::Debug;
@@ -131,6 +134,19 @@ where
     }
 }
 
+#[cfg(feature = "deepsize")]
+impl<K, D> DeepSizeOf for BoomHashMap<K, D>
+where
+    K: Hash + Debug + PartialEq + DeepSizeOf,
+    D: Debug + DeepSizeOf,
+{
+    fn deep_size_of_children(&self, context: &mut deepsize::Context) -> usize {
+        self.mphf.deep_size_of_children(context)
+            + self.keys.deep_size_of_children(context)
+            + self.values.deep_size_of_children(context)
+    }
+}
+
 impl<K, D> core::iter::FromIterator<(K, D)> for BoomHashMap<K, D>
 where
     K: Hash + Debug + PartialEq,

src/lib.rs

@@ -51,6 +51,8 @@ use std::sync::{Arc, Mutex};
 
 #[cfg(feature = "serde")]
 use serde::{self, Deserialize, Serialize};
+#[cfg(feature = "deepsize")]
+use deepsize::DeepSizeOf;
 
 #[inline]
 fn fold(v: u64) -> u32 {
@@ -408,6 +410,13 @@ impl<T: Hash + Debug + Sync + Send> Mphf<T> {
     }
 }
 
+#[cfg(feature = "deepsize")]
+impl<T> DeepSizeOf for Mphf<T> {
+    fn deep_size_of_children(&self, context: &mut deepsize::Context) -> usize {
+        self.bitvecs.deep_size_of_children(context) + self.phantom.deep_size_of_children(context)
+    }
+}
+
 struct Context {
     size: u64,
     seed: u64,