async_ach-spsc 0.1.3

Async Atomic Channel
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

use crate::heapless::Spsc;
use alloc::sync::Arc;

pub struct Sender<T: Unpin, const N: usize> {
    tx: Arc<Spsc<T, N>>,
}
impl<T: Unpin, const N: usize> Sender<T, N> {
    pub fn try_send(&mut self, val: T) -> Result<(), T> {
        self.tx.take_sender().unwrap().try_send(val)
    }
    pub async fn send(&mut self, mut val: T) {
        self.tx.take_sender().unwrap().send(val).await
    }
}

pub struct Receiver<T: Unpin, const N: usize> {
    rx: Arc<Spsc<T, N>>,
}
impl<T: Unpin, const N: usize> Receiver<T, N> {
    pub fn try_recv(&mut self) -> Option<T> {
        self.rx.take_recver().unwrap().try_recv()
    }
    pub async fn recv(&mut self) -> T {
        self.rx.take_recver().unwrap().recv().await
    }
}

pub fn channel<T: Unpin, const N: usize>() -> (Sender<T, N>, Receiver<T, N>) {
    let tx = Arc::new(Spsc::new());
    let rx = tx.clone();
    (Sender { tx }, Receiver { rx })
}