fix(storage): fix panic

risingwavelabs · Feb 2, 2024 · 2dca292 · 2dca292
1 parent dde824e
commit 2dca292
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/storage/src/hummock/sstable/block.rs b/src/storage/src/hummock/sstable/block.rs
@@ -216,9 +216,12 @@ impl Block {
                     .map_err(HummockError::decode_error)?;
                 let mut decoded = BytesMut::with_capacity(uncompressed_capacity);
                 unsafe { decoded.set_len(uncompressed_capacity) }
-                decoder
+                let actual_size = decoder
                     .read(&mut decoded)
                     .map_err(HummockError::decode_error)?;
+                unsafe {
+                    decoded.set_len(actual_size);
+                }
                 debug_assert_eq!(decoded.capacity(), uncompressed_capacity);
                 decoded.freeze()
             }
@@ -227,9 +230,12 @@ impl Block {
                     .map_err(HummockError::decode_error)?;
                 let mut decoded = BytesMut::with_capacity(uncompressed_capacity);
                 unsafe { decoded.set_len(uncompressed_capacity) }
-                decoder
+                let actual_size = decoder
                     .read(&mut decoded)
                     .map_err(HummockError::decode_error)?;
+                unsafe {
+                    decoded.set_len(actual_size);
+                }
                 debug_assert_eq!(decoded.capacity(), uncompressed_capacity);
                 decoded.freeze()
             }