chore: support stream::Entry::value method

tonbo-io · Jul 18, 2024 · 13ad69b · 13ad69b
1 parent a672a9a
commit 13ad69b
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/stream/mod.rs b/src/stream/mod.rs
@@ -35,7 +35,7 @@ impl<R> Entry<'_, R>
 where
     R: Record,
 {
-    pub fn key(&self) -> Timestamped<<R::Key as Key>::Ref<'_>> {
+    pub(crate) fn key(&self) -> Timestamped<<R::Key as Key>::Ref<'_>> {
         match self {
             Entry::Mutable(entry) => entry
                 .key()
@@ -44,6 +44,14 @@ where
             Entry::Immutable(entry) => entry.internal_key(),
         }
     }
+
+    pub(crate) fn value(&self) -> R::Ref<'_> {
+        match self {
+            Entry::Mutable(entry) => entry.value().as_ref().map(R::as_record_ref).unwrap(),
+            Entry::SsTable(entry) => entry.get(),
+            Entry::Immutable(entry) => entry.get(),
+        }
+    }
 }
 
 impl<R> fmt::Debug for Entry<'_, R>