Include empty pages in fragmented statistics

cberner · Dec 12, 2024 · e510604 · e510604
1 parent 3862242
commit e510604
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/transactions.rs b/src/transactions.rs
@@ -1449,7 +1449,8 @@ impl WriteTransaction {
             + freed_tree_stats.stored_leaf_bytes;
         let total_fragmented = data_tree_stats.fragmented_bytes()
             + system_tree_stats.fragmented_bytes
-            + freed_tree_stats.fragmented_bytes;
+            + freed_tree_stats.fragmented_bytes
+            + self.mem.count_free_pages()? * (self.mem.get_page_size() as u64);
 
         Ok(DatabaseStats {
             tree_height: data_tree_stats.tree_height(),

diff --git a/src/tree_store/page_store/page_manager.rs b/src/tree_store/page_store/page_manager.rs
@@ -1132,6 +1132,16 @@ impl TransactionalMemory {
         Ok(count)
     }
 
+    pub(crate) fn count_free_pages(&self) -> Result<u64> {
+        let state = self.state.lock().unwrap();
+        let mut count = 0u64;
+        for i in 0..state.header.layout().num_regions() {
+            count += u64::from(state.get_region(i).count_free_pages());
+        }
+
+        Ok(count)
+    }
+
     pub(crate) fn get_page_size(&self) -> usize {
         self.page_size.try_into().unwrap()
     }