mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-05-07 22:28:02 +00:00
Ashley
5f751e4472
* :) * Slight tidy * Remove ServiceBuilderCommand * Remove whitespace * Keep task manager alive for check_block/import_blocks * Pass task_manager to run_until_exit * Make task_manager in run_until_exit and make subcommands async * Change the async_run fn to return a future and task manager * async_run should take a result fn * Apply suggestions from code review Co-authored-by: Bastian Köcher <bkchr@users.noreply.github.com> * Fix spaces in export_raw_state Co-authored-by: Bastian Köcher <bkchr@users.noreply.github.com>
105 lines
2.9 KiB
Rust
105 lines
2.9 KiB
Rust
// Copyright 2017-2020 Parity Technologies (UK) Ltd.
|
|
// This file is part of Substrate.
|
|
|
|
// Substrate is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
|
|
// Substrate is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with Substrate. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
use crate::error::Error;
|
|
use log::info;
|
|
use futures::{future, prelude::*};
|
|
use sp_runtime::traits::{
|
|
Block as BlockT, NumberFor, One, Zero, SaturatedConversion
|
|
};
|
|
use sp_runtime::generic::BlockId;
|
|
use codec::Encode;
|
|
|
|
use std::{io::Write, pin::Pin};
|
|
use sc_client_api::{BlockBackend, UsageProvider};
|
|
use std::sync::Arc;
|
|
use std::task::Poll;
|
|
|
|
/// Performs the blocks export.
|
|
pub fn export_blocks<B, C>(
|
|
client: Arc<C>,
|
|
mut output: impl Write + 'static,
|
|
from: NumberFor<B>,
|
|
to: Option<NumberFor<B>>,
|
|
binary: bool
|
|
) -> Pin<Box<dyn Future<Output = Result<(), Error>>>>
|
|
where
|
|
C: BlockBackend<B> + UsageProvider<B> + 'static,
|
|
B: BlockT,
|
|
{
|
|
let mut block = from;
|
|
|
|
let last = match to {
|
|
Some(v) if v.is_zero() => One::one(),
|
|
Some(v) => v,
|
|
None => client.usage_info().chain.best_number,
|
|
};
|
|
|
|
let mut wrote_header = false;
|
|
|
|
// Exporting blocks is implemented as a future, because we want the operation to be
|
|
// interruptible.
|
|
//
|
|
// Every time we write a block to the output, the `Future` re-schedules itself and returns
|
|
// `Poll::Pending`.
|
|
// This makes it possible either to interleave other operations in-between the block exports,
|
|
// or to stop the operation completely.
|
|
let export = future::poll_fn(move |cx| {
|
|
let client = &client;
|
|
|
|
if last < block {
|
|
return Poll::Ready(Err("Invalid block range specified".into()));
|
|
}
|
|
|
|
if !wrote_header {
|
|
info!("Exporting blocks from #{} to #{}", block, last);
|
|
if binary {
|
|
let last_: u64 = last.saturated_into::<u64>();
|
|
let block_: u64 = block.saturated_into::<u64>();
|
|
let len: u64 = last_ - block_ + 1;
|
|
output.write_all(&len.encode())?;
|
|
}
|
|
wrote_header = true;
|
|
}
|
|
|
|
match client.block(&BlockId::number(block))? {
|
|
Some(block) => {
|
|
if binary {
|
|
output.write_all(&block.encode())?;
|
|
} else {
|
|
serde_json::to_writer(&mut output, &block)
|
|
.map_err(|e| format!("Error writing JSON: {}", e))?;
|
|
}
|
|
},
|
|
// Reached end of the chain.
|
|
None => return Poll::Ready(Ok(())),
|
|
}
|
|
if (block % 10000.into()).is_zero() {
|
|
info!("#{}", block);
|
|
}
|
|
if block == last {
|
|
return Poll::Ready(Ok(()));
|
|
}
|
|
block += One::one();
|
|
|
|
// Re-schedule the task in order to continue the operation.
|
|
cx.waker().wake_by_ref();
|
|
Poll::Pending
|
|
});
|
|
|
|
Box::pin(export)
|
|
}
|