asynkit 0.1.0

Abstractions for runtime-independent crates
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

use futures::FutureExt;

use crate::oneshot::{Oneshot, Receiver, Sender, TryRecvError};

pub struct TokioOneshot {}

impl Oneshot for TokioOneshot {
    type Receiver<T: Send> = tokio::sync::oneshot::Receiver<T>;
    type Sender<T: Send> = tokio::sync::oneshot::Sender<T>;

    fn channel<T: Send>() -> (Self::Sender<T>, Self::Receiver<T>) {
        tokio::sync::oneshot::channel()
    }
}

impl<T: Send> Sender<T> for tokio::sync::oneshot::Sender<T> {
    fn send(self, value: T) -> Result<(), T> {
        self.send(value)
    }
}

impl<T: Send> Receiver<T> for tokio::sync::oneshot::Receiver<T> {
    fn recv(&mut self) -> impl Future<Output = Result<T, TryRecvError>> {
        self.map(|res| match res {
            Ok(val) => Ok(val),
            Err(_) => Err(TryRecvError::Disconnected),
        })
    }
}