1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
// A port of the simplistic benchmark from
//
// http://github.com/PaulKeeble/ScalaVErlangAgents
//
// I *think* it's the same, more or less.
use std;
import io::writer;
import io::writer_util;
enum request {
get_count,
bytes(uint),
stop
}
fn server(requests: comm::port<request>, responses: comm::chan<uint>) {
let mut count = 0u;
let mut done = false;
while !done {
alt comm::recv(requests) {
get_count { comm::send(responses, copy count); }
bytes(b) { count += b; }
stop { done = true; }
}
}
comm::send(responses, count);
}
fn run(args: ~[str]) {
let from_child = comm::port();
let to_parent = comm::chan(from_child);
let to_child = do task::spawn_listener |po| {
server(po, to_parent);
};
let size = option::get(uint::from_str(args[1]));
let workers = option::get(uint::from_str(args[2]));
let start = std::time::precise_time_s();
let to_child = to_child;
let mut worker_results = ~[];
for uint::range(0u, workers) |_i| {
let builder = task::builder();
vec::push(worker_results, task::future_result(builder));
do task::run(builder) {
for uint::range(0u, size / workers) |_i| {
comm::send(to_child, bytes(100u));
}
};
}
vec::iter(worker_results, |r| { future::get(r); } );
comm::send(to_child, stop);
let result = comm::recv(from_child);
let end = std::time::precise_time_s();
let elapsed = end - start;
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);
io::stdout().write_str(#fmt("Throughput=%f per sec\n", thruput));
}
fn main(args: ~[str]) {
let args = if os::getenv("RUST_BENCH").is_some() {
~["", "1000000", "10000"]
} else if args.len() <= 1u {
~["", "10000", "4"]
} else {
args
};
#debug("%?", args);
run(args);
}
|
