Mark all generated visitor methods generated

javaparser · Apr 22, 2017 · d72a025 · d72a025
1 parent 4e845ed
commit d72a025
Show file tree

Hide file tree

Showing 11 changed files with 908 additions and 6 deletions.
diff --git a/...arser-core-generators/src/main/java/com/github/javaparser/generator/VisitorGenerator.java b/...arser-core-generators/src/main/java/com/github/javaparser/generator/VisitorGenerator.java
@@ -58,24 +58,26 @@ protected void after() throws Exception {
     }
 
     private void generateVisitMethodForNode(BaseNodeMetaModel node, ClassOrInterfaceDeclaration visitorClass, CompilationUnit compilationUnit) {
-        Optional<MethodDeclaration> visitMethod = visitorClass.getMethods().stream()
+        final Optional<MethodDeclaration> existingVisitMethod = visitorClass.getMethods().stream()
                 .filter(m -> m.getNameAsString().equals("visit"))
                 .filter(m -> m.getParameter(0).getType().toString().equals(node.getTypeName()))
                 .findFirst();
 
-        if (visitMethod.isPresent()) {
+        if (existingVisitMethod.isPresent()) {
-            generateVisitMethodBody(node, visitMethod.get(), compilationUnit);
+            generateVisitMethodBody(node, existingVisitMethod.get(), compilationUnit);
+            markGenerated(existingVisitMethod.get());
         } else if (createMissingVisitMethods) {
-            MethodDeclaration methodDeclaration = visitorClass.addMethod("visit")
+            MethodDeclaration newVisitMethod = visitorClass.addMethod("visit")
                     .addParameter(node.getTypeNameGenerified(), "n")
                     .addParameter(argumentType, "arg")
                     .setType(returnType);
             if (!visitorClass.isInterface()) {
-                methodDeclaration
+                newVisitMethod
                         .addAnnotation(new MarkerAnnotationExpr(new Name("Override")))
                         .addModifier(PUBLIC);
             }
-            generateVisitMethodBody(node, methodDeclaration, compilationUnit);
+            generateVisitMethodBody(node, newVisitMethod, compilationUnit);
+            markGenerated(newVisitMethod);
         }
     }
 

diff --git a/javaparser-core/src/main/java/com/github/javaparser/ast/visitor/CloneVisitor.java b/javaparser-core/src/main/java/com/github/javaparser/ast/visitor/CloneVisitor.java
diff --git a/javaparser-core/src/main/java/com/github/javaparser/ast/visitor/EqualsVisitor.java b/javaparser-core/src/main/java/com/github/javaparser/ast/visitor/EqualsVisitor.java