summaryrefslogtreecommitdiffstats
path: root/third_party/rust/linked-hash-map/src/heapsize.rs
diff options
context:
space:
mode:
Diffstat (limited to 'third_party/rust/linked-hash-map/src/heapsize.rs')
-rw-r--r--third_party/rust/linked-hash-map/src/heapsize.rs51
1 files changed, 51 insertions, 0 deletions
diff --git a/third_party/rust/linked-hash-map/src/heapsize.rs b/third_party/rust/linked-hash-map/src/heapsize.rs
new file mode 100644
index 0000000000..59f46297b2
--- /dev/null
+++ b/third_party/rust/linked-hash-map/src/heapsize.rs
@@ -0,0 +1,51 @@
+extern crate heapsize;
+
+use self::heapsize::{heap_size_of, HeapSizeOf};
+use std::hash::{BuildHasher, Hash};
+
+use {KeyRef, LinkedHashMap, Node};
+
+impl<K> HeapSizeOf for KeyRef<K> {
+ fn heap_size_of_children(&self) -> usize {
+ 0
+ }
+}
+
+impl<K, V> HeapSizeOf for Node<K, V>
+where
+ K: HeapSizeOf,
+ V: HeapSizeOf,
+{
+ fn heap_size_of_children(&self) -> usize {
+ self.key.heap_size_of_children() + self.value.heap_size_of_children()
+ }
+}
+
+impl<K, V, S> HeapSizeOf for LinkedHashMap<K, V, S>
+where
+ K: HeapSizeOf + Hash + Eq,
+ V: HeapSizeOf,
+ S: BuildHasher,
+{
+ fn heap_size_of_children(&self) -> usize {
+ unsafe {
+ let mut size = self.map.heap_size_of_children();
+ for &value in self.map.values() {
+ size += (*value).heap_size_of_children();
+ size += heap_size_of(value as *const _ as *const _);
+ }
+
+ if !self.head.is_null() {
+ size += heap_size_of(self.head as *const _ as *const _);
+ }
+
+ let mut free = self.free;
+ while !free.is_null() {
+ size += heap_size_of(free as *const _ as *const _);
+ free = (*free).next
+ }
+
+ size
+ }
+ }
+}