use std::{
future::Future,
marker::PhantomData,
pin::Pin,
task::{Context, Poll},
};
use super::{Result, Task};
pub struct JoinHandle<T> {
task: Task,
_mark: PhantomData<T>,
}
impl<T> JoinHandle<T> {
pub(super) fn new(task: Task) -> Self {
Self {
task,
_mark: PhantomData,
}
}
pub fn task(&self) -> &Task {
&self.task
}
}
impl<T> Drop for JoinHandle<T> {
fn drop(&mut self) {
self.task.detach();
}
}
impl<T> Unpin for JoinHandle<T> {}
impl<T> Future for JoinHandle<T> {
type Output = Result<T>;
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
self.task.join(cx.waker())
}
}