refactor: move HDD detection logic to app.rs

it would cost less binary size

Author: KSXGitHub

src/app.rs

@@ -12,8 +12,10 @@ use crate::{
     visualizer::{BarAlignment, Direction, Visualizer},
 };
 use clap::Parser;
+use hdd::any_path_is_in_hdd;
 use pipe_trait::Pipe;
 use std::{io::stdin, time::Duration};
+use sysinfo::Disks;
 
 #[cfg(unix)]
 use crate::{
@@ -44,6 +46,18 @@ impl App {
         //
         // The other operations which are invoked frequently should not utilize dynamic dispatch.
 
+        {
+            // If one of the files is on HDD, set thread number to 1
+            let disks = Disks::new_with_refreshed_list();
+            if any_path_is_in_hdd::<hdd::RealApi>(&self.args.files, &disks) {
+                eprintln!("warning: HDD detected, the thread limit will be set to 1");
+                rayon::ThreadPoolBuilder::new()
+                    .num_threads(1)
+                    .build_global()
+                    .unwrap_or_else(|_| eprintln!("warning: Failed to set thread limit to 1"));
+            }
+        }
+
         let column_width_distribution = self.args.column_width_distribution();
 
         if self.args.json_input {
@@ -202,3 +216,6 @@ impl App {
         }
     }
 }
+
+mod hdd;
+mod mount_point;

src/app/sub/hdd.rs renamed to src/app/hdd.rs

@@ -1,6 +1,3 @@
-mod hdd;
-mod mount_point;
-
 use crate::{
     args::Fraction,
     data_tree::{DataTree, DataTreeReflection},
@@ -14,10 +11,8 @@ use crate::{
     status_board::GLOBAL_STATUS_BOARD,
     visualizer::{BarAlignment, ColumnWidthDistribution, Direction, Visualizer},
 };
-use hdd::any_path_is_in_hdd;
 use serde::Serialize;
 use std::{io::stdout, iter::once, num::NonZeroUsize, path::PathBuf};
-use sysinfo::Disks;
 
 /// The sub program of the main application.
 pub struct Sub<Size, SizeGetter, Report>
@@ -74,17 +69,6 @@ where
             no_sort,
         } = self;
 
-        // If one of the files is on HDD, set thread number to 1
-        let disks = Disks::new_with_refreshed_list();
-
-        if any_path_is_in_hdd::<hdd::RealApi>(&files, &disks) {
-            eprintln!("warning: HDD detected, the thread limit will be set to 1");
-            rayon::ThreadPoolBuilder::new()
-                .num_threads(1)
-                .build_global()
-                .unwrap_or_else(|_| eprintln!("warning: Failed to set thread limit to 1"));
-        }
-
         let mut iter = files
             .into_iter()
             .map(|root| -> DataTree<OsStringDisplay, Size> {