84 lines
2.2 KiB
Rust
84 lines
2.2 KiB
Rust
// Copyright 2014 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.
|
|
|
|
// 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.
|
|
|
|
use std::sync::mpsc::channel;
|
|
use std::os;
|
|
use std::thread::Thread;
|
|
|
|
// This is a simple bench that creates M pairs of tasks. These
|
|
// tasks ping-pong back and forth over a pair of streams. This is a
|
|
// canonical message-passing benchmark as it heavily strains message
|
|
// passing and almost nothing else.
|
|
|
|
fn ping_pong_bench(n: uint, m: uint) {
|
|
|
|
// Create pairs of tasks that pingpong back and forth.
|
|
fn run_pair(n: uint) {
|
|
// Create a channel: A->B
|
|
let (atx, arx) = channel();
|
|
// Create a channel: B->A
|
|
let (btx, brx) = channel();
|
|
|
|
let guard_a = Thread::scoped(move|| {
|
|
let (tx, rx) = (atx, brx);
|
|
for _ in 0..n {
|
|
tx.send(()).unwrap();
|
|
rx.recv().unwrap();
|
|
}
|
|
});
|
|
|
|
let guard_b = Thread::scoped(move|| {
|
|
let (tx, rx) = (btx, arx);
|
|
for _ in 0..n {
|
|
rx.recv().unwrap();
|
|
tx.send(()).unwrap();
|
|
}
|
|
});
|
|
|
|
guard_a.join().ok();
|
|
guard_b.join().ok();
|
|
}
|
|
|
|
for _ in 0..m {
|
|
run_pair(n)
|
|
}
|
|
}
|
|
|
|
|
|
|
|
fn main() {
|
|
|
|
let args = os::args();
|
|
let args = args.as_slice();
|
|
let n = if args.len() == 3 {
|
|
args[1].parse::<uint>().unwrap()
|
|
} else {
|
|
10000
|
|
};
|
|
|
|
let m = if args.len() == 3 {
|
|
args[2].parse::<uint>().unwrap()
|
|
} else {
|
|
4
|
|
};
|
|
|
|
ping_pong_bench(n, m);
|
|
|
|
}
|