diff --git a/genesis/src/genesis_accounts.rs b/genesis/src/genesis_accounts.rs index 3d206511ab..af710c6753 100644 --- a/genesis/src/genesis_accounts.rs +++ b/genesis/src/genesis_accounts.rs @@ -178,13 +178,10 @@ fn add_stakes( genesis_config: &mut GenesisConfig, staker_infos: &[StakerInfo], unlock_info: &UnlockInfo, - granularity: u64, ) -> u64 { staker_infos .iter() - .map(|staker_info| { - create_and_add_stakes(genesis_config, staker_info, unlock_info, granularity) - }) + .map(|staker_info| create_and_add_stakes(genesis_config, staker_info, unlock_info, None)) .sum::() } @@ -196,23 +193,16 @@ pub fn add_genesis_accounts(genesis_config: &mut GenesisConfig, mut issued_lampo genesis_config, &BATCH_FOUR_STAKER_INFOS, &UNLOCKS_ALL_AT_9_MONTHS, - 1_000_000 * LAMPORTS_PER_SOL, ) + add_stakes( genesis_config, &FOUNDATION_STAKER_INFOS, &UNLOCKS_ALL_DAY_ZERO, - 1_000_000 * LAMPORTS_PER_SOL, - ) + add_stakes( - genesis_config, - &GRANTS_STAKER_INFOS, - &UNLOCKS_ALL_DAY_ZERO, - 1_000_000 * LAMPORTS_PER_SOL, - ) + add_stakes( - genesis_config, - &COMMUNITY_STAKER_INFOS, - &UNLOCKS_ALL_DAY_ZERO, - 1_000_000 * LAMPORTS_PER_SOL, - ); + ) + add_stakes(genesis_config, &GRANTS_STAKER_INFOS, &UNLOCKS_ALL_DAY_ZERO) + + add_stakes( + genesis_config, + &COMMUNITY_STAKER_INFOS, + &UNLOCKS_ALL_DAY_ZERO, + ); // "one thanks" (community pool) gets 500_000_000SOL (total) - above distributions create_and_add_stakes( @@ -223,7 +213,7 @@ pub fn add_genesis_accounts(genesis_config: &mut GenesisConfig, mut issued_lampo lamports: 500_000_000 * LAMPORTS_PER_SOL - issued_lamports, }, &UNLOCKS_ALL_DAY_ZERO, - 1_000_000 * LAMPORTS_PER_SOL, + None, ); } diff --git a/genesis/src/stakes.rs b/genesis/src/stakes.rs index b5093a24b7..0428697956 100644 --- a/genesis/src/stakes.rs +++ b/genesis/src/stakes.rs @@ -41,8 +41,9 @@ pub fn create_and_add_stakes( // description of how the stakes' lockups will expire unlock_info: &UnlockInfo, // the largest each stake account should be, in lamports - granularity: u64, + granularity: Option, ) -> u64 { + let granularity = granularity.unwrap_or(std::u64::MAX); let authorized = Authorized::auto( &staker_info .staker @@ -163,7 +164,7 @@ mod tests { ) { assert_eq!( total_lamports, - create_and_add_stakes(genesis_config, staker_info, unlock_info, granularity) + create_and_add_stakes(genesis_config, staker_info, unlock_info, Some(granularity)) ); assert_eq!(genesis_config.accounts.len(), len); assert_eq!(