use futures::{Async, Poll, task::Task};
use super::Never;
#[derive(Debug)]
pub(crate) struct YieldNow {
cached_task: Option<Task>,
}
impl YieldNow {
pub(crate) fn new() -> YieldNow {
YieldNow {
cached_task: None,
}
}
pub(crate) fn poll_yield(&mut self) -> Poll<Never, Never> {
if let Some(ref t) = self.cached_task {
if t.will_notify_current() {
t.notify();
return Ok(Async::NotReady);
}
}
let t = ::futures::task::current();
t.notify();
self.cached_task = Some(t);
Ok(Async::NotReady)
}
}