fix changes after rebase

Author: codetheweb

rust/worker/src/execution/operators/apply_log_to_segment_writer.rs

@@ -1,5 +1,4 @@
-use std::sync::Arc;
-
+use super::materialize_logs::MaterializeLogOutput;
 use crate::execution::operator::Operator;
 use crate::segment::metadata_segment::MetadataSegmentError;
 use crate::segment::record_segment::ApplyMaterializedLogError;
@@ -9,11 +8,10 @@ use crate::segment::SegmentWriter;
 use async_trait::async_trait;
 use chroma_error::ChromaError;
 use chroma_error::ErrorCodes;
+use std::sync::Arc;
 use thiserror::Error;
 use tracing::Instrument;
 
-use super::materialize_logs::MaterializeLogOutput;
-
 #[derive(Error, Debug)]
 pub enum ApplyLogToSegmentWriterOperatorError {
     #[error("Preparation for log materialization failed {0}")]
@@ -88,7 +86,8 @@ impl<Writer: SegmentWriter + Send + Sync + Clone>
             .segment_writer
             .apply_materialized_log_chunk(materialized_chunk.clone())
             .instrument(tracing::trace_span!(
-                "Apply materialized logs to record segment"
+                "Apply materialized logs",
+                segment = input.segment_writer.get_name()
             ))
             .await
         {

rust/worker/src/execution/orchestration/compact.rs

@@ -353,7 +353,10 @@ impl CompactOrchestrator {
         match self.dispatcher.send(task, Some(Span::current())).await {
             Ok(_) => (),
             Err(e) => {
-                tracing::error!("Error dispatching writers for compaction {:?}", e);
+                tracing::error!(
+                    "Error dispatching apply log to segment writer task: {:?}",
+                    e
+                );
                 panic!(
                     "Invariant violation. Somehow the dispatcher receiver is dropped. Error: {:?}",
                     e
@@ -671,7 +674,14 @@ impl Handler<TaskResult<MaterializeLogOutput, MaterializeLogOperatorError>>
         let materialized_result = Arc::new(materialized_result);
 
         let (record_segment_writer, hnsw_segment_writer, metadata_segment_writer) =
-            self.get_segment_writers().await.unwrap(); // todo
+            match self.get_segment_writers().await {
+                Ok(writers) => writers,
+                Err(e) => {
+                    tracing::error!("Error getting segment writers: {:?}", e);
+                    terminate_with_error(self.result_channel.take(), e, ctx);
+                    return;
+                }
+            };
 
         self.dispatch_apply_log_to_segment_writer_task(
             record_segment_writer,