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

// Copyright 2012 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.

// xfail-fast

extern mod extra;

use std::comm::Chan;
use std::comm;
use std::task;

pub fn main() { debug!("===== WITHOUT THREADS ====="); test00(); }

fn test00_start(ch: &Chan<int>, message: int, count: int) {
    debug!("Starting test00_start");
    let mut i: int = 0;
    while i < count {
        debug!("Sending Message");
        ch.send(message + 0);
        i = i + 1;
    }
    debug!("Ending test00_start");
}

fn test00() {
    let number_of_tasks: int = 16;
    let number_of_messages: int = 4;

    debug!("Creating tasks");

    let po = comm::PortSet::new();

    let mut i: int = 0;

    // Create and spawn tasks...
    let mut results = ~[];
    while i < number_of_tasks {
        let ch = po.chan();
        let mut builder = task::task();
        builder.future_result(|r| results.push(r));
        builder.spawn({
            let i = i;
            || test00_start(&ch, i, number_of_messages)
        });
        i = i + 1;
    }

    // Read from spawned tasks...
    let mut sum = 0;
    for results.each |r| {
        i = 0;
        while i < number_of_messages {
            let value = po.recv();
            sum += value;
            i = i + 1;
        }
    }

    // Join spawned tasks...
    for results.each |r| { r.recv(); }

    debug!("Completed: Final number is: ");
    error!(sum);
    // assert (sum == (((number_of_tasks * (number_of_tasks - 1)) / 2) *
    //       number_of_messages));
    assert_eq!(sum, 480);
}