uexec 0.1.4

Simple work-stealing global and local executor
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

use std::thread::{self, JoinHandle};

use parking_lot::Mutex;

/* Control the pool of workers */
#[derive(Default)]
pub(crate) struct Workers(Mutex<Vec<Worker>>);

impl Workers {
    pub(crate) fn spawn(&self, threads: u8) {
        let mut workers = self.0.lock();
        for _ in 0..threads {
            let (sender, receiver) = flume::bounded(1);
            let handle = thread::spawn(|| {
                crate::block_on(async move {
                    let _ = receiver.recv_async().await;
                })
            });
            workers.push(Worker {
                trigger: sender,
                thread: handle,
            });
        }
    }

    pub(crate) fn terminate(&self) {
        let mut workers = self.0.lock();
        for worker in workers.iter() {
            worker.trigger_termination();
        }
        for worker in workers.drain(..) {
            worker.terminate();
        }
    }
}

struct Worker {
    trigger: flume::Sender<()>,
    thread: JoinHandle<()>,
}

impl Worker {
    fn trigger_termination(&self) {
        let _ = self.trigger.send(());
    }

    fn terminate(self) {
        if let Err(err) = self.thread.join() {
            std::panic::resume_unwind(err);
        }
    }
}