Cleanup client traits and create super trait (#3728)

This commit is contained in:
Jack May
2019-04-11 00:25:14 -07:00
committed by GitHub
parent 4b6c0198ad
commit d0f46d6a8a
22 changed files with 153 additions and 115 deletions

View File

@ -1,11 +1,11 @@
use crate::bank::Bank;
use solana_sdk::async_client::AsyncClient;
use solana_sdk::client::{AsyncClient, SyncClient};
use solana_sdk::hash::Hash;
use solana_sdk::instruction::Instruction;
use solana_sdk::message::Message;
use solana_sdk::pubkey::Pubkey;
use solana_sdk::signature::Signature;
use solana_sdk::signature::{Keypair, KeypairUtil};
use solana_sdk::sync_client::SyncClient;
use solana_sdk::system_instruction;
use solana_sdk::transaction::{self, Transaction};
use solana_sdk::transport::Result;
@ -23,9 +23,13 @@ impl<'a> AsyncClient for BankClient<'a> {
Ok(transaction.signatures.get(0).cloned().unwrap_or_default())
}
fn async_send_message(&self, keypairs: &[&Keypair], message: Message) -> io::Result<Signature> {
let blockhash = self.bank.last_blockhash();
let transaction = Transaction::new(&keypairs, message, blockhash);
fn async_send_message(
&self,
keypairs: &[&Keypair],
message: Message,
recent_blockhash: Hash,
) -> io::Result<Signature> {
let transaction = Transaction::new(&keypairs, message, recent_blockhash);
self.async_send_transaction(transaction)
}
@ -33,9 +37,10 @@ impl<'a> AsyncClient for BankClient<'a> {
&self,
keypair: &Keypair,
instruction: Instruction,
recent_blockhash: Hash,
) -> io::Result<Signature> {
let message = Message::new(vec![instruction]);
self.async_send_message(&[keypair], message)
self.async_send_message(&[keypair], message, recent_blockhash)
}
/// Transfer `lamports` from `keypair` to `pubkey`
@ -44,10 +49,11 @@ impl<'a> AsyncClient for BankClient<'a> {
lamports: u64,
keypair: &Keypair,
pubkey: &Pubkey,
recent_blockhash: Hash,
) -> io::Result<Signature> {
let transfer_instruction =
system_instruction::transfer(&keypair.pubkey(), pubkey, lamports);
self.async_send_instruction(keypair, transfer_instruction)
self.async_send_instruction(keypair, transfer_instruction, recent_blockhash)
}
}
@ -86,6 +92,15 @@ impl<'a> SyncClient for BankClient<'a> {
) -> Result<Option<transaction::Result<()>>> {
Ok(self.bank.get_signature_status(signature))
}
fn get_recent_blockhash(&self) -> Result<Hash> {
let last_blockhash = self.bank.last_blockhash();
Ok(last_blockhash)
}
fn get_transaction_count(&self) -> Result<u64> {
Ok(self.bank.transaction_count())
}
}
impl<'a> BankClient<'a> {

View File

@ -1,10 +1,11 @@
use crate::bank_client::BankClient;
use serde::Serialize;
use solana_sdk::client::SyncClient;
use solana_sdk::instruction::{AccountMeta, Instruction};
use solana_sdk::loader_instruction;
use solana_sdk::pubkey::Pubkey;
use solana_sdk::signature::{Keypair, KeypairUtil};
use solana_sdk::sync_client::SyncClient;
use solana_sdk::system_instruction;
pub fn load_program(

View File

@ -107,10 +107,10 @@ mod tests {
use crate::bank::Bank;
use crate::bank_client::BankClient;
use solana_sdk::account::Account;
use solana_sdk::client::SyncClient;
use solana_sdk::genesis_block::GenesisBlock;
use solana_sdk::instruction::{AccountMeta, Instruction, InstructionError};
use solana_sdk::signature::{Keypair, KeypairUtil};
use solana_sdk::sync_client::SyncClient;
use solana_sdk::system_program;
use solana_sdk::transaction::TransactionError;