rill-engine 0.41.0

The RillRate Protocol async engine.
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

use futures::channel::mpsc;
use futures::lock::Mutex;
use meio::{Actor, Parcel};
use thiserror::Error;

#[derive(Error, Debug)]
#[error("Reserved receiver already taken.")]
pub struct AlreadyTaken;

pub(crate) struct ParcelDistributor<A: Actor> {
    pub sender: mpsc::UnboundedSender<Parcel<A>>,
    pub receiver: Mutex<Option<mpsc::UnboundedReceiver<Parcel<A>>>>,
}

impl<A: Actor> ParcelDistributor<A> {
    pub fn new() -> Self {
        let (tx, rx) = mpsc::unbounded();
        let receiver = Mutex::new(Some(rx));
        Self {
            sender: tx,
            receiver,
        }
    }

    pub async fn take_receiver(&self) -> Result<mpsc::UnboundedReceiver<Parcel<A>>, AlreadyTaken> {
        self.receiver.lock().await.take().ok_or(AlreadyTaken)
    }
}