e4f7561bcd
The entire testsuite is converted to using info! rather than debug! because some depend on the code within the debug! being trans'd.
79 lines
1.9 KiB
Rust
79 lines
1.9 KiB
Rust
// Copyright 2012 The Rust Project Developers. See the COPYRIGHT
|
|
// file at the top-level directory of this distribution and at
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
// option. This file may not be copied, modified, or distributed
|
|
// except according to those terms.
|
|
|
|
// Based on threadring.erlang by Jira Isa
|
|
|
|
use std::os;
|
|
|
|
fn start(n_tasks: int, token: int) {
|
|
let (p, ch1) = stream();
|
|
let mut p = p;
|
|
let mut ch1 = ch1;
|
|
ch1.send(token);
|
|
// XXX could not get this to work with a range closure
|
|
let mut i = 2;
|
|
while i <= n_tasks {
|
|
let (next_p, ch) = stream();
|
|
let imm_i = i;
|
|
let imm_p = p;
|
|
do spawn {
|
|
roundtrip(imm_i, n_tasks, &imm_p, &ch);
|
|
};
|
|
p = next_p;
|
|
i += 1;
|
|
}
|
|
let imm_p = p;
|
|
let imm_ch = ch1;
|
|
do spawn {
|
|
roundtrip(1, n_tasks, &imm_p, &imm_ch);
|
|
}
|
|
}
|
|
|
|
fn roundtrip(id: int, n_tasks: int, p: &Port<int>, ch: &Chan<int>) {
|
|
while (true) {
|
|
match p.recv() {
|
|
1 => {
|
|
println(fmt!("%d\n", id));
|
|
return;
|
|
}
|
|
token => {
|
|
info!("thread: %d got token: %d", id, token);
|
|
ch.send(token - 1);
|
|
if token <= n_tasks {
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
let args = if os::getenv(~"RUST_BENCH").is_some() {
|
|
~[~"", ~"2000000", ~"503"]
|
|
}
|
|
else {
|
|
os::args()
|
|
};
|
|
let token = if args.len() > 1u {
|
|
FromStr::from_str(args[1]).get()
|
|
}
|
|
else {
|
|
1000
|
|
};
|
|
let n_tasks = if args.len() > 2u {
|
|
FromStr::from_str(args[2]).get()
|
|
}
|
|
else {
|
|
503
|
|
};
|
|
start(n_tasks, token);
|
|
|
|
}
|