Use the new writer

Author: Soroosh Sarabadani

operator-framework/src/main/java/io/javaoperatorsdk/operator/processing/ControllerAnnotationProcessor.java renamed to operator-framework/src/main/java/io/javaoperatorsdk/operator/processing/annotation/ControllerAnnotationProcessor.java

@@ -1,21 +1,19 @@
-package io.javaoperatorsdk.operator.processing;
+package io.javaoperatorsdk.operator.processing.annotation;
 
 import com.google.auto.service.AutoService;
 import com.squareup.javapoet.*;
 import io.fabric8.kubernetes.api.builder.Function;
 import io.fabric8.kubernetes.client.CustomResourceDoneable;
 import io.javaoperatorsdk.operator.api.ResourceController;
+
 import javax.annotation.processing.*;
 import javax.lang.model.SourceVersion;
 import javax.lang.model.element.*;
 import javax.lang.model.type.DeclaredType;
 import javax.lang.model.type.TypeKind;
 import javax.lang.model.type.TypeMirror;
 import javax.tools.Diagnostic;
-import javax.tools.FileObject;
 import javax.tools.JavaFileObject;
-import javax.tools.StandardLocation;
-import java.io.IOException;
 import java.io.PrintWriter;
 import java.util.ArrayList;
 import java.util.HashSet;
@@ -30,23 +28,14 @@
 @SupportedSourceVersion(SourceVersion.RELEASE_8)
 @AutoService(Processor.class)
 public class ControllerAnnotationProcessor extends AbstractProcessor {
-    private FileObject resource;
-    PrintWriter printWriter = null;
+    private ControllersResourceWriter controllersResourceWriter;
     private Set<String> generatedDoneableClassFiles = new HashSet<>();
 
     @Override
     public synchronized void init(ProcessingEnvironment processingEnv) {
         super.init(processingEnv);
-        try {
-            resource = processingEnv.getFiler().createResource(StandardLocation.CLASS_OUTPUT, "", CONTROLLERS_RESOURCE_PATH);
-        } catch (IOException e) {
-            throw new RuntimeException(e);
-        }
-        try {
-            printWriter = new PrintWriter(resource.openOutputStream());
-        } catch (IOException e) {
-            throw new RuntimeException(e);
-        }
+        controllersResourceWriter = new ControllersResourceWriter(processingEnv);
+        controllersResourceWriter.loadExistingMappings();
     }
 
     @Override
@@ -57,15 +46,17 @@ public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment
                         = roundEnv.getElementsAnnotatedWith(annotation);
                 annotatedElements.stream().filter(element -> element.getKind().equals(ElementKind.CLASS))
                         .map(e -> (TypeElement) e)
-                        .forEach(e -> this.generateDoneableClass(e, printWriter));
+                        .forEach(e -> this.generateDoneableClass(e));
             }
         } finally {
-            printWriter.close();
+            if (roundEnv.processingOver()) {
+                controllersResourceWriter.flush();
+            }
         }
         return true;
     }
 
-    private void generateDoneableClass(TypeElement controllerClassSymbol, PrintWriter printWriter) {
+    private void generateDoneableClass(TypeElement controllerClassSymbol) {
         try {
             final TypeMirror resourceType = findResourceType(controllerClassSymbol);
 
@@ -84,14 +75,14 @@ private void generateDoneableClass(TypeElement controllerClassSymbol, PrintWrite
                                 destinationClassFileName,
                                 CONTROLLERS_RESOURCE_PATH)
                 );
-                printWriter.println(controllerClassSymbol.getQualifiedName() + "," + customResourceType.toString());
+                controllersResourceWriter.add(controllerClassSymbol.getQualifiedName().toString(), customResourceType.toString());
                 return;
             }
             JavaFileObject builderFile = processingEnv.getFiler()
                     .createSourceFile(destinationClassFileName);
 
             try (PrintWriter out = new PrintWriter(builderFile.openWriter())) {
-                printWriter.println(controllerClassSymbol.getQualifiedName() + "," + customResourceType.toString());
+                controllersResourceWriter.add(controllerClassSymbol.getQualifiedName().toString(), customResourceType.toString());
                 final MethodSpec constructor = MethodSpec.constructorBuilder()
                         .addModifiers(Modifier.PUBLIC)
                         .addParameter(customResourceType, "resource")
@@ -125,7 +116,6 @@ private TypeMirror findResourceType(TypeElement controllerClassSymbol) throws Ex
                     )
                     .findFirst()
                     .orElseThrow(() -> new Exception("ResourceController is not implemented by " + controllerClassSymbol.toString()));
-
             return controllerType.getTypeArguments().get(0);
         } catch (Exception e) {
             e.printStackTrace();