call shrink_to_fit() on roaring bitmaps (#27)

* bitmap shrink_to_fit * one more
spiraldb · Feb 27, 2024 · d3ef519 · d3ef519
1 parent 2a901df
commit d3ef519
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/vortex-roaring/src/boolean/compress.rs b/vortex-roaring/src/boolean/compress.rs
@@ -48,6 +48,7 @@ pub fn roaring_encode(bool_array: &BoolArray) -> RoaringBoolArray {
             .map(|(i, _)| i as u32),
     );
     bitmap.run_optimize();
+    bitmap.shrink_to_fit();
 
     RoaringBoolArray::new(bitmap, bool_array.buffer().len())
 }
diff --git a/vortex-roaring/src/integer/compress.rs b/vortex-roaring/src/integer/compress.rs
@@ -73,5 +73,6 @@ fn roaring_encode_primitive<T: NumCast + NativePType>(values: &[T]) -> RoaringIn
     let mut bitmap = Bitmap::new();
     bitmap.extend(values.iter().map(|i| i.to_u32().unwrap()));
     bitmap.run_optimize();
+    bitmap.shrink_to_fit();
     RoaringIntArray::new(bitmap, T::PTYPE)
 }