bench: Add a test for task concurrency
This commit is contained in:
parent
1248c250d8
commit
0438e6e924
1 changed files with 68 additions and 0 deletions
68
src/test/bench/task-perf-one-million.rs
Normal file
68
src/test/bench/task-perf-one-million.rs
Normal file
|
@ -0,0 +1,68 @@
|
|||
// 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 child_chs = [];
|
||||
let sum = 0;
|
||||
|
||||
iter::repeat (children) {||
|
||||
task::spawn {||
|
||||
calc(0u, chan);
|
||||
};
|
||||
}
|
||||
|
||||
iter::repeat (children) {||
|
||||
alt check comm::recv(port) {
|
||||
ready(child_ch) {
|
||||
child_chs += [child_ch];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
comm::send(parent_ch, ready(chan));
|
||||
|
||||
alt check comm::recv(port) {
|
||||
start {
|
||||
vec::iter (child_chs) { |child_ch|
|
||||
comm::send(child_ch, start);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
iter::repeat (children) {||
|
||||
alt check comm::recv(port) {
|
||||
done(child_sum) { sum += child_sum; }
|
||||
}
|
||||
}
|
||||
|
||||
comm::send(parent_ch, done(sum + 1));
|
||||
}
|
||||
|
||||
fn main(args: [str]) {
|
||||
let children = if vec::len(args) == 2u {
|
||||
uint::from_str(args[1])
|
||||
} else {
|
||||
100u
|
||||
};
|
||||
let port = comm::port();
|
||||
let chan = comm::chan(port);
|
||||
task::spawn {||
|
||||
calc(children, chan);
|
||||
};
|
||||
alt check comm::recv(port) {
|
||||
ready(chan) {
|
||||
comm::send(chan, start);
|
||||
}
|
||||
}
|
||||
let sum = alt check comm::recv(port) {
|
||||
done(sum) { sum }
|
||||
};
|
||||
#error("How many tasks? %d tasks.", sum);
|
||||
}
|
Loading…
Reference in a new issue