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

replace num_cpus dependency with available_parallelism()

Browse source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2024-05-25 20:58:37 +00:00 • committed by June 🍓🦴
parent d2aef071bc
commit a537462d51
8 changed files with 22 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/database/rocksdb/opts.rs
View file

@ -1,5 +1,7 @@
#![allow(dead_code)]
use std::collections::HashMap;
use std::{cmp, collections::HashMap};
use conduit::utils;
use super::{
rust_rocksdb::{
@ -21,10 +23,11 @@ pub(crate) fn db_options(config: &Config, env: &mut Env, row_cache: &Cache, col_
set_logging_defaults(&mut opts, config);
// Processing
const MIN_PARALLELISM: usize = 2;
let threads = if config.rocksdb_parallelism_threads == 0 {
std::cmp::max(2, num_cpus::get()) // max cores if user specified 0
cmp::max(MIN_PARALLELISM, utils::available_parallelism())
} else {
config.rocksdb_parallelism_threads
cmp::max(MIN_PARALLELISM, config.rocksdb_parallelism_threads)
};
opts.set_max_background_jobs(threads.try_into().unwrap());