YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-04-02 10:45:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

aquatic_udp: set core affinities

Browse source
This commit is contained in:
Joakim Frostegård 2021-10-21 20:41:24 +02:00
parent 2f07ba9898
commit 2c4f4a32da
5 changed files with 82 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

32
aquatic_udp/src/lib/mio/mod.rs
View file

@ -24,6 +24,12 @@ use crate::config::Config;
use common::State;
pub fn run(config: Config) -> ::anyhow::Result<()> {
if config.core_affinity.set_affinities {
core_affinity::set_for_current(
core_affinity::CoreId { id: config.core_affinity.offset }
);
}
let state = State::default();
update_access_list(&config, &state.access_list);
@ -86,6 +92,12 @@ pub fn start_workers(
Builder::new()
.name(format!("request-{:02}", i + 1))
.spawn(move || {
if config.core_affinity.set_affinities {
core_affinity::set_for_current(
core_affinity::CoreId { id: config.core_affinity.offset + 1 + i }
);
}
handlers::run_request_worker(state, config, request_receiver, response_sender)
})
.with_context(|| "spawn request worker")?;
@ -101,6 +113,12 @@ pub fn start_workers(
Builder::new()
.name(format!("socket-{:02}", i + 1))
.spawn(move || {
if config.core_affinity.set_affinities {
core_affinity::set_for_current(
core_affinity::CoreId { id: config.core_affinity.offset + 1 + config.request_workers + i }
);
}
network::run_socket_worker(
state,
config,
@ -119,10 +137,18 @@ pub fn start_workers(
Builder::new()
.name("statistics-collector".to_string())
.spawn(move || loop {
::std::thread::sleep(Duration::from_secs(config.statistics.interval));
.spawn(move || {
if config.core_affinity.set_affinities {
core_affinity::set_for_current(
core_affinity::CoreId { id: config.core_affinity.offset }
);
}
tasks::gather_and_print_statistics(&state, &config);
loop {
::std::thread::sleep(Duration::from_secs(config.statistics.interval));
tasks::gather_and_print_statistics(&state, &config);
}
})
.with_context(|| "spawn statistics worker")?;
}