aboutsummaryrefslogtreecommitdiffstats
path: root/components/util/task.rs
blob: 6047fdb3daefe49c24f5233559f1f49cb37895d7 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

use std::str::IntoMaybeOwned;
use std::task;
use std::comm::Sender;
use std::task::TaskBuilder;
use native::task::NativeTaskBuilder;
use rtinstrument;
use task_state;

pub fn spawn_named<S: IntoMaybeOwned<'static>>(name: S, f: proc():Send) {
    let builder = task::TaskBuilder::new().named(name);
    builder.spawn(proc() {
        rtinstrument::instrument(f);
    });
}

pub fn spawn_named_native<S: IntoMaybeOwned<'static>>(name: S, f: proc():Send) {
    let builder = task::TaskBuilder::new().named(name).native();
    builder.spawn(proc() {
        rtinstrument::instrument(f);
    });
}

/// Arrange to send a particular message to a channel if the task fails.
pub fn spawn_named_with_send_on_failure<T: Send>(name: &'static str,
                                                 state: task_state::TaskState,
                                                 f: proc(): Send,
                                                 msg: T,
                                                 dest: Sender<T>,
                                                 native: bool) {
    let with_state = proc() {
        task_state::initialize(state);
        rtinstrument::instrument(f);
    };

    let future_result = if native {
        TaskBuilder::new().named(name).native().try_future(with_state)
    } else {
        TaskBuilder::new().named(name).try_future(with_state)
    };

    let watched_name = name.into_string();
    let watcher_name = format!("{}Watcher", watched_name);
    TaskBuilder::new().named(watcher_name).spawn(proc() {
        rtinstrument::instrument(proc() {
            match future_result.unwrap() {
                Ok(()) => (),
                Err(..) => {
                    debug!("{} failed, notifying constellation", name);
                    dest.send(msg);
                }
            }
        });
    });
}