APT: write class file

Author: jknack

modules/jooby-apt/src/main/java/io/jooby/compiler/MvcHandlerCompiler.java

@@ -272,8 +272,8 @@ private void processReturnType(MethodVisitor visitor) throws Exception {
   }
 
   public String getGeneratedClass() {
-    return getController().getName() + "$" + httpMethod.toUpperCase() + "$" + executable
-        .getSimpleName();
+    return getController().getName() + "$" + httpMethod.toUpperCase() + pattern
+        .replace('/', '_');
   }
 
   public String getGeneratedInternalClass() {

modules/jooby-apt/src/main/java/io/jooby/compiler/MvcProcessor.java

@@ -13,6 +13,7 @@
 import org.objectweb.asm.util.TraceClassVisitor;
 
 import javax.annotation.processing.Completion;
+import javax.annotation.processing.Filer;
 import javax.annotation.processing.ProcessingEnvironment;
 import javax.annotation.processing.Processor;
 import javax.annotation.processing.RoundEnvironment;
@@ -23,7 +24,10 @@
 import javax.lang.model.element.ExecutableElement;
 import javax.lang.model.element.TypeElement;
 import javax.lang.model.type.TypeMirror;
+import javax.tools.JavaFileObject;
 import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.OutputStream;
 import java.io.PrintWriter;
 import java.util.Collections;
 import java.util.HashMap;
@@ -76,16 +80,23 @@ public boolean process(Set<? extends TypeElement> annotations,
         }
       }
     }
+    Filer filer = processingEnvironment.getFiler();
     Map<String, List<Map.Entry<String, MvcHandlerCompiler>>> classes = result.entrySet().stream()
         .collect(Collectors.groupingBy(e -> e.getValue().getController().getName()));
     for (Map.Entry<String, List<Map.Entry<String, MvcHandlerCompiler>>> entry : classes
         .entrySet()) {
       try {
         List<Map.Entry<String, MvcHandlerCompiler>> handlers = entry.getValue();
         MvcModuleCompiler module = new MvcModuleCompiler(entry.getKey());
-        modules.put(entry.getKey() + "$Module", module.compile(handlers));
+        String moduleClass = entry.getKey() + "$Module";
+        byte[] moduleBin = module.compile(handlers);
+        writeClass(filer.createClassFile(moduleClass), moduleBin);
+        modules.put(moduleClass, moduleBin);
         for (Map.Entry<String, MvcHandlerCompiler> handler : handlers) {
-          modules.put(handler.getValue().getGeneratedClass(), handler.getValue().compile());
+          String handleClass = handler.getValue().getGeneratedClass();
+          byte[] handleBin = handler.getValue().compile();
+          writeClass(filer.createClassFile(handleClass), handleBin);
+          modules.put(handleClass, handleBin);
         }
       } catch (Exception x) {
         x.printStackTrace();
@@ -95,6 +106,12 @@ public boolean process(Set<? extends TypeElement> annotations,
     return true;
   }
 
+  private void writeClass(JavaFileObject javaFileObject, byte[] bytecode) throws IOException {
+    try (OutputStream output = javaFileObject.openOutputStream()) {
+      output.write(bytecode);
+    }
+  }
+
   /*package*/ MvcHandlerCompiler compilerFor(String methodDescriptor) {
     return result.get(methodDescriptor);
   }