chore: cleanup.

oxc-project · May 2, 2024 · e879908 · e879908
1 parent f8716e9
commit e879908
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/crates/oxc_macros/src/ast_node.rs b/crates/oxc_macros/src/ast_node.rs
@@ -244,13 +244,7 @@ fn generate_traversable_vec_methods(v: &mut Vec<ImplItemFn>, field: &Field) {
         (lifetime, transform_type(generic_ty.clone()))
     };
 
-    let generator = TraversableStructVecMethodsGenerator { ident, generic_ty };
-
-    if is_ast_enum_type_name(&type_name(generic_ty)) {
-        generator.generate_as_enum(v);
-    } else {
-        generator.generate_as_struct(v);
-    }
+    TraversableStructVecMethodsGenerator { ident, generic_ty }.generate(v);
 }
 
 struct TraversableStructVecMethodsGenerator<'a> {
@@ -301,6 +295,14 @@ impl<'a> TraversableStructVecMethodsGenerator<'a> {
             }
         }
     }
+
+    fn generate(self, v: &mut Vec<ImplItemFn>) {
+        if is_ast_enum_type_name(&type_name(self.generic_ty)) {
+            self.generate_as_enum(v);
+        } else {
+            self.generate_as_struct(v);
+        }
+    }
 }
 
 fn generate_traversable_struct_method(_: &mut Vec<ImplItemFn>, _: &Field) {}