feat: return full key in state store get_keyed_row interface (#19345)

risingwavelabs · Nov 15, 2024 · f29c971 · f29c971
1 parent 4085f56
commit f29c971
Show file tree

Hide file tree

Showing 16 changed files with 473 additions and 105 deletions.
diff --git a/src/storage/hummock_test/src/snapshot_tests.rs b/src/storage/hummock_test/src/snapshot_tests.rs
@@ -32,7 +32,7 @@ use risingwave_storage::store::{
 use risingwave_storage::StateStore;
 
 use crate::local_state_store_test_utils::LocalStateStoreTestExt;
-use crate::test_utils::{gen_key_from_bytes, with_hummock_storage_v2, TestIngestBatch};
+use crate::test_utils::{gen_key_from_bytes, with_hummock_storage, TestIngestBatch};
 
 macro_rules! assert_count_range_scan {
     (
@@ -402,36 +402,36 @@ async fn test_snapshot_range_scan_inner(
 
 #[tokio::test]
 async fn test_snapshot_v2() {
-    let (storage, meta_client) = with_hummock_storage_v2(Default::default()).await;
+    let (storage, meta_client) = with_hummock_storage(Default::default()).await;
     test_snapshot_inner(storage, meta_client, false, false).await;
 }
 
 #[tokio::test]
 async fn test_snapshot_with_sync_v2() {
-    let (storage, meta_client) = with_hummock_storage_v2(Default::default()).await;
+    let (storage, meta_client) = with_hummock_storage(Default::default()).await;
     test_snapshot_inner(storage, meta_client, true, false).await;
 }
 
 #[tokio::test]
 async fn test_snapshot_with_commit_v2() {
-    let (storage, meta_client) = with_hummock_storage_v2(Default::default()).await;
+    let (storage, meta_client) = with_hummock_storage(Default::default()).await;
     test_snapshot_inner(storage, meta_client, true, true).await;
 }
 
 #[tokio::test]
 async fn test_snapshot_range_scan_v2() {
-    let (storage, meta_client) = with_hummock_storage_v2(Default::default()).await;
+    let (storage, meta_client) = with_hummock_storage(Default::default()).await;
     test_snapshot_range_scan_inner(storage, meta_client, false, false).await;
 }
 
 #[tokio::test]
 async fn test_snapshot_range_scan_with_sync_v2() {
-    let (storage, meta_client) = with_hummock_storage_v2(Default::default()).await;
+    let (storage, meta_client) = with_hummock_storage(Default::default()).await;
     test_snapshot_range_scan_inner(storage, meta_client, true, false).await;
 }
 
 #[tokio::test]
 async fn test_snapshot_range_scan_with_commit_v2() {
-    let (storage, meta_client) = with_hummock_storage_v2(Default::default()).await;
+    let (storage, meta_client) = with_hummock_storage(Default::default()).await;
     test_snapshot_range_scan_inner(storage, meta_client, true, true).await;
 }