perf(es/compat): Migrate class_properties to VisitMut partially (s…

…wc-project#2966)
nicholasxjy · Dec 18, 2021 · a968d9c · a968d9c
1 parent 50cc642
commit a968d9c
Show file tree

Hide file tree

Showing 2 changed files with 133 additions and 140 deletions.
diff --git a/crates/swc_ecma_transforms_compat/src/es2022/class_properties/mod.rs b/crates/swc_ecma_transforms_compat/src/es2022/class_properties/mod.rs
@@ -737,17 +737,19 @@ impl ClassProperties {
             members.push(ClassMember::Constructor(c));
         }
 
-        extra_stmts.extend(private_method_fn_decls.fold_with(&mut FieldAccessFolder {
+        private_method_fn_decls.visit_mut_with(&mut FieldAccessFolder {
             mark: self.mark,
             method_mark: self.method_mark,
             private_methods: &private_methods,
             statics: &statics,
             vars: vec![],
             class_name: &class_ident,
             in_assign_pat: false,
-        }));
+        });
+
+        extra_stmts.extend(private_method_fn_decls);
 
-        let members = members.fold_with(&mut FieldAccessFolder {
+        members.visit_mut_with(&mut FieldAccessFolder {
             mark: self.mark,
             method_mark: self.method_mark,
             private_methods: &private_methods,