diff --git a/src/cli.rs b/src/cli.rs index a8cfb3d..abb5598 100644 --- a/src/cli.rs +++ b/src/cli.rs @@ -498,8 +498,8 @@ pub struct Opts { /// Set number of threads to use for searching & executing (default: number /// of available CPU cores) - #[arg(long, short = 'j', value_name = "num", hide_short_help = true, value_parser = clap::value_parser!(u32).range(1..))] - pub threads: Option, + #[arg(long, short = 'j', value_name = "num", hide_short_help = true, value_parser = str::parse::)] + pub threads: Option, /// Milliseconds to buffer before streaming search results to console /// @@ -688,15 +688,8 @@ impl Opts { self.min_depth.or(self.exact_depth) } - pub fn threads(&self) -> usize { - // This will panic if the number of threads passed in is more than usize::MAX in an environment - // where usize is less than 32 bits (for example 16-bit architectures). It's pretty - // unlikely fd will be running in such an environment, and even more unlikely someone would - // be trying to use that many threads on such an environment, so I think panicing is an - // appropriate way to handle that. - self.threads.map_or_else(default_num_threads, |n| { - std::cmp::max(n.try_into().expect("too many threads"), 1) - }) + pub fn threads(&self) -> NonZeroUsize { + self.threads.unwrap_or_else(default_num_threads) } pub fn max_results(&self) -> Option { @@ -719,13 +712,14 @@ impl Opts { } /// Get the default number of threads to use, if not explicitly specified. -fn default_num_threads() -> usize { +fn default_num_threads() -> NonZeroUsize { // If we can't get the amount of parallelism for some reason, then // default to a single thread, because that is safe. + // Note that the minimum value for a NonZeroUsize is 1. + // Unfortunately, we can't do `NonZeroUsize::new(1).unwrap()` + // in a const context. const FALLBACK_PARALLELISM: NonZeroUsize = NonZeroUsize::MIN; - std::thread::available_parallelism() - .unwrap_or(FALLBACK_PARALLELISM) - .get() + std::thread::available_parallelism().unwrap_or(FALLBACK_PARALLELISM) } #[derive(Copy, Clone, PartialEq, Eq, ValueEnum)] diff --git a/src/main.rs b/src/main.rs index a1bad2b..b661f67 100644 --- a/src/main.rs +++ b/src/main.rs @@ -252,7 +252,7 @@ fn construct_config(mut opts: Opts, pattern_regexps: &[String]) -> Result