refactor: use rayon::ThreadPoolBuilder to set num threads

eatradish · eatradish · commit a7f6a38ff99e · 2025-03-19T18:40:35.000+08:00
diff --git a/src/main.rs b/src/main.rs
@@ -303,7 +303,10 @@ fn main() {
         );
     }
     if let Some(jobs) = args.jobs {
-        unsafe { std::env::set_var("RAYON_NUM_THREADS", jobs.to_string()) };
+        rayon::ThreadPoolBuilder::new()
+            .num_threads(jobs)
+            .build_global()
+            .unwrap();
     }
     let mut extra_packages = args.include.clone();
     if let Some(ref extra_files) = args.include_files {

Original file line number	Diff line number	Diff line change
`@@ -303,7 +303,10 @@ fn main() {`
`303`	`303`	`);`
`304`	`304`	`}`
`305`	`305`	`if let Some(jobs) = args.jobs {`
`306`		`- unsafe { std::env::set_var("RAYON_NUM_THREADS", jobs.to_string()) };`
	`306`	`+ rayon::ThreadPoolBuilder::new()`
	`307`	`+ .num_threads(jobs)`
	`308`	`+ .build_global()`
	`309`	`+ .unwrap();`
`307`	`310`	`}`
`308`	`311`	`let mut extra_packages = args.include.clone();`
`309`	`312`	`if let Some(ref extra_files) = args.include_files {`