src/test/run-pass/task-comm-3.rs

   1 // Copyright 2012-2014 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 #![feature(std_misc)]
  12
  13 // no-pretty-expanded FIXME #15189
  14
  15 use std::thread;
  16 use std::sync::mpsc::{channel, Sender};
  17
  18 pub fn main() { println!("===== WITHOUT THREADS ====="); test00(); }
  19
  20 fn test00_start(ch: &Sender<isize>, message: isize, count: isize) {
  21     println!("Starting test00_start");
  22     let mut i: isize = 0;
  23     while i < count {
  24         println!("Sending Message");
  25         ch.send(message + 0).unwrap();
  26         i = i + 1;
  27     }
  28     println!("Ending test00_start");
  29 }
  30
  31 fn test00() {
  32     let number_of_tasks: isize = 16;
  33     let number_of_messages: isize = 4;
  34
  35     println!("Creating tasks");
  36
  37     let (tx, rx) = channel();
  38
  39     let mut i: isize = 0;
  40
  41     // Create and spawn tasks...
  42     let mut results = Vec::new();
  43     while i < number_of_tasks {
  44         let tx = tx.clone();
  45         results.push(thread::spawn({
  46             let i = i;
  47             move|| {
  48                 test00_start(&tx, i, number_of_messages)
  49             }
  50         }));
  51         i = i + 1;
  52     }
  53
  54     // Read from spawned tasks...
  55     let mut sum = 0;
  56     for _r in &results {
  57         i = 0;
  58         while i < number_of_messages {
  59             let value = rx.recv().unwrap();
  60             sum += value;
  61             i = i + 1;
  62         }
  63     }
  64
  65     // Join spawned tasks...
  66     for r in results { r.join(); }
  67
  68     println!("Completed: Final number is: ");
  69     println!("{}", sum);
  70     // assert (sum == (((number_of_tasks * (number_of_tasks - 1)) / 2) *
  71     //       number_of_messages));
  72     assert_eq!(sum, 480);
  73 }