Use rayon

kdy1 · May 4, 2021 · 86d18a7 · 86d18a7
1 parent 25d5639
commit 86d18a7
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 5 deletions.
diff --git a/native/babel-compat/src/lib.rs b/native/babel-compat/src/lib.rs
@@ -1,5 +1,6 @@
 #![feature(type_name_of_val)]
 
+use rayon::prelude::*;
 use serde::de::DeserializeOwned;
 use serde::Serialize;
 use std::sync::Arc;
@@ -33,7 +34,7 @@ mod typescript;
 pub struct Context {
     pub fm: Arc<SourceFile>,
     pub cm: Lrc<SourceMap>,
-    pub comments: Arc<dyn Comments>,
+    pub comments: Arc<dyn Comments + Send + Sync>,
 }
 
 impl Context {
@@ -128,19 +129,27 @@ impl Context {
 }
 
 pub trait Babelify {
-    type Output: Serialize + DeserializeOwned;
+    type Output: Serialize + DeserializeOwned + Send;
+
+    fn parallel(_cnt: usize) -> bool {
+        false
+    }
 
     fn babelify(self, ctx: &Context) -> Self::Output;
 }
 
 impl<T> Babelify for Vec<T>
 where
-    T: Babelify,
+    T: Babelify + Send,
 {
     type Output = Vec<T::Output>;
 
     fn babelify(self, ctx: &Context) -> Self::Output {
-        self.into_iter().map(|v| v.babelify(ctx)).collect()
+        if T::parallel(self.len()) {
+            self.into_par_iter().map(|v| v.babelify(ctx)).collect()
+        } else {
+            self.into_iter().map(|v| v.babelify(ctx)).collect()
+        }
     }
 }
 

diff --git a/native/babel-compat/src/module.rs b/native/babel-compat/src/module.rs
@@ -127,6 +127,10 @@ pub enum ModuleItemOutput {
 impl Babelify for ModuleItem {
     type Output = ModuleItemOutput;
 
+    fn parallel(cnt: usize) -> bool {
+        cnt >= 32
+    }
+
     fn babelify(self, ctx: &Context) -> Self::Output {
         match self {
             ModuleItem::ModuleDecl(d) => ModuleItemOutput::ModuleDecl(d.babelify(ctx).into()),
@@ -168,7 +172,7 @@ fn extract_all_comments(program: &Program, ctx: &Context) -> Vec<Comment> {
 }
 
 struct CommentCollector {
-    comments: Arc<dyn Comments>,
+    comments: Arc<dyn Comments + Send + Sync>,
     collected: Vec<Comment>,
 }
 

diff --git a/native/babel-compat/src/stmt.rs b/native/babel-compat/src/stmt.rs
@@ -28,6 +28,10 @@ impl Babelify for BlockStmt {
 impl Babelify for Stmt {
     type Output = Statement;
 
+    fn parallel(cnt: usize) -> bool {
+        32 >= cnt
+    }
+
     fn babelify(self, ctx: &Context) -> Self::Output {
         match self {
             Stmt::Block(s) => Statement::Block(s.babelify(ctx)),