spmc.rc

use std::sync::mpsc::{self, Sender, Receiver};
use std::sync::{Arc, Mutex};

#[derive(Clone, Debug)]
pub struct Spmc<T> {
    pub tx: Sender<T>,
    pub rx: Reader<T>,
}

impl<T> Spmc<T> {
    pub fn new() -> Spmc<T> {
        let (tx, rx) = Reader::channel();
        Spmc { tx: tx, rx: rx }
    }
}

#[derive(Clone, Debug)]
pub struct Reader<T> {
    c: Arc<Mutex<Receiver<T>>>,
}

impl<T> Reader<T> {
    fn channel() -> (Sender<T>, Reader<T>) {
        let (tx, rx) = mpsc::channel();
        (tx, Reader { c: Arc::new(Mutex::new(rx)) })
    }
}

impl<T> Iterator for Reader<T> {
    type Item = T;
    fn next(&mut self) -> Option<T> {
        self.c.lock().ok().and_then(|c| c.recv().ok())
    }
}