unknownrori-simple-thread-pool 0.2.0

#[cfg(feature = "crossbeam")]
use crossbeam_channel::Receiver;

#[cfg(feature = "mpsc")]
use std::sync::mpsc::Receiver;

#[cfg(feature = "mpsc")]
use std::sync::{Arc, Mutex};

use std::io;
use std::thread::{self, JoinHandle};

use crate::message::Message;

#[derive(Debug)]
pub struct Worker {
    thread: Option<JoinHandle<()>>,
}

impl Worker {
    /// Creates a new [`Worker`].
    ///
    /// ## Error
    ///
    /// Will return [`Err`] if it cannot create a thread
    #[cfg(feature = "crossbeam")]
    pub fn new(receiver: Receiver<Message>, thread_builder: thread::Builder) -> io::Result<Worker> {
        let thread = thread_builder.spawn(move || loop {
            if let Ok(message) = receiver.recv() {
                let _ = match message {
                    Message::NewJob(job) => job(),
                    Message::Terminate => break,
                };
            }
        })?;

        Ok(Worker {
            thread: Some(thread),
        })
    }

    /// Creates a new [`Worker`].
    ///
    /// ## Error
    ///
    /// Will return [`Err`] if it cannot create a thread
    #[cfg(feature = "mpsc")]
    pub fn new(
        receiver: Arc<Mutex<Receiver<Message>>>,
        thread_builder: thread::Builder,
    ) -> io::Result<Worker> {
        let thread = thread_builder.spawn(move || loop {
            let _ = match receiver.lock().unwrap().recv().unwrap() {
                Message::NewJob(job) => job(),
                Message::Terminate => break,
            };
        })?;

        Ok(Worker {
            thread: Some(thread),
        })
    }

    /// Take the ownership of [`JoinHandle`]
    pub fn take_thread(&mut self) -> Option<JoinHandle<()>> {
        self.thread.take()
    }
}