Skip to content

Commit

Permalink
Rollup merge of rust-lang#78065 - tshepang:nits, r=dtolnay
Browse files Browse the repository at this point in the history
make concurrency helper more pleasant to read
  • Loading branch information
m-ou-se authored Nov 7, 2020
2 parents 4d486c9 + 628fb9f commit f21cbb4
Showing 1 changed file with 7 additions and 11 deletions.
18 changes: 7 additions & 11 deletions library/test/src/helpers/concurrency.rs
Original file line number Diff line number Diff line change
@@ -1,18 +1,14 @@
//! Helper module which helps to determine amount of threads to be used
//! during tests execution.
use std::env;
use std::thread;
use std::{env, num::NonZeroUsize, thread};

#[allow(deprecated)]
pub fn get_concurrency() -> usize {
match env::var("RUST_TEST_THREADS") {
Ok(s) => {
let opt_n: Option<usize> = s.parse().ok();
match opt_n {
Some(n) if n > 0 => n,
_ => panic!("RUST_TEST_THREADS is `{}`, should be a positive integer.", s),
}
if let Ok(value) = env::var("RUST_TEST_THREADS") {
match value.parse::<NonZeroUsize>().ok() {
Some(n) => n.get(),
_ => panic!("RUST_TEST_THREADS is `{}`, should be a positive integer.", value),
}
Err(..) => thread::available_concurrency().map(|n| n.get()).unwrap_or(1),
} else {
thread::available_concurrency().map(|n| n.get()).unwrap_or(1)
}
}

0 comments on commit f21cbb4

Please sign in to comment.