类似于Golang的WaitGroup实现，适用于同步/异步Rust，支持no_std环境。

简介

默认情况下，阻塞版本的WaitGroup已启用。

如果您使用其他异步运行时，您需要在您的Cargo.toml中启用wg::AsyncWaitGroup功能，并使用。

安装

• std

  [dependencies]
  wg = "0.9"
  

• future

  [dependencies]
  wg = { version = "0.9", features = ["future"] }
  

• no_std

  [dependencies]
  wg = { version = "0.9", default_features = false, features = ["alloc"] }
  

• no_std & future

  [dependencies]
  wg = { version = "0.9", default_features = false, features = ["alloc", "future"] }
  

示例

同步

use wg::WaitGroup;
use std::sync::Arc;
use std::sync::atomic::{AtomicUsize, Ordering};
use std::time::Duration;
use std::thread::{spawn, sleep};

fn main() {
    let wg = WaitGroup::new();
    let ctr = Arc::new(AtomicUsize::new(0));

    for _ in 0..5 {
        let ctrx = ctr.clone();
        let t_wg = wg.add(1);
        spawn(move || {
            // mock some time consuming task
            sleep(Duration::from_millis(50));
            ctrx.fetch_add(1, Ordering::Relaxed);

            // mock task is finished
            t_wg.done();
        });
    }

    wg.wait();
    assert_eq!(ctr.load(Ordering::Relaxed), 5);
}

异步

use wg::AsyncWaitGroup;
use std::sync::Arc;
use std::sync::atomic::{AtomicUsize, Ordering};
use tokio::{spawn, time::{sleep, Duration}};

#[tokio::main]
async fn main() {
    let wg = AsyncWaitGroup::new();
    let ctr = Arc::new(AtomicUsize::new(0));

    for _ in 0..5 {
        let ctrx = ctr.clone();
        let t_wg = wg.add(1);
        spawn(async move {
            // mock some time consuming task
            sleep(Duration::from_millis(50)).await;
            ctrx.fetch_add(1, Ordering::Relaxed);

            // mock task is finished
            t_wg.done();
        });
    }

    wg.wait().await;
    assert_eq!(ctr.load(Ordering::Relaxed), 5);
}

致谢

• 灵感来源于Golang的sync.WaitGroup和crossbeam_utils::WaitGroup。

许可证