use crate::multithreading::message::Message;
use std::sync::mpsc;
use std::sync::Arc;
use std::sync::Mutex;
use std::thread;
pub struct ThreadWorker {
id: usize,
pub thread: Option<thread::JoinHandle<()>>,
}
impl ThreadWorker {
pub fn new(id: usize, receiver: Arc<Mutex<mpsc::Receiver<Message>>>) -> ThreadWorker {
let thread = thread::spawn(move || loop {
let message = receiver.lock().unwrap().recv().unwrap();
match message {
Message::NewJob(job) => {
println!("Worker {} got a job; executing.", id);
job();
}
Message::Terminate => {
println!("Worker {} was told to terminate.", id);
break;
}
}
});
ThreadWorker {
id,
thread: Some(thread),
}
}
pub fn id(&self) -> usize {
self.id
}
}