blob: 8c004464fa8e74ba01ae0bb9ec3e6a81a7fbe350 (
plain)
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
|
use std::{io, num::NonZeroUsize, sync::Arc, thread};
use blocking::unblock;
use easy_parallel::Parallel;
use smol::{channel, future, future::Future, Executor};
#[allow(dead_code)]
pub async fn read_line_async() -> Result<String, io::Error> {
unblock(|| {
let mut input = String::new();
std::io::stdin().read_line(&mut input)?;
Ok(input)
})
.await
}
/// Returns an estimate of the default amount of parallelism a program should use.
/// see `std::thread::available_parallelism`
pub fn available_parallelism() -> usize {
thread::available_parallelism()
.map(NonZeroUsize::get)
.unwrap_or(1)
}
/// Run a multi-threaded executor
pub fn run_executor<T>(main_future: impl Future<Output = T>, ex: Arc<Executor<'_>>) {
let (signal, shutdown) = channel::unbounded::<()>();
let num_threads = available_parallelism();
Parallel::new()
.each(0..(num_threads), |_| {
future::block_on(ex.run(shutdown.recv()))
})
// Run the main future on the current thread.
.finish(|| {
future::block_on(async {
main_future.await;
drop(signal);
})
});
}
|
