use crate::Result; use crossbeam_channel::{bounded, unbounded, Receiver, Sender}; use drop_bomb::DropBomb; use std::thread; pub struct Worker { pub inp: Sender, pub out: Receiver, } impl Worker { pub fn spawn(name: &'static str, buf: usize, f: F) -> (Self, ThreadWatcher) where F: FnOnce(Receiver, Sender) + Send + 'static, I: Send + 'static, O: Send + 'static, { let ((inp, out), inp_r, out_s) = worker_chan(buf); let worker = Worker { inp, out }; let watcher = ThreadWatcher::spawn(name, move || f(inp_r, out_s)); (worker, watcher) } pub fn stop(self) -> Receiver { self.out } pub fn send(&self, item: I) { self.inp.send(item) } } pub struct ThreadWatcher { name: &'static str, thread: thread::JoinHandle<()>, bomb: DropBomb, } impl ThreadWatcher { fn spawn(name: &'static str, f: impl FnOnce() + Send + 'static) -> ThreadWatcher { let thread = thread::spawn(f); ThreadWatcher { name, thread, bomb: DropBomb::new(format!("ThreadWatcher {} was not stopped", name)), } } pub fn stop(mut self) -> Result<()> { info!("waiting for {} to finish ...", self.name); let name = self.name; self.bomb.defuse(); let res = self .thread .join() .map_err(|_| format_err!("ThreadWatcher {} died", name)); match &res { Ok(()) => info!("... {} terminated with ok", name), Err(_) => error!("... {} terminated with err", name), } res } } /// Sets up worker channels in a deadlock-avoind way. /// If one sets both input and output buffers to a fixed size, /// a worker might get stuck. fn worker_chan(buf: usize) -> ((Sender, Receiver), Receiver, Sender) { let (input_sender, input_receiver) = bounded::(buf); let (output_sender, output_receiver) = unbounded::(); ( (input_sender, output_receiver), input_receiver, output_sender, ) }