improved tracing and cleanup

tensorlakeai · Dec 2, 2024 · 3cf4f75 · 3cf4f75
1 parent fba76e0
commit 3cf4f75
Show file tree

Hide file tree

Showing 2 changed files with 157 additions and 169 deletions.
diff --git a/server/src/scheduler.rs b/server/src/scheduler.rs
@@ -594,7 +594,7 @@ mod tests {
         let state = IndexifyState::new(temp_dir.path().join("state")).await?;
         let scheduler = Scheduler::new(
             state.clone(),
-            Arc::new(scheduler_stats::Metrics::new(state.clone())),
+            Arc::new(scheduler_stats::Metrics::new(state.metrics.clone())),
         );
 
         let graph = {
@@ -867,7 +867,7 @@ mod tests {
         let state = IndexifyState::new(temp_dir.path().join("state")).await?;
         let scheduler = Scheduler::new(
             state.clone(),
-            Arc::new(scheduler_stats::Metrics::new(state.clone())),
+            Arc::new(scheduler_stats::Metrics::new(state.metrics.clone())),
         );
 
         let graph = {
@@ -1191,7 +1191,7 @@ mod tests {
         let state = IndexifyState::new(temp_dir.path().join("state")).await?;
         let scheduler = Scheduler::new(
             state.clone(),
-            Arc::new(scheduler_stats::Metrics::new(state.clone())),
+            Arc::new(scheduler_stats::Metrics::new(state.metrics.clone())),
         );
 
         let graph = {