Fix multi-threading issue.

Findomain · Nov 14, 2019 · 853673d · 853673d
1 parent eae4dba
commit 853673d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -543,10 +543,6 @@ fn async_resolver(args: &mut args::Args) -> HashMap<&String, String> {
     } else if args.subdomains.len() > 50 && args.subdomains.len() < 500 {
         args.threads = 50
     }
-    rayon::ThreadPoolBuilder::new()
-        .num_threads(args.threads)
-        .build()
-        .unwrap();
     if !args.quiet_flag {
         println!(
             "Performing asynchronous resolution for {} subdomains with {} threads, it will take a while. 🧐\n",

diff --git a/src/main.rs b/src/main.rs
@@ -3,6 +3,10 @@ use findomain::{args, get_subdomains, read_from_file};
 
 fn run() -> Result<()> {
     let mut arguments = args::get_args();
+    rayon::ThreadPoolBuilder::new()
+        .num_threads(arguments.threads)
+        .build_global()
+        .unwrap();
     if !arguments.target.is_empty() {
         get_subdomains(&mut arguments)
     } else if !arguments.file.is_empty() {