Use config commitment for balance checks (solana-labs#842)

vkomenda · Nov 14, 2020 · 6ee1f09 · 6ee1f09
1 parent c4ec3b3
commit 6ee1f09
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/token/cli/src/main.rs b/token/cli/src/main.rs
@@ -163,7 +163,10 @@ fn new_throwaway_signer() -> (Option<Box<dyn Signer>>, Option<Pubkey>) {
 }
 
 fn check_fee_payer_balance(config: &Config, required_balance: u64) -> Result<(), Error> {
-    let balance = config.rpc_client.get_balance(&config.fee_payer)?;
+    let balance = config
+        .rpc_client
+        .get_balance_with_commitment(&config.fee_payer, config.commitment_config)?
+        .value;
     if balance < required_balance {
         Err(format!(
             "Fee payer, {}, has insufficient balance: {} required, {} available",
@@ -178,7 +181,10 @@ fn check_fee_payer_balance(config: &Config, required_balance: u64) -> Result<(),
 }
 
 fn check_owner_balance(config: &Config, required_balance: u64) -> Result<(), Error> {
-    let balance = config.rpc_client.get_balance(&config.owner)?;
+    let balance = config
+        .rpc_client
+        .get_balance_with_commitment(&config.owner, config.commitment_config)?
+        .value;
     if balance < required_balance {
         Err(format!(
             "Owner, {}, has insufficient balance: {} required, {} available",