Add #[heapsize]/#[derive(HeapSizeOf)] plugin to auto-derive `HeapSizeOf` impls

(fixes #5914)

1 files changed, 28 insertions, 0 deletions

diff --git a/components/util/mem.rs b/components/util/mem.rs
index cd1b9834e48..175863ddb9e 100644
--- a/components/util/mem.rs
+++ b/components/util/mem.rs
@@ -158,3 +158,31 @@ impl<T> Drop for LinkedList2<T> {
     fn drop(&mut self) {}
 }
 
+/// For use on types defined in external crates
+/// with known heap sizes
+#[macro_export]
+macro_rules! known_heap_size(
+    ($size:expr, $($ty:ident),+) => (
+        $(
+            impl $crate::mem::HeapSizeOf for $ty {
+                #[inline]
+                fn heap_size_of_children(&self) -> usize {
+                    $size
+                }
+            }
+        )+
+    );
+    ($size: expr, $ty:ident<$($gen:ident),+>) => (
+        impl<$($gen),+> $crate::mem::HeapSizeOf for $ty<$($gen),+> {
+            #[inline]
+            fn heap_size_of_children(&self) -> usize {
+                $size
+            }
+        }
+    );
+);
+
+
+known_heap_size!(0, u8, u16, u32, u64, usize);
+known_heap_size!(0, i8, i16, i32, i64, isize);
+known_heap_size!(0, bool);
\ No newline at end of file