pkthread 0.1.0

PKaiser's threading utilities.
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

use std::sync::{mpsc::channel, Arc, Barrier};
use pkthread::pool::worker::{Worker, WorkerOptions};

/// Asserts that a [`Worker`] becomes 'busy' if it is processing work.
#[test]
fn worker_is_busy() {
    let worker = spawn_worker();
    let (sender, receiver) = channel::<bool>();
    let barrier = Arc::new(Barrier::new(2));

    let barrier_thread = barrier.clone();
    worker.queue(move || {
        sender
            .send(true)
            .unwrap();
        barrier_thread.wait();
    });

    let received = receiver
        .recv()
        .unwrap();
    if received {
        let is_busy = worker.is_busy();
        assert!(is_busy);
    }

    barrier.wait();
}

/// Asserts that a [`Worker`] executes work.
#[test]
fn work_is_executed() {
    let worker = spawn_worker();
    let (sender, receiver) = channel::<bool>();

    worker.queue(move || {
        sender
            .send(true)
            .unwrap();
    });

    let value = receiver
        .recv()
        .expect("Failed to receive value.");
    
    assert!(value);
}

fn spawn_worker() -> Worker {
    let options = WorkerOptions::default();

    Worker::spawn(options)
        .expect("Failed to initialize the worker thread.")
}