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
30
31
32
33
34

use futures::FutureExt;
use tokio::sync::mpsc::error::SendError;

use crate::mpsc::{Mpsc, Receiver, Sender, TryRecvError, TrySendError};

pub struct TokioMpsc {}

impl Mpsc for TokioMpsc {
    type Receiver<T: Send> = tokio::sync::mpsc::Receiver<T>;

    type Sender<T: Send> = tokio::sync::mpsc::Sender<T>;

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

impl<T: Send> Sender<T> for tokio::sync::mpsc::Sender<T> {
    fn send(&self, value: T) -> impl Future<Output = Result<(), TrySendError<T>>> {
        self.send(value).map(|res| match res {
            Ok(_) => Ok(()),
            Err(SendError(val)) => Err(TrySendError::Closed(val)),
        })
    }
}

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