From 3de979aa7c598b7287bc9e73df053822c9e25a86 Mon Sep 17 00:00:00 2001 From: Tyera Eulberg Date: Wed, 11 Jul 2018 10:35:06 -0600 Subject: [PATCH] Check client balance and only airdrop if less than TPS quota --- src/bin/client-demo.rs | 24 ++++++++++++++---------- 1 file changed, 14 insertions(+), 10 deletions(-) diff --git a/src/bin/client-demo.rs b/src/bin/client-demo.rs index c7d3f47774..9f510fcf67 100644 --- a/src/bin/client-demo.rs +++ b/src/bin/client-demo.rs @@ -224,18 +224,22 @@ fn main() { let mut client = mk_client(&leader); let starting_balance = client.poll_get_balance(&id.pubkey()).unwrap(); - let txs: i64 = 500_000; - println!("Airdropping {:?} tokens", txs); - let _airdrop = request_airdrop(&drone_addr, &id, txs as u64).unwrap(); - // TODO: return airdrop Result from Drone - sleep(Duration::from_millis(100)); - let balance = client.poll_get_balance(&id.pubkey()).unwrap(); - println!("Your balance is: {:?}", balance); - if balance < txs || (starting_balance == balance) { - println!("TPS airdrop limit reached; wait 60sec to retry"); - exit(1); + if starting_balance < txs { + let airdrop_amount = txs - starting_balance; + println!("Airdropping {:?} tokens", airdrop_amount); + let _airdrop = request_airdrop(&drone_addr, &id, airdrop_amount as u64).unwrap(); + // TODO: return airdrop Result from Drone + sleep(Duration::from_millis(100)); + + let balance = client.poll_get_balance(&id.pubkey()).unwrap(); + println!("Your balance is: {:?}", balance); + + if balance < txs || (starting_balance == balance) { + println!("TPS airdrop limit reached; wait 60sec to retry"); + exit(1); + } } println!("Get last ID...");