taskio 0.1.0-alpha-1

Abstractions for non-blocking tasks and IO
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

use crate::{Poll, Task};
use core::{
    marker::PhantomData,
    ops::{Generator, GeneratorState},
    pin::Pin,
};
use pin_project_lite::pin_project;

pub fn from_generator<O, G: Generator<(), Yield = (), Return = O>>(
    generator: G,
) -> FromGenerator<O, G> {
    FromGenerator {
        generator,
        _output: PhantomData,
    }
}

pin_project! {
    pub struct FromGenerator<O, G> {
        #[pin]
        generator: G,
        _output: PhantomData<O>,
    }
}

impl<O, G> Task for FromGenerator<O, G>
where
    G: Generator<(), Yield = (), Return = O>,
{
    type Output = O;

    fn poll(self: Pin<&mut Self>) -> Poll<Self::Output> {
        match self.project().generator.resume(()) {
            GeneratorState::Yielded(()) => Poll::Pending,
            GeneratorState::Complete(output) => Poll::Ready(output),
        }
    }
}