rafaelcaricio/relay
1
0
Fork 0
forked from mirrors/relay
Code Pull requests Activity

Split available signature threads between sign & verify

Browse source
This commit is contained in:
asonix 2023-07-27 11:08:20 -05:00
parent f24685e700
commit f3ff8ae5f7
2 changed files with 28 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/config.rs
View file

@ -259,12 +259,14 @@ impl Config {
} }
pub(crate) fn signature_threads(&self) -> usize { pub(crate) fn signature_threads(&self) -> usize {
self.signature_threads.unwrap_or_else(|| { self.signature_threads
std::thread::available_parallelism() .unwrap_or_else(|| {
.map(usize::from) std::thread::available_parallelism()
.map_err(|e| tracing::warn!("Failed to get parallelism, {e}")) .map(usize::from)
.unwrap_or(1) .map_err(|e| tracing::warn!("Failed to get parallelism, {e}"))
}) .unwrap_or(1)
})
.max(1)
} }
pub(crate) fn client_timeout(&self) -> u64 { pub(crate) fn client_timeout(&self) -> u64 {

22
src/main.rs
View file

@ -242,6 +242,8 @@ fn server_main(
actix_rt::spawn(do_server_main(db, actors, media, collector, config)) actix_rt::spawn(do_server_main(db, actors, media, collector, config))
} }
const VERIFY_RATIO: usize = 7;
async fn do_server_main( async fn do_server_main(
db: Db, db: Db,
actors: ActorCache, actors: ActorCache,
@ -259,8 +261,24 @@ async fn do_server_main(
let keys = config.open_keys()?; let keys = config.open_keys()?;
let spawner = Spawner::build("signature-thread", config.signature_threads())?; let (signature_threads, verify_threads) = match config.signature_threads() {
let verify_spawner = Spawner::build("verify-thread", (config.signature_threads() / 8).max(1))?; 0 | 1 => (1, 1),
n if n <= VERIFY_RATIO => {
let verify_threads = (n / VERIFY_RATIO).max(1);
let signature_threads = n.saturating_sub(verify_threads).max(n);
(signature_threads, verify_threads)
}
n => {
let verify_threads = (n / VERIFY_RATIO).max(1);
let signature_threads = n.saturating_sub(verify_threads).max(VERIFY_RATIO);
(signature_threads, verify_threads)
}
};
let spawner = Spawner::build("signature-thread", signature_threads)?;
let verify_spawner = Spawner::build("verify-thread", verify_threads)?;
let bind_address = config.bind_address(); let bind_address = config.bind_address();
let server = HttpServer::new(move || { let server = HttpServer::new(move || {