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() {
handle.join().expect(context)?
}
Ok(())
}
pub(crate) fn is_current(&self) -> bool {
self.0
.lock()
.as_ref()
.map(|handle| handle.thread().id() == thread::current().id())
.unwrap_or_default()
}
}