rust/src/test/bench/msgsend-pipes.rs

97 lines
2.6 KiB
Rust
Raw Normal View History

// A port of the simplistic benchmark from
//
// http://github.com/PaulKeeble/ScalaVErlangAgents
//
// I *think* it's the same, more or less.
2012-07-12 22:09:19 -05:00
// xfail-pretty
use std;
2012-09-05 14:32:05 -05:00
use io::Writer;
use io::WriterUtil;
2012-09-05 14:32:05 -05:00
use pipes::{Port, PortSet, Chan};
macro_rules! move_out (
{ $x:expr } => { unsafe { let y <- *ptr::addr_of($x); y } }
)
enum request {
get_count,
bytes(uint),
stop
}
2012-08-28 13:11:15 -05:00
fn server(requests: PortSet<request>, responses: pipes::Chan<uint>) {
let mut count = 0u;
let mut done = false;
while !done {
2012-08-06 14:34:08 -05:00
match requests.try_recv() {
2012-08-20 14:23:37 -05:00
Some(get_count) => { responses.send(copy count); }
Some(bytes(b)) => {
2012-08-22 19:24:52 -05:00
//error!("server: received %? bytes", b);
count += b;
}
2012-08-20 14:23:37 -05:00
None => { done = true; }
2012-08-03 21:59:04 -05:00
_ => { }
}
}
responses.send(count);
2012-08-22 19:24:52 -05:00
//error!("server exiting");
}
fn run(args: &[~str]) {
let (to_parent, from_child) = pipes::stream();
let (to_child, from_parent_) = pipes::stream();
let from_parent = PortSet();
from_parent.add(from_parent_);
let size = option::get(uint::from_str(args[1]));
let workers = option::get(uint::from_str(args[2]));
let num_bytes = 100;
let start = std::time::precise_time_s();
let mut worker_results = ~[];
for uint::range(0u, workers) |i| {
let (to_child, from_parent_) = pipes::stream();
from_parent.add(from_parent_);
do task::task().future_result(|+r| {
vec::push(worker_results, r);
}).spawn {
for uint::range(0u, size / workers) |_i| {
2012-08-22 19:24:52 -05:00
//error!("worker %?: sending %? bytes", i, num_bytes);
to_child.send(bytes(num_bytes));
}
2012-08-22 19:24:52 -05:00
//error!("worker %? exiting", i);
};
}
do task::spawn {
server(from_parent, to_parent);
}
vec::iter(worker_results, |r| { future::get(&r); } );
2012-08-22 19:24:52 -05:00
//error!("sending stop message");
to_child.send(stop);
2012-08-22 19:24:52 -05:00
move_out!(to_child);
let result = from_child.recv();
let end = std::time::precise_time_s();
let elapsed = end - start;
2012-08-22 19:24:52 -05:00
io::stdout().write_str(fmt!("Count is %?\n", result));
io::stdout().write_str(fmt!("Test took %? seconds\n", elapsed));
let thruput = ((size / workers * workers) as float) / (elapsed as float);
2012-08-22 19:24:52 -05:00
io::stdout().write_str(fmt!("Throughput=%f per sec\n", thruput));
assert result == num_bytes * size;
}
fn main(args: ~[~str]) {
let args = if os::getenv(~"RUST_BENCH").is_some() {
~[~"", ~"1000000", ~"8"]
} else if args.len() <= 1u {
~[~"", ~"10000", ~"4"]
} else {
copy args
};
2012-08-22 19:24:52 -05:00
debug!("%?", args);
run(args);
}