use futures::Future;
use std::io;
use std::time;
use super::Executor;
use fiber::{self, Spawn};
use io::poll;
#[derive(Debug)]
pub struct InPlaceExecutor {
scheduler: fiber::Scheduler,
poller: poll::Poller,
}
impl InPlaceExecutor {
pub fn new() -> io::Result<Self> {
let poller = poll::Poller::new()?;
Ok(InPlaceExecutor {
scheduler: fiber::Scheduler::new(poller.handle()),
poller,
})
}
}
impl Executor for InPlaceExecutor {
type Handle = InPlaceExecutorHandle;
fn handle(&self) -> Self::Handle {
InPlaceExecutorHandle {
scheduler: self.scheduler.handle(),
}
}
fn run_once(&mut self) -> io::Result<()> {
self.scheduler.run_once(false);
self.poller.poll(Some(time::Duration::from_millis(1)))?;
Ok(())
}
}
impl Spawn for InPlaceExecutor {
fn spawn_boxed(&self, fiber: Box<dyn Future<Item = (), Error = ()> + Send>) {
self.handle().spawn_boxed(fiber)
}
}
#[derive(Debug, Clone)]
pub struct InPlaceExecutorHandle {
scheduler: fiber::SchedulerHandle,
}
impl Spawn for InPlaceExecutorHandle {
fn spawn_boxed(&self, fiber: Box<dyn Future<Item = (), Error = ()> + Send>) {
self.scheduler.spawn_boxed(fiber)
}
}