async-cpupool 0.4.0

A simple async threadpool for CPU-bound tasks
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

#[cfg(not(loom))]
use std::time::Duration;
#[cfg(not(loom))]
use tracing_subscriber::{fmt, prelude::*, EnvFilter};

#[cfg(not(loom))]
fn main() -> Result<(), Box<dyn std::error::Error>> {
    tracing_subscriber::registry()
        .with(fmt::layer())
        .with(EnvFilter::from_default_env())
        .init();

    let pool = async_cpupool::CpuPool::configure()
        .min_threads(1)
        .max_threads(10)
        .buffer_multiplier(8)
        .build()?;

    smol::block_on(async {
        // scale up
        let tasks: Vec<_> = (0..30)
            .map(|i| {
                let pool = pool.clone();

                smol::spawn(async move {
                    tracing::info!("Spawning");
                    let out = pool
                        .spawn(move || {
                            std::thread::sleep(Duration::from_millis(400));
                            i
                        })
                        .await;

                    pool.close().await;

                    out
                })
            })
            .collect();

        for task in tasks {
            let i = task.await?;
            tracing::info!("Awaited {i}");
        }

        // ramp down
        for _ in 0..10 {
            pool.spawn(move || {
                std::thread::sleep(Duration::from_millis(400));
            })
            .await?;
        }

        assert!(pool.close().await);
        Ok(())
    })
}

#[cfg(loom)]
fn main() {}