Mark all generated getMetaModel methods generated

1 files changed, 8 insertions, 7 deletions

diff --git a/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetMetaModelGenerator.java b/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetMetaModelGenerator.java
index 62ff9e06a..48ab86b1a 100644
--- a/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetMetaModelGenerator.java
+++ b/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetMetaModelGenerator.java
@@ -3,7 +3,9 @@ package com.github.javaparser.generator.core.node;
 import com.github.javaparser.JavaParser;
 import com.github.javaparser.ast.CompilationUnit;
 import com.github.javaparser.ast.Node;
+import com.github.javaparser.ast.body.BodyDeclaration;
 import com.github.javaparser.ast.body.ClassOrInterfaceDeclaration;
+import com.github.javaparser.ast.body.MethodDeclaration;
 import com.github.javaparser.generator.NodeGenerator;
 import com.github.javaparser.metamodel.BaseNodeMetaModel;
 import com.github.javaparser.metamodel.JavaParserMetaModel;
@@ -19,15 +21,14 @@ public class GetMetaModelGenerator extends NodeGenerator {
 
     @Override
     protected void generateNode(BaseNodeMetaModel nodeMetaModel, CompilationUnit nodeCu, ClassOrInterfaceDeclaration nodeCoid) {
-        nodeCoid.getMethodsByName("getMetaModel").forEach(Node::remove);
-
-        nodeCu.addImport(nodeMetaModel.getClass().getName());
-        nodeCu.addImport(JavaParserMetaModel.class);
-        final String method = f("%s public %s getMetaModel() { return JavaParserMetaModel.%s; }",
+        final MethodDeclaration getMetaModelMethod = (MethodDeclaration)parseClassBodyDeclaration(f("%s public %s getMetaModel() { return JavaParserMetaModel.%s; }",
                 nodeMetaModel.isRootNode() ? "" : "@Override",
                 nodeMetaModel.getClass().getSimpleName(),
-                nodeMetaModel.getMetaModelFieldName());
+                nodeMetaModel.getMetaModelFieldName()));
 
-        nodeCoid.addMember(parseClassBodyDeclaration(method));
+        addOrReplaceWhenSameSignature(nodeCoid, getMetaModelMethod);
+        nodeCu.addImport(nodeMetaModel.getClass().getName());
+        nodeCu.addImport(JavaParserMetaModel.class);
+        markGenerated(getMetaModelMethod);
     }
 }