feat(breeze.rs): make parse_files take in a number of jobs

nvim-neorg · May 6, 2023 · db2c87c · db2c87c
1 parent defd3f1
commit db2c87c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/breeze.rs b/src/breeze.rs
@@ -29,11 +29,12 @@ fn parse_file(filepath: &std::path::PathBuf, parser: &mut Parser) -> Result<Tree
 pub fn parse_files(
     files: Vec<PathBuf>,
     language: Language,
+    num_jobs: Option<usize>,
     callback: &'static (dyn Fn(Tree) + Send + Sync),
 ) {
     let threadpool = Builder::new()
         .thread_name("neorg".into())
-        .num_threads(8)
+        .num_threads(num_jobs.unwrap_or(4))
         .build();
 
     for file in files {
@@ -75,6 +76,7 @@ mod tests {
         parse_files(
             workspace.files(),
             tree_sitter_norg::language(),
+            None,
             &|tree: Tree| assert!(tree.root_node().kind() == "document"),
         );
     }