From 5c0262832eb6b981ec970cb04ba4537c976ad26e Mon Sep 17 00:00:00 2001
From: marvin-j97 <marvin.janke.97@gmail.com>
Date: Sat, 7 Dec 2024 23:45:57 +0100
Subject: [PATCH] catch another ? in compaction worker, #87

---
 src/compaction/worker.rs | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/src/compaction/worker.rs b/src/compaction/worker.rs
index 5bec5d36..25011f6a 100644
--- a/src/compaction/worker.rs
+++ b/src/compaction/worker.rs
@@ -243,7 +243,7 @@ fn merge_segments(
 
     let start = Instant::now();
 
-    let mut segment_writer = MultiWriter::new(
+    let Ok(segment_writer) = MultiWriter::new(
         opts.segment_id_generator.clone(),
         payload.target_size,
         crate::segment::writer::Options {
@@ -252,8 +252,19 @@ fn merge_segments(
             data_block_size: opts.config.data_block_size,
             index_block_size: opts.config.index_block_size,
         },
-    )?
-    .use_compression(opts.config.compression);
+    ) else {
+        log::error!("Compaction failed");
+
+        // IMPORTANT: Show the segments again, because compaction failed
+        opts.levels
+            .write()
+            .expect("lock is poisoned")
+            .show_segments(payload.segment_ids.iter().copied());
+
+        return Ok(());
+    };
+
+    let mut segment_writer = segment_writer.use_compression(opts.config.compression);
 
     #[cfg(feature = "bloom")]
     {