Fixed logical dump in new ZffWriter

zff-team · Jan 27, 2025 · 3e082bf · 3e082bf
1 parent 262e0d4
commit 3e082bf
Showing 1 changed file with 14 additions and 9 deletions.
diff --git a/src/lib/io/zffwriter/mod.rs b/src/lib/io/zffwriter/mod.rs
@@ -931,17 +931,22 @@ impl<R: Read> Read for ZffWriter<R> {
                                         }
                                 };
 
-                                if !self.in_progress_data.chunkmaps.offset_map.add_chunk_entry(
-                                    current_chunk_number, self.in_progress_data.bytes_read.current_segment) {
-                                    self.flush_chunkmap(ChunkMapType::OffsetMap)?;
-                                    self.read_state = ReadState::ChunkOffsetMap;
-                                    self.in_progress_data.chunkmaps.offset_map.add_chunk_entry(
-                                        current_chunk_number, self.in_progress_data.bytes_read.current_segment);
+                                match data {
+                                    PreparedData::PreparedChunk(_) => {
+                                        if !self.in_progress_data.chunkmaps.offset_map.add_chunk_entry(
+                                            current_chunk_number, self.in_progress_data.bytes_read.current_segment) {
+                                            self.flush_chunkmap(ChunkMapType::OffsetMap)?;
+                                            self.read_state = ReadState::ChunkOffsetMap;
+                                            self.in_progress_data.chunkmaps.offset_map.add_chunk_entry(
+                                                current_chunk_number, self.in_progress_data.bytes_read.current_segment);
+                                        }
+                                        self.in_progress_data.current_prepared_data_queue_state = PreparedDataQueueState::Flags;
+                                    },
+                                    _ => {
+                                        self.in_progress_data.current_prepared_data_queue_state = PreparedDataQueueState::Data;
+                                    },
                                 }
-
                                 self.in_progress_data.current_prepared_data_queue = Some(data);
-                                self.in_progress_data.current_prepared_data_queue_state = PreparedDataQueueState::Flags;
-                                self.in_progress_data.current_encoded_chunked_data_read_bytes = ReadBytes::NotRead;
                             },
                             PreparedDataQueueState::Flags => {
                                 let flags = match &self.in_progress_data.current_prepared_data_queue {