mirror of
https://github.com/bootandy/dust.git
synced 2026-06-08 11:29:05 +03:00
Compare commits
4 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 7bbecd9f87 | |||
| 137bde1099 | |||
| 0d0206a5a5 | |||
| dfce0eec66 |
+12
-11
@@ -11,6 +11,7 @@ mod utils;
|
|||||||
|
|
||||||
use crate::cli::build_cli;
|
use crate::cli::build_cli;
|
||||||
use std::collections::HashSet;
|
use std::collections::HashSet;
|
||||||
|
use std::panic;
|
||||||
use std::process;
|
use std::process;
|
||||||
use sysinfo::{System, SystemExt};
|
use sysinfo::{System, SystemExt};
|
||||||
|
|
||||||
@@ -156,7 +157,10 @@ fn main() {
|
|||||||
by_filecount,
|
by_filecount,
|
||||||
ignore_hidden: config.get_ignore_hidden(&options),
|
ignore_hidden: config.get_ignore_hidden(&options),
|
||||||
};
|
};
|
||||||
let _rayon = init_rayon();
|
let pool = panic::catch_unwind(init_rayon);
|
||||||
|
if pool.is_err() {
|
||||||
|
eprintln!("Warning: Could not configure threads {:?}", pool.err());
|
||||||
|
}
|
||||||
|
|
||||||
let iso = config.get_iso(&options);
|
let iso = config.get_iso(&options);
|
||||||
let (top_level_nodes, has_errors) = walk_it(simplified_dirs, walk_data);
|
let (top_level_nodes, has_errors) = walk_it(simplified_dirs, walk_data);
|
||||||
@@ -191,17 +195,14 @@ fn main() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn init_rayon() -> Result<(), ThreadPoolBuildError> {
|
fn init_rayon() -> Result<(), ThreadPoolBuildError> {
|
||||||
let large_stack = usize::pow(1024, 3);
|
let s = System::new_all();
|
||||||
// Warning: Creating System is slow, takes ~ 100ms
|
let av = s.get_available_memory();
|
||||||
let s = System::new();
|
let free = s.get_free_memory();
|
||||||
let available = s.get_available_memory() * 1024;
|
println!("{}", av);
|
||||||
|
println!("{}", free);
|
||||||
if available > large_stack.try_into().unwrap() {
|
|
||||||
// Larger stack size to handle cases with lots of nested directories
|
// Larger stack size to handle cases with lots of nested directories
|
||||||
rayon::ThreadPoolBuilder::new()
|
rayon::ThreadPoolBuilder::new()
|
||||||
.stack_size(large_stack)
|
.stack_size(usize::pow(1024, 3))
|
||||||
.build_global()
|
.build_global()
|
||||||
} else {
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user