thread.rs

use crate::Result;
use parking_lot::Mutex;
use std::{sync::Arc, thread};

pub type JoinHandle = thread::JoinHandle<Result<()>>;

#[derive(Clone)]
pub struct ThreadHandle(Arc<Mutex<Option<JoinHandle>>>);

impl Default for ThreadHandle {
    fn default() -> Self {
        Self(Arc::new(Mutex::new(None)))
    }
}

impl ThreadHandle {
    pub fn new(handle: JoinHandle) -> Self {
        Self(Arc::new(Mutex::new(Some(handle))))
    }

    pub(crate) fn register(&self, handle: JoinHandle) {
        *self.0.lock() = Some(handle);
    }

    fn take(&self) -> Option<JoinHandle> {
        self.0.lock().take()
    }

    pub(crate) fn wait(&self, context: &'static str) -> Result<()> {
        if let Some(handle) = self.take() {
            if handle.thread().id() != thread::current().id() {
                handle.join().expect(context)?;
            }
        }
        Ok(())
    }
}