#158: Fixed CompletableFuture usages not writing JsonType variable values if no other endpoints are defined with the same return type.

Author: Haydon Perrin

http-generator-core/src/main/java/io/avaje/http/generator/core/JsonBUtil.java

@@ -33,7 +33,13 @@ public static Map<String, UType> jsonTypes(ControllerReader reader) {
             methodReader -> {
               addJsonBodyType(methodReader, addToMap);
               if (!methodReader.isVoid()) {
-                addToMap.accept(UType.parse(methodReader.returnType()));
+                UType uType = UType.parse(methodReader.returnType());
+
+                if (uType.mainType().equals("java.util.concurrent.CompletableFuture")) {
+                  uType = uType.paramRaw();
+                }
+
+                addToMap.accept(uType);
               }
             });
 
@@ -50,12 +56,6 @@ private static void addJsonBodyType(MethodReader methodReader, Consumer<UType> a
   }
 
   public static void writeJsonbType(UType type, Append writer) {
-    // Support for CompletableFuture's.
-    if (type.mainType().equals("java.util.concurrent.CompletableFuture")) {
-      writeJsonbType(type.paramRaw(), writer);
-      return;
-    }
-
     writer.append("    this.%sJsonType = jsonB.type(", type.shortName());
     if (!type.isGeneric()) {
       writer.append("%s.class)", Util.shortName(PrimitiveUtil.wrap(type.full())));

http-generator-javalin/src/main/java/io/avaje/http/generator/javalin/ControllerWriter.java

@@ -88,17 +88,6 @@ private void writeClassStart() {
     }
 
     for (UType type : jsonTypes.values()) {
-      // Support for CompletableFuture's.
-      if (type.mainType().equals("java.util.concurrent.CompletableFuture")) {
-        type = type.paramRaw();
-
-        if (this.jsonTypes.containsKey(type.full())) {
-          // Already written before -- we can skip.
-          continue;
-        }
-      }
-
-      // Everything else
       final var typeString = PrimitiveUtil.wrap(type.shortType()).replace(",", ", ");
       writer.append("  private final JsonType<%s> %sJsonType;", typeString, type.shortName()).eol();
     }
@@ -117,13 +106,7 @@ private void writeClassStart() {
       writer.append("    this.validator = validator;").eol();
     }
     if (useJsonB) {
-      for (final UType type : jsonTypes.values()) {
-        // Skip trying to assign a global variable value for any UType that is a Completable Future. Because the paramRaw() should
-        // already be in this jsonTypes map anyway and write the assignment all by itself.
-        if (type.mainType().equals("java.util.concurrent.CompletableFuture")) {
-          continue;
-        }
-
+      for (UType type : jsonTypes.values()) {
         JsonBUtil.writeJsonbType(type, writer);
       }
     }