Expose memory usage in SingleSegmentIndexWriter (#1508)

quickwit-oss · Sep 7, 2022 · 4d634d6 · 4d634d6
1 parent 1f3d8ca
commit 4d634d6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/core/single_segment_index_writer.rs b/src/core/single_segment_index_writer.rs
@@ -21,6 +21,10 @@ impl SingleSegmentIndexWriter {
         })
     }
 
+    pub fn mem_usage(&self) -> usize {
+        self.segment_writer.mem_usage()
+    }
+
     pub fn add_document(&mut self, document: Document) -> crate::Result<()> {
         let opstamp = self.opstamp;
         self.opstamp += 1;