rust
/
smol
mirror of https://github.com/stjepang/smol
1
0
Fork 0
Code Issues Releases Wiki Activity

bring back SMOL_THREADS 

Signed-off-by: Marc-Antoine Perennou <Marc-Antoine@Perennou.com>
This commit is contained in:
Marc-Antoine Perennou 2020-08-27 12:21:56 +02:00
parent 396fb2006a
commit c39584d4b5
1 changed files with 18 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
src/lib.rs
View File

@ -79,9 +79,11 @@ pub mod prelude {
/// Spawns a task onto the single-threaded global executor.
///
/// There is a single-threaded global executor that gets lazily initialized on first use. It is
/// advisable to use it in tests or small programs, but it is otherwise a better idea to define
/// There is a (single-threaded by default) global executor that gets lazily initialized on first use.
/// It is advisable to use it in tests or small programs, but it is otherwise a better idea to define
/// your own [`Executor`]s.
/// You can configure the number of threads of the global executor using the `SMOL_THREADS` environment
/// variable.
///
/// # Examples
///
@ -96,15 +98,22 @@ pub mod prelude {
/// ```
pub fn spawn<T: Send + 'static>(future: impl Future<Output = T> + Send + 'static) -> Task<T> {
static GLOBAL: Lazy<Executor> = Lazy::new(|| {
thread::Builder::new()
.name("smol".to_string())
.spawn(|| {
loop {
let num_threads = {
// Parse SMOL_THREADS or run a monothreaded executor.
std::env::var("SMOL_THREADS")
.ok()
.and_then(|s| s.parse().ok())
.unwrap_or(1)
};
for n in 1..=num_threads {
thread::Builder::new()
.name(format!("smol-{}", n))
.spawn(|| loop {
let _ =
catch_unwind(|| async_io::block_on(GLOBAL.run(future::pending::<()>())));
}
})
.unwrap();
})
.expect("cannot spawn executor thread");
}
Executor::new()
});
GLOBAL.spawn(future)