mirror of
https://github.com/servo/servo.git
synced 2025-06-15 03:44:30 +00:00
6df1cc8e4c
During debugging, I found it useful to hook all task creation in a central location, and util::task was the perfect place for it. r? @pcwalton (or maybe someone else, I'm kinda sending you a bunch of reviews today because I don't know who better to give them to)
55 lines
1.4 KiB
Rust
55 lines
1.4 KiB
Rust
/* 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/. */
|
|
|
|
//! A load-balancing task pool.
|
|
//!
|
|
//! This differs in implementation from std::sync::TaskPool in that each job is
|
|
//! up for grabs by any of the child tasks in the pool.
|
|
//!
|
|
|
|
//
|
|
// This is based on the cargo task pool.
|
|
// https://github.com/rust-lang/cargo/blob/master/src/cargo/util/pool.rs
|
|
//
|
|
// The only difference is that a normal channel is used instead of a sync_channel.
|
|
//
|
|
|
|
use task::spawn_named;
|
|
use std::sync::{Arc, Mutex};
|
|
|
|
pub struct TaskPool {
|
|
tx: Sender<proc():Send>,
|
|
}
|
|
|
|
impl TaskPool {
|
|
pub fn new(tasks: uint) -> TaskPool {
|
|
assert!(tasks > 0);
|
|
let (tx, rx) = channel();
|
|
|
|
let state = Arc::new(Mutex::new(rx));
|
|
|
|
for i in range(0, tasks) {
|
|
let state = state.clone();
|
|
spawn_named(
|
|
format!("TaskPoolWorker {}/{}", i+1, tasks),
|
|
proc() worker(&*state));
|
|
}
|
|
|
|
return TaskPool { tx: tx };
|
|
|
|
fn worker(rx: &Mutex<Receiver<proc():Send>>) {
|
|
loop {
|
|
let job = rx.lock().recv_opt();
|
|
match job {
|
|
Ok(job) => job(),
|
|
Err(..) => break,
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn execute(&self, job: proc():Send) {
|
|
self.tx.send(job);
|
|
}
|
|
}
|