magmaus3/continuwuity
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

support core affinity mask

Browse source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2024-12-30 09:35:44 +00:00 • committed by strawberry
parent 0b2fca5ad9
commit 1792711d09
5 changed files with 39 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/database/pool.rs
View file

@ -13,7 +13,7 @@ use conduwuit::{
debug, debug_warn, defer, err, implement,
result::DebugInspect,
trace,
utils::sys::compute::{get_affinity, get_core_available, set_affinity},
utils::sys::compute::{get_affinity, nth_core_available, set_affinity},
Result, Server,
};
use futures::{channel::oneshot, TryFutureExt};
@ -270,7 +270,7 @@ fn worker_init(&self, id: usize) {
.enumerate()
.filter(|_| self.queues.len() > 1)
.filter_map(|(core_id, &queue_id)| (group == queue_id).then_some(core_id))
.filter_map(get_core_available);
.filter_map(nth_core_available);
// affinity is empty (no-op) if there's only one queue
set_affinity(affinity.clone());