29 lines
629 B
Rust
29 lines
629 B
Rust
use tokio::sync::mpsc::{channel, Receiver as _Receiver, Sender as _Sender};
|
|
|
|
pub struct Receiver<T>(_Receiver<T>);
|
|
|
|
impl<T> Receiver<T> {
|
|
pub(super) async fn recv(&mut self) -> Option<T> {
|
|
self.0.recv().await
|
|
}
|
|
}
|
|
|
|
#[derive(Clone)]
|
|
pub struct Sender<T>(_Sender<T>);
|
|
|
|
// TODO: Handle error
|
|
impl<T> Sender<T> {
|
|
pub(super) async fn send(self, t: T) {
|
|
let _ = self.0.send(t).await;
|
|
}
|
|
}
|
|
|
|
// TODO: Rename the name of the following fn
|
|
pub fn oneshot<T>() -> (Sender<T>, Receiver<T>) {
|
|
let (tx, rx) = channel(32);
|
|
let sender = Sender(tx);
|
|
let receiver = Receiver(rx);
|
|
|
|
(sender, receiver)
|
|
}
|