fibers/executor/in_place.rs
1// Copyright (c) 2016 DWANGO Co., Ltd. All Rights Reserved.
2// See the LICENSE file at the top-level directory of this distribution.
3
4use futures::Future;
5use std::io;
6use std::time;
7
8use super::Executor;
9use fiber::{self, Spawn};
10use io::poll;
11
12/// An executor that executes spawned fibers and I/O event polling on current thread.
13///
14/// # Examples
15///
16/// An example to calculate fibonacci numbers:
17///
18/// ```
19/// # extern crate fibers;
20/// # extern crate futures;
21/// use fibers::{Spawn, Executor, InPlaceExecutor};
22/// use futures::{Async, Future};
23///
24/// fn fib<H: Spawn + Clone>(n: usize, handle: H) -> Box<dyn Future<Item=usize, Error=()> + Send> {
25/// if n < 2 {
26/// Box::new(futures::finished(n))
27/// } else {
28/// let f0 = handle.spawn_monitor(fib(n - 1, handle.clone()));
29/// let f1 = handle.spawn_monitor(fib(n - 2, handle.clone()));
30/// Box::new(f0.join(f1).map(|(a0, a1)| a0 + a1).map_err(|_| ()))
31/// }
32/// }
33///
34/// let mut executor = InPlaceExecutor::new().unwrap();
35/// let mut monitor = executor.spawn_monitor(fib(7, executor.handle()));
36/// loop {
37/// if let Async::Ready(answer) = monitor.poll().unwrap() {
38/// assert_eq!(answer, 13);
39/// return;
40/// } else {
41/// executor.run_once().unwrap();
42/// }
43/// }
44/// ```
45#[derive(Debug)]
46pub struct InPlaceExecutor {
47 scheduler: fiber::Scheduler,
48 poller: poll::Poller,
49}
50impl InPlaceExecutor {
51 /// Creates a new instance of `InPlaceExecutor`.
52 pub fn new() -> io::Result<Self> {
53 let poller = poll::Poller::new()?;
54 Ok(InPlaceExecutor {
55 scheduler: fiber::Scheduler::new(poller.handle()),
56 poller,
57 })
58 }
59}
60impl Executor for InPlaceExecutor {
61 type Handle = InPlaceExecutorHandle;
62 fn handle(&self) -> Self::Handle {
63 InPlaceExecutorHandle {
64 scheduler: self.scheduler.handle(),
65 }
66 }
67 fn run_once(&mut self) -> io::Result<()> {
68 self.scheduler.run_once(false);
69 self.poller.poll(Some(time::Duration::from_millis(1)))?;
70 Ok(())
71 }
72}
73impl Spawn for InPlaceExecutor {
74 fn spawn_boxed(&self, fiber: Box<dyn Future<Item = (), Error = ()> + Send>) {
75 self.handle().spawn_boxed(fiber)
76 }
77}
78
79/// A handle of an `InPlaceExecutor` instance.
80#[derive(Debug, Clone)]
81pub struct InPlaceExecutorHandle {
82 scheduler: fiber::SchedulerHandle,
83}
84impl Spawn for InPlaceExecutorHandle {
85 fn spawn_boxed(&self, fiber: Box<dyn Future<Item = (), Error = ()> + Send>) {
86 self.scheduler.spawn_boxed(fiber)
87 }
88}