2018-04-24 10:15:08 -07:00
|
|
|
extern crate futures;
|
2018-04-09 11:28:54 -07:00
|
|
|
extern crate getopts;
|
2018-04-21 06:12:57 -07:00
|
|
|
extern crate isatty;
|
2018-03-28 15:51:18 -07:00
|
|
|
extern crate rayon;
|
2018-03-05 14:34:15 -08:00
|
|
|
extern crate serde_json;
|
2018-03-27 15:24:05 -07:00
|
|
|
extern crate solana;
|
2018-04-16 13:51:06 -07:00
|
|
|
extern crate untrusted;
|
2018-02-28 09:07:54 -08:00
|
|
|
|
2018-04-24 10:15:08 -07:00
|
|
|
use futures::Future;
|
2018-04-11 19:24:14 -07:00
|
|
|
use getopts::Options;
|
2018-04-21 06:12:57 -07:00
|
|
|
use isatty::stdin_isatty;
|
2018-04-02 20:15:21 -07:00
|
|
|
use rayon::prelude::*;
|
2018-04-16 13:51:06 -07:00
|
|
|
use solana::mint::MintDemo;
|
2018-05-08 21:03:05 -07:00
|
|
|
use solana::signature::{GenKeys, KeyPair, KeyPairUtil};
|
2018-05-08 17:59:01 -07:00
|
|
|
use solana::thin_client::ThinClient;
|
2018-03-27 15:24:05 -07:00
|
|
|
use solana::transaction::Transaction;
|
2018-04-11 19:24:14 -07:00
|
|
|
use std::env;
|
2018-04-21 06:12:57 -07:00
|
|
|
use std::io::{stdin, Read};
|
2018-05-02 20:38:07 -07:00
|
|
|
use std::net::{SocketAddr, UdpSocket};
|
2018-04-19 07:06:19 -07:00
|
|
|
use std::process::exit;
|
2018-05-04 11:11:39 -07:00
|
|
|
use std::thread::sleep;
|
|
|
|
use std::time::Duration;
|
2018-04-26 12:17:36 -07:00
|
|
|
use std::time::Instant;
|
2018-04-16 13:51:06 -07:00
|
|
|
use untrusted::Input;
|
2018-03-04 00:21:40 -08:00
|
|
|
|
2018-04-19 10:16:20 -07:00
|
|
|
fn print_usage(program: &str, opts: Options) {
|
|
|
|
let mut brief = format!("Usage: cat <mint.json> | {} [options]\n\n", program);
|
|
|
|
brief += " Solana client demo creates a number of transactions and\n";
|
|
|
|
brief += " sends them to a target node.";
|
|
|
|
brief += " Takes json formatted mint file to stdin.";
|
|
|
|
|
|
|
|
print!("{}", opts.usage(&brief));
|
|
|
|
}
|
|
|
|
|
2018-02-28 09:07:54 -08:00
|
|
|
fn main() {
|
2018-04-09 11:28:54 -07:00
|
|
|
let mut threads = 4usize;
|
2018-05-22 09:46:52 -07:00
|
|
|
let mut server_addr: String = "127.0.0.1:8000".to_string();
|
2018-05-11 15:35:53 -07:00
|
|
|
let mut requests_addr: String = "127.0.0.1:8010".to_string();
|
2018-03-03 20:15:42 -08:00
|
|
|
|
2018-04-09 11:28:54 -07:00
|
|
|
let mut opts = Options::new();
|
|
|
|
opts.optopt("s", "", "server address", "host:port");
|
|
|
|
opts.optopt("c", "", "client address", "host:port");
|
2018-05-04 11:11:39 -07:00
|
|
|
opts.optopt("t", "", "number of threads", &format!("{}", threads));
|
2018-04-19 10:16:20 -07:00
|
|
|
opts.optflag("h", "help", "print help");
|
2018-04-09 11:28:54 -07:00
|
|
|
let args: Vec<String> = env::args().collect();
|
|
|
|
let matches = match opts.parse(&args[1..]) {
|
|
|
|
Ok(m) => m,
|
2018-04-19 07:06:19 -07:00
|
|
|
Err(e) => {
|
|
|
|
eprintln!("{}", e);
|
|
|
|
exit(1);
|
|
|
|
}
|
2018-04-09 11:28:54 -07:00
|
|
|
};
|
|
|
|
|
2018-04-19 10:16:20 -07:00
|
|
|
if matches.opt_present("h") {
|
|
|
|
let program = args[0].clone();
|
|
|
|
print_usage(&program, opts);
|
|
|
|
return;
|
|
|
|
}
|
2018-04-09 11:28:54 -07:00
|
|
|
if matches.opt_present("s") {
|
2018-05-22 09:46:52 -07:00
|
|
|
server_addr = matches.opt_str("s").unwrap();
|
2018-04-09 11:28:54 -07:00
|
|
|
}
|
|
|
|
if matches.opt_present("c") {
|
2018-05-11 15:35:53 -07:00
|
|
|
requests_addr = matches.opt_str("c").unwrap();
|
2018-04-09 11:28:54 -07:00
|
|
|
}
|
|
|
|
if matches.opt_present("t") {
|
|
|
|
threads = matches.opt_str("t").unwrap().parse().expect("integer");
|
|
|
|
}
|
2018-04-21 06:12:57 -07:00
|
|
|
|
2018-05-11 15:35:53 -07:00
|
|
|
let mut events_addr: SocketAddr = requests_addr.parse().unwrap();
|
|
|
|
let requests_port = events_addr.port();
|
|
|
|
events_addr.set_port(requests_port + 1);
|
|
|
|
|
2018-04-21 06:12:57 -07:00
|
|
|
if stdin_isatty() {
|
|
|
|
eprintln!("nothing found on stdin, expected a json file");
|
|
|
|
exit(1);
|
|
|
|
}
|
|
|
|
|
|
|
|
let mut buffer = String::new();
|
|
|
|
let num_bytes = stdin().read_to_string(&mut buffer).unwrap();
|
|
|
|
if num_bytes == 0 {
|
|
|
|
eprintln!("empty file on stdin, expected a json file");
|
|
|
|
exit(1);
|
|
|
|
}
|
|
|
|
|
2018-04-20 22:28:55 -07:00
|
|
|
println!("Parsing stdin...");
|
2018-04-26 08:35:10 -07:00
|
|
|
let demo: MintDemo = serde_json::from_str(&buffer).unwrap_or_else(|e| {
|
2018-04-19 07:55:47 -07:00
|
|
|
eprintln!("failed to parse json: {}", e);
|
|
|
|
exit(1);
|
|
|
|
});
|
2018-02-28 09:07:54 -08:00
|
|
|
|
2018-05-11 15:35:53 -07:00
|
|
|
println!("Binding to {}", requests_addr);
|
|
|
|
let requests_socket = UdpSocket::bind(&requests_addr).unwrap();
|
|
|
|
requests_socket
|
|
|
|
.set_read_timeout(Some(Duration::new(5, 0)))
|
|
|
|
.unwrap();
|
|
|
|
let events_socket = UdpSocket::bind(&events_addr).unwrap();
|
2018-05-22 09:46:52 -07:00
|
|
|
let requests_addr: SocketAddr = server_addr.parse().unwrap();
|
|
|
|
let requests_port = requests_addr.port();
|
|
|
|
let mut events_server_addr = requests_addr.clone();
|
|
|
|
events_server_addr.set_port(requests_port + 3);
|
|
|
|
let mut client = ThinClient::new(
|
|
|
|
requests_addr,
|
|
|
|
requests_socket,
|
|
|
|
events_server_addr,
|
|
|
|
events_socket,
|
|
|
|
);
|
2018-04-20 22:28:55 -07:00
|
|
|
|
|
|
|
println!("Get last ID...");
|
2018-05-11 15:15:14 -07:00
|
|
|
let last_id = client.get_last_id().wait().unwrap();
|
2018-05-02 20:38:07 -07:00
|
|
|
println!("Got last ID {:?}", last_id);
|
2018-03-05 14:34:15 -08:00
|
|
|
|
2018-05-08 21:03:05 -07:00
|
|
|
let rnd = GenKeys::new(demo.mint.keypair().public_key_bytes());
|
|
|
|
let tokens_per_user = 1_000;
|
|
|
|
|
2018-05-11 09:55:05 -07:00
|
|
|
let users = rnd.gen_n_keys(demo.num_accounts, tokens_per_user);
|
2018-05-08 21:03:05 -07:00
|
|
|
|
2018-04-20 22:28:55 -07:00
|
|
|
println!("Creating keypairs...");
|
2018-05-11 09:55:05 -07:00
|
|
|
let txs = demo.num_accounts / 2;
|
2018-05-08 21:03:05 -07:00
|
|
|
let keypairs: Vec<_> = users
|
2018-04-16 13:51:06 -07:00
|
|
|
.into_par_iter()
|
|
|
|
.map(|(pkcs8, _)| KeyPair::from_pkcs8(Input::from(&pkcs8)).unwrap())
|
|
|
|
.collect();
|
|
|
|
let keypair_pairs: Vec<_> = keypairs.chunks(2).collect();
|
2018-03-05 14:34:15 -08:00
|
|
|
|
2018-03-03 09:23:31 -08:00
|
|
|
println!("Signing transactions...");
|
|
|
|
let now = Instant::now();
|
2018-04-16 13:51:06 -07:00
|
|
|
let transactions: Vec<_> = keypair_pairs
|
2018-03-28 15:51:18 -07:00
|
|
|
.into_par_iter()
|
2018-04-16 13:51:06 -07:00
|
|
|
.map(|chunk| Transaction::new(&chunk[0], chunk[1].pubkey(), 1, last_id))
|
2018-03-03 10:45:21 -08:00
|
|
|
.collect();
|
2018-05-04 11:11:39 -07:00
|
|
|
let mut duration = now.elapsed();
|
2018-04-16 13:51:06 -07:00
|
|
|
let ns = duration.as_secs() * 1_000_000_000 + u64::from(duration.subsec_nanos());
|
|
|
|
let bsps = txs as f64 / ns as f64;
|
|
|
|
let nsps = ns as f64 / txs as f64;
|
2018-03-03 09:23:31 -08:00
|
|
|
println!(
|
2018-03-03 10:52:46 -08:00
|
|
|
"Done. {} thousand signatures per second, {}us per signature",
|
|
|
|
bsps * 1_000_000_f64,
|
|
|
|
nsps / 1_000_f64
|
2018-03-03 09:23:31 -08:00
|
|
|
);
|
|
|
|
|
2018-05-11 15:15:14 -07:00
|
|
|
let initial_tx_count = client.transaction_count();
|
2018-05-04 11:11:39 -07:00
|
|
|
println!("initial count {}", initial_tx_count);
|
2018-04-17 15:41:58 -07:00
|
|
|
|
2018-04-09 11:28:54 -07:00
|
|
|
println!("Transfering {} transactions in {} batches", txs, threads);
|
2018-03-01 11:23:27 -08:00
|
|
|
let now = Instant::now();
|
2018-04-09 11:28:54 -07:00
|
|
|
let sz = transactions.len() / threads;
|
|
|
|
let chunks: Vec<_> = transactions.chunks(sz).collect();
|
2018-04-16 13:51:06 -07:00
|
|
|
chunks.into_par_iter().for_each(|trs| {
|
2018-05-02 20:38:07 -07:00
|
|
|
println!("Transferring 1 unit {} times... to", trs.len());
|
2018-05-22 09:46:52 -07:00
|
|
|
let requests_addr: SocketAddr = server_addr.parse().unwrap();
|
|
|
|
let mut requests_cb_addr = requests_addr.clone();
|
|
|
|
requests_cb_addr.set_port(0);
|
|
|
|
let requests_socket = UdpSocket::bind(requests_cb_addr).unwrap();
|
2018-05-11 15:35:53 -07:00
|
|
|
requests_socket
|
|
|
|
.set_read_timeout(Some(Duration::new(5, 0)))
|
|
|
|
.unwrap();
|
|
|
|
let mut events_addr: SocketAddr = requests_addr.clone();
|
2018-05-22 09:46:52 -07:00
|
|
|
events_addr.set_port(0);
|
2018-05-11 15:35:53 -07:00
|
|
|
let events_socket = UdpSocket::bind(&events_addr).unwrap();
|
2018-05-22 09:46:52 -07:00
|
|
|
let client = ThinClient::new(
|
|
|
|
requests_addr,
|
|
|
|
requests_socket,
|
|
|
|
events_server_addr,
|
|
|
|
events_socket,
|
|
|
|
);
|
2018-04-16 13:51:06 -07:00
|
|
|
for tr in trs {
|
2018-05-11 15:15:14 -07:00
|
|
|
client.transfer_signed(tr.clone()).unwrap();
|
2018-04-16 13:51:06 -07:00
|
|
|
}
|
|
|
|
});
|
|
|
|
|
2018-05-04 11:11:39 -07:00
|
|
|
println!("Waiting for transactions to complete...",);
|
|
|
|
let mut tx_count;
|
2018-05-08 15:21:28 -07:00
|
|
|
for _ in 0..10 {
|
2018-05-11 15:15:14 -07:00
|
|
|
tx_count = client.transaction_count();
|
2018-05-04 11:11:39 -07:00
|
|
|
duration = now.elapsed();
|
|
|
|
let txs = tx_count - initial_tx_count;
|
|
|
|
println!("Transactions processed {}", txs);
|
|
|
|
let ns = duration.as_secs() * 1_000_000_000 + u64::from(duration.subsec_nanos());
|
|
|
|
let tps = (txs * 1_000_000_000) as f64 / ns as f64;
|
|
|
|
println!("{} tps", tps);
|
|
|
|
sleep(Duration::new(1, 0));
|
2018-04-17 15:41:58 -07:00
|
|
|
}
|
2018-02-28 09:07:54 -08:00
|
|
|
}
|