Track processed classes. (#123)

Second round of compilation results in visiting the same class again. This prevents that. Fixes #122
micronaut-projects · Jul 23, 2024 · 3a884e6 · 3a884e6
1 parent 0281342
commit 3a884e6
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/...tor/src/main/java/io/micronaut/sourcegen/generator/visitors/BuilderAnnotationVisitor.java b/...tor/src/main/java/io/micronaut/sourcegen/generator/visitors/BuilderAnnotationVisitor.java
@@ -38,6 +38,8 @@
 import io.micronaut.sourcegen.model.TypeDef;
 import io.micronaut.sourcegen.model.VariableDef;
 
+import java.util.HashSet;
+import java.util.Set;
 import javax.lang.model.element.Modifier;
 import java.io.IOException;
 import java.util.ArrayList;
@@ -52,13 +54,23 @@
 @Internal
 public final class BuilderAnnotationVisitor implements TypeElementVisitor<Builder, Object> {
 
+    private final Set<String> processed = new HashSet<>();
+
     @Override
     public @NonNull VisitorKind getVisitorKind() {
         return VisitorKind.ISOLATING;
     }
 
+    @Override
+    public void start(VisitorContext visitorContext) {
+        processed.clear();
+    }
+
     @Override
     public void visitClass(ClassElement element, VisitorContext context) {
+        if (processed.contains(element.getName())) {
+            return;
+        }
         String simpleName = element.getSimpleName() + "Builder";
         String builderClassName = element.getPackageName() + "." + simpleName;
 
@@ -125,6 +137,7 @@ public void visitClass(ClassElement element, VisitorContext context) {
         }
 
         ClassDef builderDef = builder.build();
+        processed.add(element.getName());
         context.visitGeneratedSourceFile(
             builderDef.getPackageName(),
             builderDef.getSimpleName(),

diff --git a/...ator/src/main/java/io/micronaut/sourcegen/generator/visitors/WitherAnnotationVisitor.java b/...ator/src/main/java/io/micronaut/sourcegen/generator/visitors/WitherAnnotationVisitor.java
@@ -37,6 +37,8 @@
 import io.micronaut.sourcegen.model.TypeDef;
 import io.micronaut.sourcegen.model.VariableDef;
 
+import java.util.HashSet;
+import java.util.Set;
 import javax.lang.model.element.Modifier;
 import java.io.IOException;
 import java.util.ArrayList;
@@ -54,13 +56,23 @@
 @Internal
 public final class WitherAnnotationVisitor implements TypeElementVisitor<Wither, Object> {
 
+    private final Set<String> processed = new HashSet<>();
+
     @Override
     public @NonNull VisitorKind getVisitorKind() {
         return VisitorKind.ISOLATING;
     }
 
+    @Override
+    public void start(VisitorContext visitorContext) {
+        processed.clear();
+    }
+
     @Override
     public void visitClass(ClassElement recordElement, VisitorContext context) {
+        if (processed.contains(recordElement.getName())) {
+            return;
+        }
         if (!recordElement.isRecord()) {
             throw new ProcessingException(recordElement, "Only records can be annotated with @Wither");
         }
@@ -154,6 +166,7 @@ public void visitClass(ClassElement recordElement, VisitorContext context) {
         }
 
         InterfaceDef witherDef = wither.build();
+        processed.add(recordElement.getName());
         context.visitGeneratedSourceFile(
             witherDef.getPackageName(),
             witherDef.getSimpleName(),