Using a compiler argument to toggle incremental annotation processing support.

imeszaros · imeszaros · commit c954eb4b4cec · 2020-07-19T19:58:42.000+02:00
diff --git a/modules/jooby-apt/src/main/java/io/jooby/apt/JoobyProcessor.java b/modules/jooby-apt/src/main/java/io/jooby/apt/JoobyProcessor.java
@@ -5,15 +5,13 @@
  */
 package io.jooby.apt;
 
+import io.jooby.Jooby;
 import io.jooby.MvcFactory;
 import io.jooby.SneakyThrows;
 import io.jooby.internal.apt.HandlerCompiler;
 import io.jooby.internal.apt.ModuleCompiler;
 
-import javax.annotation.processing.AbstractProcessor;
-import javax.annotation.processing.Filer;
-import javax.annotation.processing.ProcessingEnvironment;
-import javax.annotation.processing.RoundEnvironment;
+import javax.annotation.processing.*;
 import javax.lang.model.SourceVersion;
 import javax.lang.model.element.AnnotationMirror;
 import javax.lang.model.element.Element;
@@ -29,13 +27,7 @@
 import java.io.IOException;
 import java.io.OutputStream;
 import java.io.PrintWriter;
-import java.util.ArrayList;
-import java.util.Collections;
-import java.util.LinkedHashMap;
-import java.util.LinkedHashSet;
-import java.util.List;
-import java.util.Map;
-import java.util.Set;
+import java.util.*;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
 
@@ -44,8 +36,14 @@
  *
  * @since 2.1.0
  */
+@SupportedOptions({
+    JoobyProcessor.OPT_DEBUG,
+    JoobyProcessor.OPT_INCREMENTAL })
 public class JoobyProcessor extends AbstractProcessor {
 
+  protected static final String OPT_DEBUG = "jooby.debug";
+  protected static final String OPT_INCREMENTAL = "jooby.incremental";
+
   private ProcessingEnvironment processingEnv;
 
   /**
@@ -57,9 +55,21 @@ public class JoobyProcessor extends AbstractProcessor {
   private Map<TypeElement, Map<TypeElement, List<ExecutableElement>>> routeMap = new LinkedHashMap<>();
 
   private boolean debug;
+  private boolean incremental;
 
   private int round;
 
+  @Override public Set<String> getSupportedOptions() {
+    Set<String> options = new HashSet<>(super.getSupportedOptions());
+
+    if (incremental) {
+      // enables incremental annotation processing support in Gradle
+      options.add("org.gradle.annotation.processing.isolating");
+    }
+
+    return options;
+  }
+
   @Override public Set<String> getSupportedAnnotationTypes() {
     return Stream.concat(Annotations.PATH.stream(), Annotations.HTTP_METHODS.stream())
         .collect(Collectors.toCollection(LinkedHashSet::new));
@@ -71,7 +81,11 @@ public class JoobyProcessor extends AbstractProcessor {
 
   @Override public void init(ProcessingEnvironment processingEnvironment) {
     this.processingEnv = processingEnvironment;
-    debug = Boolean.parseBoolean(processingEnvironment.getOptions().getOrDefault("debug", "false"));
+
+    debug = boolOpt(processingEnv, OPT_DEBUG, false);
+    incremental = boolOpt(processingEnv, OPT_INCREMENTAL, false);
+
+    debug("Incremental annotation processing is turned %s.", incremental ? "ON" : "OFF");
   }
 
   @Override
@@ -183,7 +197,11 @@ private void build(Filer filer) throws Exception {
       moduleList.add(moduleClass);
     }
 
-    //doServices(filer, moduleList);
+    if (!incremental) {
+      // writing resource files would prevent incremental annotation processing in Gradle:
+      // https://docs.gradle.org/5.0/userguide/java_plugin.html#sec:incremental_annotation_processing
+      doServices(filer, moduleList);
+    }
   }
 
   private String signature(ExecutableElement method) {
@@ -291,4 +309,9 @@ private List<String> path(String method, List<? extends AnnotationMirror> annota
         .distinct()
         .collect(Collectors.toList());
   }
+
+  private boolean boolOpt(ProcessingEnvironment processingEnvironment, String option, boolean defaultValue) {
+    return Boolean.parseBoolean(processingEnvironment
+        .getOptions().getOrDefault(option, String.valueOf(defaultValue)));
+  }
 }
diff --git a/modules/jooby-apt/src/main/resources/META-INF/gradle/incremental.annotation.processors b/modules/jooby-apt/src/main/resources/META-INF/gradle/incremental.annotation.processors
@@ -1 +1 @@
-io.jooby.apt.JoobyProcessor,isolating
+io.jooby.apt.JoobyProcessor,dynamic

Original file line number	Diff line number	Diff line change
`@@ -1 +1 @@`
`1`		`-io.jooby.apt.JoobyProcessor,isolating`
	`1`	`+io.jooby.apt.JoobyProcessor,dynamic`