/
task-perf-one-million.rs
78 lines (68 loc) · 1.83 KB
/
task-perf-one-million.rs
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
74
75
76
77
78
// Test for concurrent tasks
enum msg {
ready(comm::Chan<msg>),
start,
done(int),
}
fn calc(children: uint, parent_ch: comm::Chan<msg>) {
let port = comm::Port();
let chan = comm::Chan(&port);
let mut child_chs = ~[];
let mut sum = 0;
for iter::repeat (children) {
do task::spawn {
calc(0u, chan);
};
}
for iter::repeat (children) {
match comm::recv(port) {
ready(child_ch) => {
child_chs.push(child_ch);
}
_ => fail ~"task-perf-one-million failed (port not ready)"
}
}
comm::send(parent_ch, ready(chan));
match comm::recv(port) {
start => {
for vec::each(child_chs) |child_ch| {
comm::send(*child_ch, start);
}
}
_ => fail ~"task-perf-one-million failed (port not in start state)"
}
for iter::repeat (children) {
match comm::recv(port) {
done(child_sum) => { sum += child_sum; }
_ => fail ~"task-perf-one-million failed (port not done)"
}
}
comm::send(parent_ch, done(sum + 1));
}
fn main() {
let args = os::args();
let args = if os::getenv(~"RUST_BENCH").is_some() {
~[~"", ~"100000"]
} else if args.len() <= 1u {
~[~"", ~"100"]
} else {
args
};
let children = uint::from_str(args[1]).get();
let port = comm::Port();
let chan = comm::Chan(&port);
do task::spawn {
calc(children, chan);
};
match comm::recv(port) {
ready(chan) => {
comm::send(chan, start);
}
_ => fail ~"task-perf-one-million failed (port not ready)"
}
let sum = match comm::recv(port) {
done(sum) => { sum }
_ => fail ~"task-perf-one-million failed (port not done)"
};
error!("How many tasks? %d tasks.", sum);
}