github/ouch
1
1
Fork 0
mirror of https://github.com/ouch-org/ouch.git synced 2025-06-05 02:55:31 +00:00
Code Issues 15 Packages Projects Releases Wiki Activity

feat: add concurrent working threads option to CLI args

Browse Source
This commit is contained in:
ttyS3 2024-11-30 17:43:50 +00:00 committed by João Marcos
parent 195483a182
commit 6b38e1dd46
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/cli/args.rs
View File

@ -45,6 +45,10 @@ pub struct CliArgs {
#[arg(short = 'p', long = "password", global = true)] #[arg(short = 'p', long = "password", global = true)]
pub password: Option<OsString>, pub password: Option<OsString>,
/// cocurrent working threads
#[arg(short = 't', long, global = true)]
pub threads: Option<usize>,
// Ouch and claps subcommands // Ouch and claps subcommands
#[command(subcommand)] #[command(subcommand)]
pub cmd: Subcommand, pub cmd: Subcommand,
@ -142,6 +146,7 @@ mod tests {
format: None, format: None,
// This is usually replaced in assertion tests // This is usually replaced in assertion tests
password: None, password: None,
threads: None,
cmd: Subcommand::Decompress { cmd: Subcommand::Decompress {
// Put a crazy value here so no test can assert it unintentionally // Put a crazy value here so no test can assert it unintentionally
files: vec!["\x00\x11\x22".into()], files: vec!["\x00\x11\x22".into()],

7
src/cli/mod.rs
View File

@ -28,6 +28,13 @@ impl CliArgs {
set_accessible(args.accessible); set_accessible(args.accessible);
if let Some(threads) = args.threads {
rayon::ThreadPoolBuilder::new()
.num_threads(threads)
.build_global()
.unwrap();
}
let (Subcommand::Compress { files, .. } let (Subcommand::Compress { files, .. }
| Subcommand::Decompress { files, .. } | Subcommand::Decompress { files, .. }
| Subcommand::List { archives: files, .. }) = &mut args.cmd; | Subcommand::List { archives: files, .. }) = &mut args.cmd;