itsscb/tokio
1
0
Fork 0
mirror of https://github.com/tokio-rs/tokio.git synced 2025-09-28 12:10:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

more fix shutdown

Browse Source
This commit is contained in:
Carl Lerche 2023-06-12 12:57:54 -07:00
parent 904dabb23d
commit 22ccfe48c1
No known key found for this signature in database
GPG Key ID: FC5ADF3A4B2E5977
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tokio/src/runtime/scheduler/multi_thread/idle.rs
View File

@ -184,16 +184,24 @@ impl Idle {
}
pub(super) fn shutdown(&self, synced: &mut worker::Synced, shared: &Shared) {
while let Some(mut core) = synced.idle.available_cores.pop() {
// First, set the shutdown flag on each core
for core in &mut synced.idle.available_cores {
core.is_shutdown = true;
}
// Wake every sleeping worker and assign a core to it. There may not be
// enough sleeping workers for all cores, but other workers will
// eventually find the cores and shut them down.
while !synced.idle.sleepers.is_empty() && !synced.idle.available_cores.is_empty() {
let worker = synced.idle.sleepers.pop().unwrap();
let core = synced.idle.available_cores.pop().unwrap();
synced.assigned_cores[worker] = Some(core);
shared.condvars[worker].notify_one();
}
self.num_idle.store(0, Release);
self.num_idle
.store(synced.idle.available_cores.len(), Release);
}
// Wake up any other workers
while let Some(index) = synced.idle.sleepers.pop() {