job-pool 0.6.0

Thread Pool
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
55
56
57
58
59
60
61
62
63
64
65
66
67

use std::ptr;

use crate::ThreadPool;

#[repr(C)]
pub struct PoolConfig {
    n_workers: u16,
    max_jobs: i32,
    incoming_buf_size: i32,
}

impl PoolConfig {
    fn convert(&self) -> crate::PoolConfig {
        crate::PoolConfig {
            n_workers: self.n_workers,
            max_jobs: if self.max_jobs > 0 { Some(self.max_jobs as u16) } else { None },
            incoming_buf_size: if self.incoming_buf_size > 0 { Some(self.incoming_buf_size as u16) } else { None },
        }
    }
}

#[unsafe(no_mangle)]
pub extern "C"
fn pool_default_conf() -> PoolConfig {
    PoolConfig {
        n_workers: 16,
        max_jobs: -1,
        incoming_buf_size: -1,
    }
}

#[unsafe(no_mangle)]
pub extern "C"
fn pool_init(conf: PoolConfig) -> *mut ThreadPool {
    let conf = conf.convert();

    let Ok(pool) = ThreadPool::new(conf) else {
        return ptr::null_mut()
    };

    let pool = Box::new(pool);
    Box::into_raw(pool)
}

#[unsafe(no_mangle)]
pub extern "C"
fn pool_execute_job(pool: *mut ThreadPool, f: extern "C" fn ()) {
    unsafe {
        (*pool).execute(move || { f(); });
    }
}

#[unsafe(no_mangle)]
pub extern "C"
fn pool_join(pool: *mut ThreadPool) {
    unsafe { (*pool).join(); }
}

#[unsafe(no_mangle)]
pub extern "C"
fn pool_free(pool: *mut ThreadPool) {
    if pool.is_null() { return }
    unsafe {
        let pool = Box::from_raw(pool);
        drop(pool);
    }
}