solana/src/bin/client-demo.rs

extern crate serde_json;
extern crate silk;

use silk::accountant_stub::AccountantStub;
use silk::event::Event;
use silk::signature::{generate_keypair, get_pubkey};
use silk::transaction::{sign_transaction_data, Transaction};
use silk::genesis::Genesis;
use std::time::Instant;
use std::net::UdpSocket;
use std::io::stdin;

fn main() {
    let addr = "127.0.0.1:8000";
    let send_addr = "127.0.0.1:8001";

    let gen: Genesis = serde_json::from_reader(stdin()).unwrap();
    let alice_keypair = gen.get_keypair();
    let alice_pubkey = gen.get_pubkey();

    let socket = UdpSocket::bind(send_addr).unwrap();
    let mut acc = AccountantStub::new(addr, socket);
    let last_id = acc.get_last_id().unwrap();

    let txs = acc.get_balance(&alice_pubkey).unwrap().unwrap();
    println!("Alice's Initial Balance {}", txs);

    let one = 1;
    println!("Signing transactions...");
    let now = Instant::now();
    let sigs: Vec<(_, _)> = (0..txs)
        .map(|_| {
            let rando_keypair = generate_keypair();
            let rando_pubkey = get_pubkey(&rando_keypair);
            let sig = sign_transaction_data(&one, &alice_keypair, &rando_pubkey, &last_id);
            (rando_pubkey, sig)
        })
        .collect();
    let duration = now.elapsed();
    let ns = duration.as_secs() * 1_000_000_000 + duration.subsec_nanos() as u64;
    let bsps = txs as f64 / ns as f64;
    let nsps = ns as f64 / txs as f64;
    println!(
        "Done. {} thousand signatures per second, {}us per signature",
        bsps * 1_000_000_f64,
        nsps / 1_000_f64
    );

    println!("Verify signatures...");
    let now = Instant::now();
    for &(k, s) in &sigs {
        let e = Event::Transaction(Transaction {
            from: alice_pubkey,
            to: k,
            asset: one,
            last_id,
            sig: s,
        });
        assert!(e.verify());
    }
    let duration = now.elapsed();
    let ns = duration.as_secs() * 1_000_000_000 + duration.subsec_nanos() as u64;
    let bsvps = txs as f64 / ns as f64;
    let nspsv = ns as f64 / txs as f64;
    println!(
        "Done. {} thousand signature verifications per second, {}us per signature verification",
        bsvps * 1_000_000_f64,
        nspsv / 1_000_f64
    );

    println!("Transferring 1 unit {} times...", txs);
    let now = Instant::now();
    let mut sig = Default::default();
    for (k, s) in sigs {
        acc.transfer_signed(alice_pubkey, k, one, last_id, s)
            .unwrap();
        sig = s;
    }
    println!("Waiting for last transaction to be confirmed...",);
    acc.wait_on_signature(&sig).unwrap();

    let duration = now.elapsed();
    let ns = duration.as_secs() * 1_000_000_000 + duration.subsec_nanos() as u64;
    let tps = (txs * 1_000_000_000) as f64 / ns as f64;
    println!("Done. {} tps!", tps);
    let val = acc.get_balance(&alice_pubkey).unwrap().unwrap();
    println!("Alice's Final Balance {}", val);
    assert_eq!(val, 0);
}