sc_cli/commands/
export_blocks_cmd.rs1use crate::{
20 error,
21 params::{DatabaseParams, GenericNumber, PruningParams, SharedParams},
22 CliConfiguration,
23};
24use clap::Parser;
25use log::info;
26use sc_client_api::{BlockBackend, HeaderBackend, UsageProvider};
27use sc_service::{chain_ops::export_blocks, config::DatabaseSource};
28use sp_runtime::traits::{Block as BlockT, Header as HeaderT};
29use std::{fmt::Debug, fs, io, path::PathBuf, str::FromStr, sync::Arc};
30
31#[derive(Debug, Clone, Parser)]
33pub struct ExportBlocksCmd {
34 #[arg()]
36 pub output: Option<PathBuf>,
37
38 #[arg(long, value_name = "BLOCK")]
41 pub from: Option<GenericNumber>,
42
43 #[arg(long, value_name = "BLOCK")]
46 pub to: Option<GenericNumber>,
47
48 #[arg(long)]
50 pub binary: bool,
51
52 #[allow(missing_docs)]
53 #[clap(flatten)]
54 pub shared_params: SharedParams,
55
56 #[allow(missing_docs)]
57 #[clap(flatten)]
58 pub pruning_params: PruningParams,
59
60 #[allow(missing_docs)]
61 #[clap(flatten)]
62 pub database_params: DatabaseParams,
63}
64
65impl ExportBlocksCmd {
66 pub async fn run<B, C>(
68 &self,
69 client: Arc<C>,
70 database_config: DatabaseSource,
71 ) -> error::Result<()>
72 where
73 B: BlockT,
74 C: HeaderBackend<B> + BlockBackend<B> + UsageProvider<B> + 'static,
75 <<B::Header as HeaderT>::Number as FromStr>::Err: Debug,
76 {
77 if let Some(path) = database_config.path() {
78 info!("DB path: {}", path.display());
79 }
80
81 let from = self.from.as_ref().and_then(|f| f.parse().ok()).unwrap_or(1u32);
82 let to = self.to.as_ref().and_then(|t| t.parse().ok());
83
84 let binary = self.binary;
85
86 let file: Box<dyn io::Write> = match &self.output {
87 Some(filename) => Box::new(fs::File::create(filename)?),
88 None => Box::new(io::stdout()),
89 };
90
91 export_blocks(client, file, from.into(), to, binary).await.map_err(Into::into)
92 }
93}
94
95impl CliConfiguration for ExportBlocksCmd {
96 fn shared_params(&self) -> &SharedParams {
97 &self.shared_params
98 }
99
100 fn pruning_params(&self) -> Option<&PruningParams> {
101 Some(&self.pruning_params)
102 }
103
104 fn database_params(&self) -> Option<&DatabaseParams> {
105 Some(&self.database_params)
106 }
107}