shuumatsu / waitgroup-rs Goto Github PK

use std::thread;

let wg = WaitGroup::new();
let (tx, rx) = mpsc::channel();

for _ in 0..THREADS {
    let wg = wg.clone();
    let tx = tx.clone();

    thread::spawn(move || {
        wg.wait();
        tx.send(()).unwrap();
    });
}

thread::sleep(Duration::from_millis(100));

// At this point, all spawned threads should be blocked, so we shouldn't get anything from the channel.
assert!(rx.try_recv().is_err());

wg.wait();

// Now, the wait group is cleared and we should receive messages.
for _ in 0..THREADS {
    rx.recv().unwrap();
}