Add db params for export-state CLI command (#10830)

paritytech · Feb 9, 2022 · 3aecef7 · 3aecef7
1 parent e9d79da
commit 3aecef7
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/client/cli/src/commands/export_blocks_cmd.rs b/client/cli/src/commands/export_blocks_cmd.rs
@@ -76,7 +76,7 @@ impl ExportBlocksCmd {
 		C: BlockBackend<B> + UsageProvider<B> + 'static,
 		<<B::Header as HeaderT>::Number as FromStr>::Err: Debug,
 	{
-		if let DatabaseSource::RocksDb { ref path, .. } = database_config {
+		if let Some(path) = database_config.path() {
 			info!("DB path: {}", path.display());
 		}
 

diff --git a/client/cli/src/commands/export_state_cmd.rs b/client/cli/src/commands/export_state_cmd.rs
@@ -18,7 +18,7 @@
 
 use crate::{
 	error,
-	params::{BlockNumberOrHash, PruningParams, SharedParams},
+	params::{BlockNumberOrHash, DatabaseParams, PruningParams, SharedParams},
 	CliConfiguration,
 };
 use clap::Parser;
@@ -42,6 +42,10 @@ pub struct ExportStateCmd {
 	#[allow(missing_docs)]
 	#[clap(flatten)]
 	pub pruning_params: PruningParams,
+
+	#[allow(missing_docs)]
+	#[clap(flatten)]
+	pub database_params: DatabaseParams,
 }
 
 impl ExportStateCmd {
@@ -81,4 +85,8 @@ impl CliConfiguration for ExportStateCmd {
 	fn pruning_params(&self) -> Option<&PruningParams> {
 		Some(&self.pruning_params)
 	}
+
+	fn database_params(&self) -> Option<&DatabaseParams> {
+		Some(&self.database_params)
+	}
 }