job-pool 0.6.0

Thread Pool
Documentation 
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

use std::sync::mpsc::SendError;
use std::sync::{mpsc, Arc, Mutex};

pub fn sync_channel<T>(bound: usize) -> (SenderWrapper<T>, ReceiverWrapper<T>) {
    let (s,r) = mpsc::sync_channel(bound);
    let s = SenderWrapper::Bounded(s);
    let r = ReceiverWrapper(Arc::new(Mutex::new(r)));
    (s,r)
}

pub fn channel<T>() -> (SenderWrapper<T>, ReceiverWrapper<T>) {
    let (s,r) = mpsc::channel();
    let s = SenderWrapper::Unbounded(s);
    let r = ReceiverWrapper(Arc::new(Mutex::new(r)));
    (s,r)
}

pub enum SenderWrapper<T> {
    Bounded(mpsc::SyncSender<T>),
    Unbounded(mpsc::Sender<T>),
}

impl<T> SenderWrapper<T> {
    pub fn send(&self, t: T) -> std::result::Result<(),SendError<T>> {
        match self {
            SenderWrapper::Bounded(b) => b.send(t),
            SenderWrapper::Unbounded(u) => u.send(t),
        }
    }
}

pub struct ReceiverWrapper<T>(pub Arc<Mutex<mpsc::Receiver<T>>>);

impl<T> Clone for ReceiverWrapper<T> {
    fn clone(&self) -> Self {
        Self(Arc::clone(&self.0))
    }
}

impl<T> ReceiverWrapper<T> {
    pub fn recv(&self) -> Result<T, mpsc::RecvError> {
        self.0.lock().unwrap().recv()
    }
}