feat: add concurrent working threads option to CLI args

ouch-org · Nov 30, 2024 · 639ffbe · 639ffbe
1 parent 8c32d2c
commit 639ffbe
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/cli/args.rs b/src/cli/args.rs
@@ -45,6 +45,10 @@ pub struct CliArgs {
     #[arg(short = 'p', long = "password", global = true)]
     pub password: Option<OsString>,
 
+    /// cocurrent working threads
+    #[arg(short = 't', long, global = true)]
+    pub threads: Option<usize>,
+
     // Ouch and claps subcommands
     #[command(subcommand)]
     pub cmd: Subcommand,

diff --git a/src/cli/mod.rs b/src/cli/mod.rs
@@ -28,6 +28,11 @@ impl CliArgs {
 
         set_accessible(args.accessible);
 
+        rayon::ThreadPoolBuilder::new()
+            .num_threads(args.threads.unwrap_or(0))
+            .build_global()
+            .unwrap();
+
         let (Subcommand::Compress { files, .. }
         | Subcommand::Decompress { files, .. }
         | Subcommand::List { archives: files, .. }) = &mut args.cmd;