1 // Copyright 2012 The Rust Project Developers. See the COPYRIGHT
2 // file at the top-level directory of this distribution and at
3 // http://rust-lang.org/COPYRIGHT.
5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8 // option. This file may not be copied, modified, or distributed
9 // except according to those terms.
11 // This test creates a bunch of tasks that simultaneously send to each
12 // other in a ring. The messages should all be basically
14 // This is like msgsend-ring-pipes but adapted to use Arcs.
16 // This also serves as a pipes test, because Arcs are implemented with pipes.
18 // no-pretty-expanded FIXME #15189
19 // ignore-lexer-test FIXME #15679
22 use std::str::from_str;
23 use std::sync::{Arc, Future, Mutex, Condvar};
24 use std::time::Duration;
28 type pipe = Arc<(Mutex<Vec<uint>>, Condvar)>;
30 fn send(p: &pipe, msg: uint) {
31 let &(ref lock, ref cond) = &**p;
32 let mut arr = lock.lock().unwrap();
36 fn recv(p: &pipe) -> uint {
37 let &(ref lock, ref cond) = &**p;
38 let mut arr = lock.lock().unwrap();
39 while arr.is_empty() {
40 arr = cond.wait(arr).unwrap();
45 fn init() -> (pipe,pipe) {
46 let m = Arc::new((Mutex::new(Vec::new()), Condvar::new()));
51 fn thread_ring(i: uint, count: uint, num_chan: pipe, num_port: pipe) {
52 let mut num_chan = Some(num_chan);
53 let mut num_port = Some(num_port);
54 // Send/Receive lots of messages.
55 for j in range(0u, count) {
56 //println!("task %?, iter %?", i, j);
57 let num_chan2 = num_chan.take().unwrap();
58 let num_port2 = num_port.take().unwrap();
59 send(&num_chan2, i * j);
60 num_chan = Some(num_chan2);
61 let _n = recv(&num_port2);
63 num_port = Some(num_port2);
68 let args = os::args();
69 let args = if os::getenv("RUST_BENCH").is_some() {
70 vec!("".to_string(), "100".to_string(), "10000".to_string())
71 } else if args.len() <= 1u {
72 vec!("".to_string(), "10".to_string(), "100".to_string())
74 args.clone().into_iter().collect()
77 let num_tasks = from_str::<uint>(args[1].as_slice()).unwrap();
78 let msg_per_task = from_str::<uint>(args[2].as_slice()).unwrap();
80 let (mut num_chan, num_port) = init();
82 let mut p = Some((num_chan, num_port));
83 let dur = Duration::span(|| {
84 let (mut num_chan, num_port) = p.take().unwrap();
87 let mut futures = Vec::new();
89 for i in range(1u, num_tasks) {
90 //println!("spawning %?", i);
91 let (new_chan, num_port) = init();
92 let num_chan_2 = num_chan.clone();
93 let new_future = Future::spawn(move|| {
94 thread_ring(i, msg_per_task, num_chan_2, num_port)
96 futures.push(new_future);
101 thread_ring(0, msg_per_task, num_chan, num_port);
104 for f in futures.iter_mut() {
109 // all done, report stats.
110 let num_msgs = num_tasks * msg_per_task;
111 let rate = (num_msgs as f64) / (dur.num_milliseconds() as f64);
113 println!("Sent {} messages in {} ms", num_msgs, dur.num_milliseconds());
114 println!(" {} messages / second", rate / 1000.0);
115 println!(" {} μs / message", 1000000. / rate / 1000.0);