Send all transactions as quickly as possible, then wait (solana-labs#26)

* Send all transactions as quickly as possible, then wait

* Exit when finalized or blockhashes have expired

* Don't need blockhash in the CSV output

* Better types

CSV library was choking on Pubkey as a type. PickleDb doesn't have that problem.

* Resend if blockhash has not expired

* Attempt to fix CI

.travis.yml

@@ -1,5 +1,4 @@
 language: rust
-cache: cargo
 addons:
   apt:
     packages:

Cargo.toml

@@ -13,6 +13,7 @@ csv = "1.1.3"
 clap = "2.33.0"
 console = "0.10.3"
 indexmap = "1.3.2"
+indicatif = "0.14.0"
 itertools = "0.9.0"
 pickledb = "0.4.1"
 serde = { version = "1.0", features = ["derive"] }

src/thin_client.rs

@@ -1,6 +1,7 @@
 use solana_client::rpc_client::RpcClient;
 use solana_runtime::bank_client::BankClient;
 use solana_sdk::{
+    account::Account,
     client::{AsyncClient, SyncClient},
     fee_calculator::FeeCalculator,
     hash::Hash,
@@ -9,6 +10,10 @@ use solana_sdk::{
     signature::{Signature, Signer},
     signers::Signers,
     system_instruction,
+    sysvar::{
+        recent_blockhashes::{self, RecentBlockhashes},
+        Sysvar,
+    },
     transaction::Transaction,
     transport::{Result, TransportError},
 };
@@ -26,6 +31,7 @@ pub trait Client {
     ) -> Result<Vec<Option<TransactionStatus>>>;
     fn get_balance1(&self, pubkey: &Pubkey) -> Result<u64>;
     fn get_recent_blockhash1(&self) -> Result<(Hash, FeeCalculator)>;
+    fn get_account1(&self, pubkey: &Pubkey) -> Result<Option<Account>>;
 }
 
 impl Client for RpcClient {
@@ -58,6 +64,12 @@ impl Client for RpcClient {
         self.get_recent_blockhash()
             .map_err(|e| TransportError::Custom(e.to_string()))
     }
+
+    fn get_account1(&self, pubkey: &Pubkey) -> Result<Option<Account>> {
+        self.get_account(pubkey)
+            .map(Some)
+            .map_err(|e| TransportError::Custom(e.to_string()))
+    }
 }
 
 impl Client for BankClient {
@@ -97,6 +109,10 @@ impl Client for BankClient {
     fn get_recent_blockhash1(&self) -> Result<(Hash, FeeCalculator)> {
         self.get_recent_blockhash()
     }
+
+    fn get_account1(&self, pubkey: &Pubkey) -> Result<Option<Account>> {
+        self.get_account(pubkey)
+    }
 }
 
 pub struct ThinClient<C: Client>(pub C);
@@ -144,4 +160,16 @@ impl<C: Client> ThinClient<C> {
     pub fn get_balance(&self, pubkey: &Pubkey) -> Result<u64> {
         self.0.get_balance1(pubkey)
     }
+
+    pub fn get_account(&self, pubkey: &Pubkey) -> Result<Option<Account>> {
+        self.0.get_account1(pubkey)
+    }
+
+    pub fn get_recent_blockhashes(&self) -> Result<Vec<Hash>> {
+        let opt_blockhashes_account = self.get_account(&recent_blockhashes::id())?;
+        let blockhashes_account = opt_blockhashes_account.unwrap();
+        let recent_blockhashes = RecentBlockhashes::from_account(&blockhashes_account).unwrap();
+        let hashes = recent_blockhashes.iter().map(|x| x.blockhash).collect();
+        Ok(hashes)
+    }
 }