forked from mirror/async-std
Improve thread creating algorithm in spawn_blocking
This commit is contained in:
parent
980c30e90f
commit
6677d52c2d
1 changed files with 18 additions and 7 deletions
|
@ -68,16 +68,13 @@ static POOL: Lazy<Pool> = Lazy::new(|| {
|
||||||
|
|
||||||
fn start_thread() {
|
fn start_thread() {
|
||||||
SLEEPING.fetch_add(1, Ordering::SeqCst);
|
SLEEPING.fetch_add(1, Ordering::SeqCst);
|
||||||
|
let timeout = Duration::from_secs(10);
|
||||||
// Generate a random duration of time between 1 second and 10 seconds. If the thread doesn't
|
|
||||||
// receive the next task in this duration of time, it will stop running.
|
|
||||||
let timeout = Duration::from_millis(1000 + u64::from(random(9_000)));
|
|
||||||
|
|
||||||
thread::Builder::new()
|
thread::Builder::new()
|
||||||
.name("async-std/blocking".to_string())
|
.name("async-std/blocking".to_string())
|
||||||
.spawn(move || {
|
.spawn(move || {
|
||||||
loop {
|
loop {
|
||||||
let task = match POOL.receiver.recv_timeout(timeout) {
|
let mut task = match POOL.receiver.recv_timeout(timeout) {
|
||||||
Ok(task) => task,
|
Ok(task) => task,
|
||||||
Err(_) => {
|
Err(_) => {
|
||||||
// Check whether this is the last sleeping thread.
|
// Check whether this is the last sleeping thread.
|
||||||
|
@ -100,8 +97,22 @@ fn start_thread() {
|
||||||
start_thread();
|
start_thread();
|
||||||
}
|
}
|
||||||
|
|
||||||
// Run the task.
|
loop {
|
||||||
abort_on_panic(|| task.run());
|
// Run the task.
|
||||||
|
abort_on_panic(|| task.run());
|
||||||
|
|
||||||
|
// Try taking another task if there are any available.
|
||||||
|
task = match POOL.receiver.try_recv() {
|
||||||
|
Ok(task) => task,
|
||||||
|
Err(_) => break,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// If there is at least one sleeping thread, stop this thread instead of putting it
|
||||||
|
// to sleep.
|
||||||
|
if SLEEPING.load(Ordering::SeqCst) > 0 {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
SLEEPING.fetch_add(1, Ordering::SeqCst);
|
SLEEPING.fetch_add(1, Ordering::SeqCst);
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue