diff --git a/src/test/bench/shootout-threadring.rs b/src/test/bench/shootout-threadring.rs index 54881a63694ee..a845481f0e028 100644 --- a/src/test/bench/shootout-threadring.rs +++ b/src/test/bench/shootout-threadring.rs @@ -1,4 +1,4 @@ -// Copyright 2012 The Rust Project Developers. See the COPYRIGHT +// Copyright 2012-2014 The Rust Project Developers. See the COPYRIGHT // file at the top-level directory of this distribution and at // http://rust-lang.org/COPYRIGHT. // @@ -8,69 +8,39 @@ // 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; +#![feature(phase)] +#[phase(syntax)] extern crate green; +green_start!(main) fn start(n_tasks: int, token: int) { let (tx, mut rx) = channel(); tx.send(token); - // FIXME could not get this to work with a range closure - let mut i = 2; - while i <= n_tasks { + for i in range(2, n_tasks + 1) { let (tx, next_rx) = channel(); - let imm_i = i; - let imm_rx = rx; - spawn(proc() { - roundtrip(imm_i, n_tasks, &imm_rx, &tx); - }); + spawn(proc() roundtrip(i, tx, rx)); rx = next_rx; - i += 1; } - let imm_rx = rx; - spawn(proc() { - roundtrip(1, n_tasks, &imm_rx, &tx); - }); + spawn(proc() roundtrip(1, tx, rx)); } -fn roundtrip(id: int, n_tasks: int, p: &Receiver, ch: &Sender) { - loop { - match p.recv() { - 1 => { - println!("{}\n", id); - return; - } - token => { - println!("thread: {} got token: {}", id, token); - ch.send(token - 1); - if token <= n_tasks { - return; - } - } +fn roundtrip(id: int, tx: Sender, rx: Receiver) { + for token in rx.iter() { + if token == 1 { + println!("{}", id); + break; } + tx.send(token - 1); } } fn main() { - use std::from_str::FromStr; - - let args = if os::getenv("RUST_BENCH").is_some() { - vec!("".to_owned(), "2000000".to_owned(), "503".to_owned()) + let args = std::os::args(); + let token = if std::os::getenv("RUST_BENCH").is_some() { + 2000000 } else { - os::args().move_iter().collect() - }; - let token = if args.len() > 1u { - FromStr::from_str(*args.get(1)).unwrap() - } - else { - 1000 + args.get(1).and_then(|arg| from_str(*arg)).unwrap_or(1000) }; - let n_tasks = if args.len() > 2u { - FromStr::from_str(*args.get(2)).unwrap() - } - else { - 503 - }; - start(n_tasks, token); + let n_tasks = args.get(2).and_then(|arg| from_str(*arg)).unwrap_or(503); + start(n_tasks, token); }