track memory arena memory usage

quickwit-oss · Aug 16, 2023 · 09b4159 · 09b4159
1 parent 62ece86
commit 09b4159
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/aggregation/bucket/term_agg.rs b/src/aggregation/bucket/term_agg.rs
@@ -1269,6 +1269,7 @@ mod tests {
         ];
 
         let index = get_test_index_from_terms(false, &terms_per_segment)?;
+        assert_eq!(index.searchable_segments().unwrap().len(), 2);
 
         let agg_req: Aggregations = serde_json::from_value(json!({
             "my_texts": {

diff --git a/stacker/src/arena_hashmap.rs b/stacker/src/arena_hashmap.rs
@@ -164,7 +164,7 @@ impl ArenaHashMap {
 
     #[inline]
     pub fn mem_usage(&self) -> usize {
-        self.table.len() * mem::size_of::<KeyValue>()
+        self.table.len() * mem::size_of::<KeyValue>() + self.memory_arena.mem_usage()
     }
 
     #[inline]