feat(sst-dump): support more features (risingwavelabs#8580)

Author: hzxa21

Cargo.lock

@@ -26,6 +26,7 @@ use size::Size;
 use tokio::task::JoinHandle;
 
 use super::table::{get_table_catalog, make_state_table};
+use crate::common::HummockServiceOpts;
 use crate::CtlContext;
 
 #[derive(Subcommand)]
@@ -37,6 +38,7 @@ pub enum BenchCommands {
         /// number of futures doing scan
         #[clap(long, default_value_t = 1)]
         threads: usize,
+        data_dir: Option<String>,
     },
 }
 
@@ -74,11 +76,17 @@ impl InterestedMetrics {
 
 pub async fn do_bench(context: &CtlContext, cmd: BenchCommands) -> Result<()> {
     let meta = context.meta_client().await?;
-    let (hummock, metrics) = context.hummock_store_with_metrics().await?;
     let next_cnt = Arc::new(AtomicU64::new(0));
     let iter_cnt = Arc::new(AtomicU64::new(0));
     match cmd {
-        BenchCommands::Scan { mv_name, threads } => {
+        BenchCommands::Scan {
+            mv_name,
+            threads,
+            data_dir,
+        } => {
+            let (hummock, metrics) = context
+                .hummock_store_with_metrics(HummockServiceOpts::from_env(data_dir)?)
+                .await?;
             let table = get_table_catalog(meta.clone(), mv_name).await?;
             let mut handlers = vec![];
             for i in 0..threads {

src/ctl/src/cmd_impl/bench.rs

@@ -17,10 +17,18 @@ use core::ops::Bound::Unbounded;
 use risingwave_common::catalog::TableId;
 use risingwave_storage::store::{PrefetchOptions, ReadOptions, StateStoreReadExt};
 
+use crate::common::HummockServiceOpts;
 use crate::CtlContext;
 
-pub async fn list_kv(context: &CtlContext, epoch: u64, table_id: u32) -> anyhow::Result<()> {
-    let hummock = context.hummock_store().await?;
+pub async fn list_kv(
+    context: &CtlContext,
+    epoch: u64,
+    table_id: u32,
+    data_dir: Option<String>,
+) -> anyhow::Result<()> {
+    let hummock = context
+        .hummock_store(HummockServiceOpts::from_env(data_dir)?)
+        .await?;
     if epoch == u64::MAX {
         tracing::info!("using u64::MAX as epoch");
     }