nexus-acto-rs 0.4.2

A Rust crate for Actors
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

use std::sync::Arc;

use tokio::sync::{Mutex, Notify};

#[derive(Debug, Clone)]
pub struct AsyncBarrier {
  notify: Arc<Notify>,
  count: Arc<Mutex<usize>>,
}

impl AsyncBarrier {
  pub fn new(count: usize) -> Self {
    AsyncBarrier {
      notify: Arc::new(Notify::new()),
      count: Arc::new(Mutex::new(count)),
    }
  }

  pub async fn wait(&self) {
    let mut count = self.count.lock().await;
    *count -= 1;
    if *count == 0 {
      self.notify.notify_waiters();
    } else {
      drop(count);
      self.notify.notified().await;
    }
  }
}