reuse in replay

risingwavelabs · Jan 15, 2025 · 9ad0c6b · 9ad0c6b
1 parent 17e16c0
commit 9ad0c6b
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/src/storage/hummock_test/src/bin/replay/replay_impl.rs b/src/storage/hummock_test/src/bin/replay/replay_impl.rs
@@ -14,7 +14,6 @@
 
 use std::ops::Bound;
 
-use bytes::Bytes;
 use futures::stream::BoxStream;
 use futures::{Stream, StreamExt, TryStreamExt};
 use futures_async_stream::try_stream;
@@ -251,11 +250,10 @@ impl LocalReplayRead for LocalReplayImpl {
     ) -> Result<Option<TracedBytes>> {
         Ok(self
             .0
-            .on_key_value(TableKey(key.into()), read_options.into(), |_, value| {
-                Ok(TracedBytes::from(Bytes::copy_from_slice(value)))
-            })
+            .get(TableKey(key.into()), read_options.into())
             .await
-            .unwrap())
+            .unwrap()
+            .map(TracedBytes::from))
     }
 }