add an async bean generation

Author: SentryMan

.gitignore

@@ -13,3 +13,4 @@ build/
 inject-generator/avaje-inject
 inject-generator/avaje-inject-generator
 inject-generator/avaje-processors.txt
+*.csv

inject-generator/src/main/java/io/avaje/inject/generator/BeanReader.java

@@ -7,6 +7,7 @@
 import java.util.List;
 import java.util.Optional;
 import java.util.Set;
+import java.util.concurrent.CompletableFuture;
 import java.util.stream.Stream;
 
 import javax.lang.model.element.Element;
@@ -38,6 +39,7 @@ final class BeanReader {
   private final boolean prototype;
   private final boolean primary;
   private final boolean secondary;
+  private final boolean async;
   private final boolean lazy;
   private final boolean proxy;
   private final BeanAspects aspects;
@@ -58,6 +60,7 @@ final class BeanReader {
     this.primary = PrimaryPrism.isPresent(beanType);
     this.secondary = !primary && SecondaryPrism.isPresent(beanType);
     this.lazy = !FactoryPrism.isPresent(beanType) && LazyPrism.isPresent(beanType);
+    this.async = !FactoryPrism.isPresent(beanType) && AsyncBeanPrism.isPresent(beanType);
     final var beantypes = BeanTypesPrism.getOptionalOn(beanType);
     beantypes.ifPresent(p -> Util.validateBeanTypes(beanType, p.value()));
     this.typeReader =
@@ -128,13 +131,17 @@ BeanAspects aspects() {
   }
 
   boolean registerProvider() {
-    return prototype || lazy;
+    return prototype || async || lazy;
   }
 
   boolean lazy() {
     return lazy;
   }
 
+  boolean async() {
+    return async;
+  }
+
   boolean importedComponent() {
     return importedComponent;
   }
@@ -304,7 +311,7 @@ void buildAddFor(Append writer) {
   }
 
   void buildRegister(Append writer) {
-    if (prototype || lazy) {
+    if (prototype || lazy || async) {
       return;
     }
     writer.indent("      ");
@@ -349,15 +356,20 @@ void prototypePostConstruct(Append writer, String indent) {
 
   private void lifeCycleNotSupported(String lifecycle) {
     if (registerProvider()) {
-      logError(
-        beanType,
-        "%s scoped bean does not support the %s lifecycle method",
-        prototype ? "@Prototype" : "@Lazy",
-        lifecycle);
+      String scope;
+      if (lazy) scope = "@Lazy";
+      if (async) scope = "@AsyncBean";
+      else scope = "@Prototype";
+
+      logError(beanType, "%s scoped bean does not support the %s lifecycle method", scope, lifecycle);
     }
   }
 
   private Set<String> importTypes() {
+    importTypes.add(type);
+    if (async) {
+      importTypes.add(CompletableFuture.class.getCanonicalName());
+    }
     importTypes.add(type);
     typeReader.extraImports(importTypes);
     requestParams.addImports(importTypes);

inject-generator/src/main/java/io/avaje/inject/generator/SimpleBeanWriter.java

@@ -14,6 +14,7 @@
 import java.util.Map;
 import java.util.Objects;
 import java.util.Set;
+import java.util.concurrent.CompletableFuture;
 
 import javax.lang.model.type.TypeKind;
 import javax.tools.JavaFileObject;
@@ -175,7 +176,17 @@ private void writeAddFor(MethodReader constructor) {
     beanReader.buildAddFor(writer);
     if (beanReader.registerProvider()) {
       indent += "  ";
-      writer.append("      builder.%s(() -> {", beanReader.lazy() ? "registerProvider" : "asPrototype().registerProvider").eol();
+
+      final String registerProvider;
+      if (beanReader.async()) {
+        registerProvider = "registerProvider(CompletableFuture.supplyAsync";
+      } else if (beanReader.lazy()) {
+        registerProvider = "registerProvider";
+      } else {
+        registerProvider = "asPrototype().registerProvider";
+      }
+
+      writer.append("      builder.%s(() -> {", registerProvider).eol();
     }
     constructor.startTry(writer);
     writeCreateBean(constructor);
@@ -187,7 +198,7 @@ private void writeAddFor(MethodReader constructor) {
     if (beanReader.registerProvider()) {
       beanReader.prototypePostConstruct(writer, indent);
       writer.indent("        return bean;").eol();
-      writer.indent("      });").eol();
+      writer.indent("      })").append("%s;", beanReader.async() ? "::join)" : "").eol();
     }
     writeObserveMethods();
     constructor.endTry(writer);

inject-generator/src/main/java/io/avaje/inject/generator/package-info.java

@@ -3,6 +3,7 @@
 @GeneratePrism(value = Aspect.Import.class, name = "AspectImportPrism")
 @GeneratePrism(Assisted.class)
 @GeneratePrism(AssistFactory.class)
+@GeneratePrism(AsyncBean.class)
 @GeneratePrism(Bean.class)
 @GeneratePrism(Component.class)
 @GeneratePrism(Component.Import.class)

inject-generator/src/test/java/io/avaje/inject/generator/InjectProcessorTest.java

@@ -41,7 +41,7 @@ void deleteGeneratedFiles() {
     }
   }
 
-  @Disabled
+
   @Test
   void testGeneration() throws Exception {
     final String source =

inject-generator/src/test/java/io/avaje/inject/generator/models/valid/lazy/BackgroundBean.java

@@ -0,0 +1,12 @@
+package io.avaje.inject.generator.models.valid.lazy;
+
+import io.avaje.inject.AsyncBean;
+import jakarta.inject.Inject;
+import jakarta.inject.Provider;
+import jakarta.inject.Singleton;
+
+@Singleton
+@AsyncBean
+public class BackgroundBean {
+  @Inject Provider<Integer> intProvider;
+}

inject/src/main/java/io/avaje/inject/AsyncBean.java

@@ -0,0 +1,8 @@
+package io.avaje.inject;
+
+import java.lang.annotation.*;
+
+/** Limits the types exposed by this bean to the given types. */
+@Retention(RetentionPolicy.SOURCE)
+@Target({ElementType.METHOD, ElementType.TYPE})
+public @interface AsyncBean {}