Generate config metadata fix. (helidon-io#7145)

trentjeff · Jul 4, 2023 · a61a942 · a61a942
1 parent 47e23a3
commit a61a942
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/...a-processor/src/main/java/io/helidon/config/metadata/processor/ConfigMetadataHandler.java b/...a-processor/src/main/java/io/helidon/config/metadata/processor/ConfigMetadataHandler.java
@@ -20,6 +20,7 @@
 import java.io.PrintWriter;
 import java.util.ArrayList;
 import java.util.HashMap;
+import java.util.LinkedHashSet;
 import java.util.LinkedList;
 import java.util.List;
 import java.util.Map;
@@ -75,7 +76,7 @@ class ConfigMetadataHandler {
     private final Map<String, ConfiguredType> newOptions = new HashMap<>();
     // map of module name to list of classes that belong to it
     private final Map<String, List<String>> moduleTypes = new HashMap<>();
-
+    private final Set<Element> classesToHandle = new LinkedHashSet<>();
     /*
      * Compiler utilities for annotation processing
      */
@@ -249,9 +250,10 @@ private static String javadoc(String docComment) {
     }
 
     private boolean doProcess(RoundEnvironment roundEnv) {
+        // we need to collect all types for processing
+        classesToHandle.addAll(roundEnv.getElementsAnnotatedWith(configuredElement));
         if (roundEnv.processingOver()) {
-            Set<? extends Element> classes = roundEnv.getElementsAnnotatedWith(configuredElement);
-            for (Element aClass : classes) {
+            for (Element aClass : classesToHandle) {
                 processClass(aClass);
             }