src/test/bench/rt-parfib.rs

   1 // Copyright 2013 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.
   4 //
   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.
  10
  11 use std::sync::mpsc::channel;
  12 use std::env;
  13 use std::thread;
  14
  15 // A simple implementation of parfib. One subtree is found in a new
  16 // thread and communicated over a oneshot pipe, the other is found
  17 // locally. There is no sequential-mode threshold.
  18
  19 fn parfib(n: u64) -> u64 {
  20     if n == 0 || n == 1 {
  21         return 1;
  22     }
  23
  24     let (tx, rx) = channel();
  25     thread::spawn(move|| {
  26         tx.send(parfib(n-1)).unwrap();
  27     });
  28     let m2 = parfib(n-2);
  29     return rx.recv().unwrap() + m2;
  30 }
  31
  32 fn main() {
  33     let mut args = env::args();
  34     let n = if args.len() == 2 {
  35         args.nth(1).unwrap().parse::<u64>().unwrap()
  36     } else {
  37         10
  38     };
  39
  40     parfib(n);
  41
  42 }