byor 1.0.0

Bring your own runtime!
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

use crate::{channel::mpsc::*, runtime::Tokio};
use tokio::sync::mpsc::{Sender as TokioSender, UnboundedSender as TokioUnboundedSender};

impl<T: 'static> Sender<T> for TokioSender<T> {
    type SendError = tokio::sync::mpsc::error::SendError<T>;

    fn is_closed(&self) -> bool {
        self.is_closed()
    }
}

impl<T: 'static> SenderExt<T> for TokioSender<T> {
    fn closed(&mut self) -> impl Future<Output = ()> {
        TokioSender::closed(self)
    }

    fn same_channel(&self, other: &Self) -> bool {
        self.same_channel(other)
    }
}

impl<T: 'static> BoundedSender<T> for TokioSender<T> {
    type TrySendError = tokio::sync::mpsc::error::TrySendError<T>;

    fn send(&mut self, message: T) -> impl Future<Output = Result<(), Self::SendError>> {
        TokioSender::send(self, message)
    }

    fn try_send(&mut self, message: T) -> Result<(), Self::TrySendError> {
        TokioSender::try_send(self, message)
    }
}

impl<T: 'static> Receiver<T> for tokio_stream::wrappers::ReceiverStream<T> {
    type TryRecvError = tokio::sync::mpsc::error::TryRecvError;

    fn close(&mut self) {
        self.close()
    }

    fn try_recv(&mut self) -> Result<Option<T>, Self::TryRecvError> {
        match self.as_mut().try_recv() {
            Ok(t) => Ok(Some(t)),
            Err(Self::TryRecvError::Empty) => Ok(None),
            Err(Self::TryRecvError::Disconnected) => Err(Self::TryRecvError::Disconnected),
        }
    }
}

impl<T: 'static> Sender<T> for TokioUnboundedSender<T> {
    type SendError = <TokioSender<T> as Sender<T>>::SendError;

    fn is_closed(&self) -> bool {
        self.is_closed()
    }
}

impl<T: 'static> SenderExt<T> for TokioUnboundedSender<T> {
    fn closed(&mut self) -> impl Future<Output = ()> {
        TokioUnboundedSender::closed(self)
    }

    fn same_channel(&self, other: &Self) -> bool {
        self.same_channel(other)
    }
}

impl<T: 'static> UnboundedSender<T> for TokioUnboundedSender<T> {
    fn send(&self, message: T) -> Result<(), Self::SendError> {
        self.send(message)
    }
}

impl<T: 'static> Receiver<T> for tokio_stream::wrappers::UnboundedReceiverStream<T> {
    type TryRecvError = tokio::sync::mpsc::error::TryRecvError;

    fn close(&mut self) {
        self.close()
    }

    fn try_recv(&mut self) -> Result<Option<T>, Self::TryRecvError> {
        match self.as_mut().try_recv() {
            Ok(t) => Ok(Some(t)),
            Err(Self::TryRecvError::Empty) => Ok(None),
            Err(Self::TryRecvError::Disconnected) => Err(Self::TryRecvError::Disconnected),
        }
    }
}

impl RuntimeMpsc for Tokio {
    type BoundedSender<T: 'static> = tokio::sync::mpsc::Sender<T>;
    type BoundedReceiver<T: 'static> = tokio_stream::wrappers::ReceiverStream<T>;

    fn bounded_channel<T: 'static>(
        buffer: usize,
    ) -> (Self::BoundedSender<T>, Self::BoundedReceiver<T>) {
        let (tx, rx) = tokio::sync::mpsc::channel(buffer);

        (tx, rx.into())
    }

    type UnboundedSender<T: 'static> = tokio::sync::mpsc::UnboundedSender<T>;
    type UnboundedReceiver<T: 'static> = tokio_stream::wrappers::UnboundedReceiverStream<T>;

    fn unbounded_channel<T: 'static>() -> (Self::UnboundedSender<T>, Self::UnboundedReceiver<T>) {
        let (tx, rx) = tokio::sync::mpsc::unbounded_channel();

        (tx, rx.into())
    }
}