blockworks-foundation
/
solana
mirror of https://github.com/blockworks-foundation/solana.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add --remove-account option to create-snapshot command

This commit is contained in:
Michael Vines 2020-12-04 22:46:32 -08:00 committed by mergify[bot]
parent 882d033233
commit c59cb28bbf
1 changed files with 23 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
ledger-tool/src/main.rs
View File

@ -9,7 +9,9 @@ use serde::Serialize;
use serde_json::json;
use solana_clap_utils::{
input_parsers::{cluster_type_of, pubkey_of, pubkeys_of},
input_validators::{is_parsable, is_pubkey_or_keypair, is_slot, is_valid_percentage},
input_validators::{
is_parsable, is_pubkey, is_pubkey_or_keypair, is_slot, is_valid_percentage,
},
};
use solana_ledger::entry::Entry;
use solana_ledger::{
@ -1139,6 +1141,16 @@ fn main() {
.takes_value(false)
.help("Re-calculate the bank hash and overwrite the original bank hash."),
)
.arg(
Arg::with_name("accounts_to_remove")
.required(false)
.long("remove-account")
.takes_value(true)
.value_name("PUBKEY")
.validator(is_pubkey)
.multiple(true)
.help("List if accounts to remove while creating the snapshot"),
)
.arg(
Arg::with_name("remove_stake_accounts")
.required(false)
@ -1678,6 +1690,8 @@ fn main() {
exit(1);
}
let bootstrap_validator_pubkeys = pubkeys_of(&arg_matches, "bootstrap_validator");
let accounts_to_remove =
pubkeys_of(&arg_matches, "accounts_to_remove").unwrap_or_default();
let snapshot_version =
arg_matches
@ -1749,6 +1763,14 @@ fn main() {
}
}
for address in accounts_to_remove {
if let Some(mut account) = bank.get_account(&address) {
rehash = true;
account.lamports = 0;
bank.store_account(&address, &account);
}
}
if let Some(bootstrap_validator_pubkeys) = bootstrap_validator_pubkeys {
rehash = true;