Attach io_thread_count() to Process

Author: djc

src/diskio/mod.rs

@@ -66,7 +66,6 @@ use std::{fmt::Debug, fs::OpenOptions};
 use anyhow::Result;
 
 use crate::process::Process;
-use crate::utils;
 use crate::utils::notifications::Notification;
 use threaded::PoolReference;
 
@@ -449,7 +448,7 @@ pub(crate) fn get_executor<'a>(
     process: &Process,
 ) -> anyhow::Result<Box<dyn Executor + 'a>> {
     // If this gets lots of use, consider exposing via the config file.
-    Ok(match utils::io_thread_count(process)? {
+    Ok(match process.io_thread_count()? {
         0 | 1 => Box::new(immediate::ImmediateUnpacker::new()),
         n => Box::new(threaded::Threaded::new(notify_handler, n, ram_budget)),
     })

src/install.rs

@@ -42,7 +42,7 @@ impl InstallMethod<'_> {
         // Initialize rayon for use by the remove_dir_all crate limiting the number of threads.
         // This will error if rayon is already initialized but it's fine to ignore that.
         let _ = rayon::ThreadPoolBuilder::new()
-            .num_threads(utils::io_thread_count(self.cfg().process)?)
+            .num_threads(self.cfg().process.io_thread_count()?)
             .build_global();
         let nh = &self.cfg().notify_handler;
         match self {

src/process.rs

@@ -1,18 +1,19 @@
-use std::env;
 use std::ffi::OsString;
 use std::fmt::Debug;
 use std::io;
 use std::io::IsTerminal;
 use std::path::PathBuf;
+use std::str::FromStr;
 #[cfg(feature = "test")]
 use std::{
     collections::HashMap,
     io::Cursor,
     path::Path,
     sync::{Arc, Mutex},
 };
+use std::{env, thread};
 
-use anyhow::{Context, Result};
+use anyhow::{Context, Result, bail};
 #[cfg(feature = "test")]
 use tracing::subscriber::DefaultGuard;
 #[cfg(feature = "test")]
@@ -64,6 +65,26 @@ impl Process {
         home::env::rustup_home_with_env(self).context("failed to determine rustup home dir")
     }
 
+    pub fn io_thread_count(&self) -> anyhow::Result<usize> {
+        if let Ok(n) = self.var("RUSTUP_IO_THREADS") {
+            let threads = usize::from_str(&n).context(
+                "invalid value in RUSTUP_IO_THREADS -- must be a natural number greater than zero",
+            )?;
+            match threads {
+                0 => bail!("RUSTUP_IO_THREADS must be a natural number greater than zero"),
+                _ => return Ok(threads),
+            }
+        };
+
+        Ok(match thread::available_parallelism() {
+            // Don't spawn more than 8 I/O threads unless the user tells us to.
+            // Feel free to increase this value if it improves performance.
+            Ok(threads) => Ord::min(threads.get(), 8),
+            // Unknown for target platform or no permission to query.
+            Err(_) => 1,
+        })
+    }
+
     pub fn var(&self, key: &str) -> Result<String, env::VarError> {
         match self {
             Process::OsProcess(_) => env::var(key),

src/utils/mod.rs

@@ -7,8 +7,6 @@ use std::io::{self, BufReader, Write};
 use std::ops::{BitAnd, BitAndAssign};
 use std::path::{Path, PathBuf};
 use std::process::ExitStatus;
-use std::str::FromStr;
-use std::thread;
 
 use anyhow::{Context, Result, anyhow, bail};
 use retry::delay::{Fibonacci, jitter};
@@ -28,26 +26,6 @@ pub(crate) mod notify;
 pub mod raw;
 pub(crate) mod units;
 
-pub fn io_thread_count(process: &Process) -> anyhow::Result<usize> {
-    if let Ok(n) = process.var("RUSTUP_IO_THREADS") {
-        let threads = usize::from_str(&n).context(
-            "invalid value in RUSTUP_IO_THREADS -- must be a natural number greater than zero",
-        )?;
-        match threads {
-            0 => bail!("RUSTUP_IO_THREADS must be a natural number greater than zero"),
-            _ => return Ok(threads),
-        }
-    };
-
-    Ok(match thread::available_parallelism() {
-        // Don't spawn more than 8 I/O threads unless the user tells us to.
-        // Feel free to increase this value if it improves performance.
-        Ok(threads) => Ord::min(threads.get(), 8),
-        // Unknown for target platform or no permission to query.
-        Err(_) => 1,
-    })
-}
-
 #[must_use]
 #[derive(Debug, PartialEq, Eq)]
 pub struct ExitCode(pub i32);