From cbdeff003dc39e1b5ba9c9671288ebd89f7338e6 Mon Sep 17 00:00:00 2001
From: Charles Hall <charles@computer.surgery>
Date: Thu, 24 Oct 2024 11:57:49 -0700
Subject: [PATCH] show column family sizes at trace level

---
 src/database/abstraction/rocksdb.rs | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/database/abstraction/rocksdb.rs b/src/database/abstraction/rocksdb.rs
index 90cdb06a..797a6fd5 100644
--- a/src/database/abstraction/rocksdb.rs
+++ b/src/database/abstraction/rocksdb.rs
@@ -1,5 +1,5 @@
 use std::{
-    collections::HashSet,
+    collections::{HashMap, HashSet},
     future::Future,
     pin::Pin,
     sync::{Arc, Mutex, RwLock},
@@ -104,6 +104,16 @@ impl KeyValueDatabaseEngine for Arc<Engine> {
             }),
         )?;
 
+        tracing::trace!(
+            cf_sizes = ?db.live_files().map(|x| {
+                x.into_iter().fold(HashMap::<_, usize>::new(), |mut acc, x| {
+                    *acc.entry(x.column_family_name).or_default() += x.size;
+                    acc
+                })
+            }),
+            "Size in bytes of all column families",
+        );
+
         Ok(Arc::new(Engine {
             rocks: db,
             max_open_files: config.database.rocksdb_max_open_files,