Break down file storage usage gauge by component (#30077)

GitOrigin-RevId: 077fab0b09ee47d5681ff043846a8a7a2931f78f
get-convex · Sep 24, 2024 · 7985643 · 7985643
1 parent fc68aa9
commit 7985643
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/crates/events/src/testing.rs b/crates/events/src/testing.rs
@@ -176,7 +176,10 @@ impl UsageCounterState {
             },
             UsageEvent::CurrentVectorStorage { tables: _ } => todo!(),
             UsageEvent::CurrentDatabaseStorage { tables: _ } => todo!(),
-            UsageEvent::CurrentFileStorage { total_size: _ } => todo!(),
+            UsageEvent::CurrentFileStorage {
+                total_size: _,
+                by_component: _,
+            } => todo!(),
             UsageEvent::CurrentDocumentCounts { tables: _ } => todo!(),
         }
     }

diff --git a/crates/events/src/usage.rs b/crates/events/src/usage.rs
@@ -1,3 +1,5 @@
+use std::collections::BTreeMap;
+
 use async_trait::async_trait;
 use serde::{
     Deserialize,
@@ -108,7 +110,10 @@ pub enum UsageEvent {
         tables: Vec<TableDatabaseStorage>,
     },
     CurrentFileStorage {
+        // TODO(ENG-7423): Clean up this old field once no databricks tables and queries rely on
+        // it. They should use the by_component field instead.
         total_size: u64,
+        by_component: BTreeMap<Option<String>, u64>,
     },
     CurrentDocumentCounts {
         tables: Vec<TableDocumentCount>,