move get to test_utils

src/storage/src/hummock/test_utils.rs

@@ -584,3 +584,24 @@ impl<S: StateStore> StateStoreReadTestExt for S {
         Ok(ret)
     }
 }
+
+pub trait StateStoreGetTestExt: StateStoreGet {
+    fn get(
+        &self,
+        key: TableKey<Bytes>,
+        read_options: ReadOptions,
+    ) -> impl StorageFuture<'_, Option<Bytes>>;
+}
+
+impl<S: StateStoreGet> StateStoreGetTestExt for S {
+    async fn get(
+        &self,
+        key: TableKey<Bytes>,
+        read_options: ReadOptions,
+    ) -> StorageResult<Option<Bytes>> {
+        self.on_key_value(key, read_options, |_, value| {
+            Ok(Bytes::copy_from_slice(value))
+        })
+        .await
+    }
+}

src/storage/src/store.rs

@@ -264,22 +264,6 @@ pub trait StateStoreGet: StaticSendSync {
         read_options: ReadOptions,
         on_key_value_fn: impl KeyValueFn<O>,
     ) -> impl StorageFuture<'_, Option<O>>;
-
-    /// Point gets a value from the state store.
-    /// The result is based on a snapshot corresponding to the given `epoch`.
-    /// Only the value is returned.
-    fn get(
-        &self,
-        key: TableKey<Bytes>,
-        read_options: ReadOptions,
-    ) -> impl StorageFuture<'_, Option<Bytes>> {
-        async move {
-            self.on_key_value(key, read_options, |_, value| {
-                Ok(Bytes::copy_from_slice(value))
-            })
-            .await
-        }
-    }
 }
 
 pub trait StateStoreRead: StateStoreGet + StaticSendSync {