Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: add script to snapshot all staking accounts #542

Merged
merged 8 commits into from
Nov 12, 2024
Merged

feat: add script to snapshot all staking accounts #542

Changes from 1 commit
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
bcb68b6
checkpoint
guibescos Nov 11, 2024
f4293c0
checkpoint
guibescos Nov 11, 2024
123f65d
checkpoint
guibescos Nov 11, 2024
2ab9776
go
guibescos Nov 11, 2024
42a66df
imports
guibescos Nov 11, 2024
b27489c
clippy
guibescos Nov 11, 2024
733dd5a
some clean up
guibescos Nov 11, 2024
30b946e
add custody authority
guibescos Nov 12, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
checkpoint
  • Loading branch information
@guibescos
guibescos committed Nov 11, 2024
commit 123f65d5e6f0f9f176c689132e0176de3fd8ed4e
1 change: 1 addition & 0 deletions staking/Cargo.lock
View file

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

1 change: 1 addition & 0 deletions staking/cli/Cargo.toml
View file
Original file line number Diff line number Diff line change
@@ -26,3 +26,4 @@ serde_json = "1.0.128"
uriparse = "0.6.4"
solana-remote-wallet = "1.18.16"
solana-account-decoder = "1.18.16"
chrono = "0.4.38"
2 changes: 1 addition & 1 deletion staking/cli/src/cli.rs
View file
Original file line number Diff line number Diff line change
@@ -109,7 +109,7 @@ pub enum Action {
#[clap(long, help = "Publisher caps")]
publisher_caps: Pubkey,
},
GetAllAccounts {},
SaveStakeAccountsSnapshot {},
}

pub enum SignerSource {
140 changes: 5 additions & 135 deletions staking/cli/src/instructions.rs
View file
Original file line number Diff line number Diff line change
@@ -107,9 +107,11 @@ use {
HashSet,
},
convert::TryInto,
fmt::Debug,
mem::size_of,
str::FromStr,
thread::current,
time::SystemTime,
},
wormhole_core_bridge_solana::sdk::{
WriteEncodedVaaArgs,
@@ -871,7 +873,7 @@ pub fn update_y(rpc_client: &RpcClient, signer: &dyn Signer, y: u64) {
process_transaction(rpc_client, &[instruction], &[signer]).unwrap();
}

pub fn get_all_accounts(rpc_client: &RpcClient) {
pub fn save_stake_accounts_snapshot(rpc_client: &RpcClient) {
let data: Vec<(Pubkey, DynamicPositionArrayAccount, Pubkey, Pubkey)> = rpc_client
.get_program_accounts_with_config(
&staking::ID,
@@ -1061,7 +1063,8 @@ pub fn get_all_accounts(rpc_client: &RpcClient) {
},
};

let file = File::create("output.csv").unwrap();
let timestamp = chrono::Utc::now().format("%Y-%m-%d_%H:%M:%S").to_string();
let file = File::create(format!("snapshots/snapshot-{}.csv", timestamp)).unwrap();
let mut writer = BufWriter::new(file);

// Write the header
@@ -1090,137 +1093,4 @@ pub fn get_all_accounts(rpc_client: &RpcClient) {
)
.unwrap();
}

// // let current_epoch = get_current_epoch(rpc_client);

// // let governance_stake_amounts = position_accounts.iter().enumerate().map(|(index, (pubkey,
// // account))| { let mut account = DynamicPositionArrayAccount{
// // key: *pubkey,
// // lamports: account.lamports,
// // data: account.data.clone(),
// // };
// // let dynamic_position_array = account.to_dynamic_position_array();

// // println!("Account {}", index);

// // compute_voter_weight(
// // &dynamic_position_array,
// // current_epoch,
// // MAX_VOTER_WEIGHT,
// // MAX_VOTER_WEIGHT
// // ).unwrap()
// // }).collect::<Vec<_>>();

// // let ois_stake_amounts = position_accounts.iter().enumerate().map(|(index, (pubkey,
// account))| // { let mut account = DynamicPositionArrayAccount{
// // key: *pubkey,
// // lamports: account.lamports,
// // data: account.data.clone(),
// // };
// // let dynamic_position_array = account.to_dynamic_position_array();

// // let mut raw_voter_weight = 0u64;
// // for i in 0..dynamic_position_array.get_position_capacity() {
// // if let Some(position) = dynamic_position_array.read_position(i).unwrap(){
// // match position.get_current_position(current_epoch).unwrap() {
// // PositionState::LOCKED | PositionState::PREUNLOCKING => {
// // if !position.is_voting() {
// // // position.amount is trusted, so I don't think this can overflow,
// // // but still probably better to use checked math
// // raw_voter_weight = raw_voter_weight + position.amount;
// // }
// // }
// // _ => {}
// // }
// // }
// // }
// // raw_voter_weight
// // }).collect::<Vec<_>>();

// // let total_capacity: u64 = governance_stake_amounts.iter().sum();
// // println!("Total capacity: {}", total_capacity);
// // println!("Total OIS capacity: {}", ois_stake_amounts.iter().sum::<u64>());

// // let mut frequencies = std::collections::HashMap::new();
// // for capacity in &position_account_capacities {
// // *frequencies.entry(*capacity).or_insert(0) += 1;
// // }

// // let mut sorted_frequencies: Vec<_> = frequencies.iter().collect();
// // sorted_frequencies.sort_by_key(|&(capacity, _)| capacity);
// // for (capacity, count) in sorted_frequencies {
// // println!("{}: {}", capacity, count);
// // }

// let stake_account_metadata_accounts = rpc_client
// .get_program_accounts_with_config(
// &staking::ID,
// RpcProgramAccountsConfig {
// filters: Some(vec![RpcFilterType::Memcmp(Memcmp::new(
// 0,
// MemcmpEncodedBytes::Bytes(StakeAccountMetadataV2::discriminator().to_vec()),
// ))]),
// account_config: RpcAccountInfoConfig {
// encoding: Some(UiAccountEncoding::Base64Zstd),
// data_slice: None,
// commitment: None,
// min_context_slot: None,
// },
// with_context: None,
// },
// )
// .unwrap()
// .into_iter()
// .map(|(pubkey, account)| {
// (
// pubkey,
// StakeAccountMetadataV2::try_deserialize(&mut account.data.as_slice()).unwrap(),
// )
// })
// .collect::<Vec<_>>();

// let stake_account_metadata_accounts = stake_account_metadata_accounts
// .iter()
// .filter(|(_, metadata)| !matches!(metadata.lock, VestingSchedule::FullyVested))
// .collect::<Vec<_>>();
// println!("{}", stake_account_metadata_accounts.len());

// let mut custody_accounts: Vec<Pubkey> = stake_account_metadata_accounts
// .iter()
// .map(|(pubkey, _)| get_stake_account_custody_address(*pubkey))
// .collect::<Vec<_>>();

// let current_time = get_current_time(rpc_client);
// let stake_account_locked_amount = stake_account_metadata_accounts
// .iter()
// .map(|(pubkey, metadata)| {
// metadata
// .lock
// .get_unvested_balance(current_time, config.pyth_token_list_time)
// .unwrap()
// })
// .collect::<Vec<_>>();

// println!("{:?}", stake_account_locked_amount.iter().sum::<u64>());

// let mut token_accounts: Vec<TokenAccount> = Vec::new();
// for chunk in custody_accounts.chunks(100) {
// let batch = rpc_client
// .get_multiple_accounts(&chunk)
// .unwrap()
// .into_iter()
// .map(|account| {
// TokenAccount::try_deserialize(&mut account.unwrap().data.as_slice()).unwrap()
// })
// .collect::<Vec<_>>();
// token_accounts.extend_from_slice(batch.as_slice());
// }

// let custody_accounts = rpc_client.get_token_largest_accounts(
// &Pubkey::from_str("HZ1JovNiVvGrGNiiYvEozEVgZ58xaU3RKwX8eACQBCt3").unwrap(),
// );

// // println!("{:?}", positio);
// // println!("{:?}", stake_account_metadata_accounts.unwrap().len());
// println!("{:?}", custody_accounts.unwrap().len());
}
6 changes: 3 additions & 3 deletions staking/cli/src/main.rs
View file
Original file line number Diff line number Diff line change
@@ -11,9 +11,9 @@ use {
close_publisher_caps,
create_slash_event,
fetch_publisher_caps_and_advance,
get_all_accounts,
initialize_pool,
initialize_reward_custody,
save_stake_accounts_snapshot,
set_publisher_stake_account,
slash,
update_delegation_fee,
@@ -94,8 +94,8 @@ fn main() {
Action::ClosePublisherCaps { publisher_caps } => {
close_publisher_caps(&rpc_client, keypair.as_ref(), publisher_caps)
}
Action::GetAllAccounts {} => {
get_all_accounts(&rpc_client);
Action::SaveStakeAccountsSnapshot {} => {
save_stake_accounts_snapshot(&rpc_client);
}
}
}
Loading