Trace one off collection ids

chroma-core · Jan 15, 2025 · 2e0c103 · 2e0c103
1 parent 7b6487b
commit 2e0c103
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/rust/worker/src/compactor/compaction_manager.rs b/rust/worker/src/compactor/compaction_manager.rs
@@ -334,9 +334,12 @@ impl Handler<OneOffCompactionMessage> for CompactionManager {
         message: OneOffCompactionMessage,
         _ctx: &ComponentContext<CompactionManager>,
     ) {
-        tracing::info!("CompactionManager: Performing one-off compaction");
         self.scheduler
             .add_oneoff_collections(message.collection_ids);
+        tracing::info!(
+            "One-off collections queued: {:?}",
+            self.scheduler.get_oneoff_collections()
+        );
     }
 }
 

diff --git a/rust/worker/src/compactor/scheduler.rs b/rust/worker/src/compactor/scheduler.rs
@@ -66,6 +66,10 @@ impl Scheduler {
         self.oneoff_collections.extend(ids);
     }
 
+    pub(crate) fn get_oneoff_collections(&self) -> Vec<CollectionUuid> {
+        self.oneoff_collections.iter().cloned().collect()
+    }
+
     async fn get_collections_with_new_data(&mut self) -> Vec<CollectionInfo> {
         let collections = self
             .log