Impl Drop for los allocator

wenyuzhao · Jul 1, 2024 · 1525772 · 1525772
1 parent 6f29430
commit 1525772
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/mallockit/src/space/large_object_space.rs b/mallockit/src/space/large_object_space.rs
@@ -168,3 +168,15 @@ where
         }
     }
 }
+
+impl<S: PageSize, const MAX_CACHEABLE_SIZE: usize, const THRESHOLD_SLOP: usize> Drop
+    for LargeObjectAllocator<S, MAX_CACHEABLE_SIZE, THRESHOLD_SLOP>
+where
+    [(); bins::<S>(MAX_CACHEABLE_SIZE)]: Sized,
+{
+    fn drop(&mut self) {
+        if Self::CACHE_ENABLED {
+            self.clear_bins();
+        }
+    }
+}