Merge pull request #754 from avaje/feature/destoryMethodChaining

#749 Add support for chained methods in @bean(destroyMethod)

Author: SentryMan

blackbox-test-inject/src/main/java/org/example/myapp/MyNestedDestroy.java

@@ -0,0 +1,29 @@
+package org.example.myapp;
+
+import java.util.concurrent.atomic.AtomicInteger;
+
+public class MyNestedDestroy {
+
+  public static AtomicInteger started = new AtomicInteger();
+  public static AtomicInteger stopped = new AtomicInteger();
+
+  public static void reset() {
+    started.set(0);
+    stopped.set(0);
+  }
+
+  public void start() {
+    started.incrementAndGet();
+  }
+
+  public Reaper reaper() {
+    return new Reaper();
+  }
+
+  public static class Reaper {
+
+    public void stop() {
+      stopped.incrementAndGet();
+    }
+  }
+}

blackbox-test-inject/src/main/java/org/example/myapp/config/AFactory.java

@@ -2,12 +2,18 @@
 
 import io.avaje.inject.Bean;
 import io.avaje.inject.Factory;
+import org.example.myapp.MyNestedDestroy;
 
 import java.io.IOException;
 
 @Factory
 class AFactory {
 
+  @Bean(initMethod = "start", destroyMethod = "reaper().stop()")
+  MyNestedDestroy lifecycle2() {
+    return new MyNestedDestroy();
+  }
+
   @Bean
   A0.Builder build0() {
     return new I0();

blackbox-test-inject/src/test/java/org/example/myapp/HelloServiceTest.java

@@ -14,6 +14,17 @@
 
 class HelloServiceTest {
 
+  @Test
+  void lifecycles() {
+    MyNestedDestroy.reset();
+    try (BeanScope beanScope = BeanScope.builder().build()) {
+      assertThat(beanScope.get(MyNestedDestroy.class)).isNotNull();
+      assertThat(MyNestedDestroy.started.get()).isEqualTo(1);
+      assertThat(MyNestedDestroy.stopped.get()).isEqualTo(0);
+    }
+    assertThat(MyNestedDestroy.stopped.get()).isEqualTo(1);
+  }
+
   /**
    * No mocking, no use of <code>@TestScope</code> so just like main.
    */

blackbox-test-inject/src/test/java/org/example/myapp/testconfig/CountTestScopeStart.java

@@ -13,6 +13,5 @@ public void onStart() {
 
   public void onStop() {
     stopped.incrementAndGet();
-    System.out.println("STOPPED !! ");
   }
 }

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

@@ -321,7 +321,6 @@ void buildRegister(Append writer) {
   }
 
   void addLifecycleCallbacks(Append writer, String indent) {
-
     if (postConstructMethod != null && !registerProvider()) {
       writer.indent(indent).append(" builder.addPostConstruct($bean::%s);", postConstructMethod.getSimpleName()).eol();
     }

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

@@ -287,7 +287,7 @@ void builderBuildAddBean(Append writer) {
       if (hasInitMethod) {
         var addPostConstruct =
           multiRegister
-            ? "    .peek(b -> builder.addPostConstruct(b::%s))"
+            ? "    .peek($bean -> builder.addPostConstruct($bean::%s))"
             : "builder.addPostConstruct($bean::%s);";
         writer.indent(indent).append(addPostConstruct, initMethod).eol();
       }
@@ -296,14 +296,14 @@ void builderBuildAddBean(Append writer) {
       if (notEmpty(destroyMethod)) {
         var addPreDestroy =
           multiRegister
-            ? "    .forEach(b -> builder.addPreDestroy(b::%s%s));"
-            : "builder.addPreDestroy($bean::%s%s);";
-        writer.indent(indent).append(addPreDestroy, destroyMethod, priority).eol();
+            ? "    .forEach($bean -> builder.addPreDestroy(%s%s));"
+            : "builder.addPreDestroy(%s%s);";
+        writer.indent(indent).append(addPreDestroy, addPreDestroy(destroyMethod), priority).eol();
 
       } else if (typeReader != null && typeReader.isClosable()) {
         var addPreDestroy =
           multiRegister
-            ? "    .forEach(b -> builder.addPreDestroy(b::close%s));"
+            ? "    .forEach($bean -> builder.addPreDestroy($bean::close%s));"
             : "builder.addPreDestroy($bean::close%s);";
         writer.indent(indent).append(addPreDestroy, priority).eol();
 
@@ -324,6 +324,13 @@ void builderBuildAddBean(Append writer) {
     }
   }
 
+  static String addPreDestroy(String destroyMethod) {
+    if (!destroyMethod.contains(".")) {
+      return "$bean::" + destroyMethod;
+    }
+    return "() -> $bean." + destroyMethod;
+  }
+
   private boolean hasLifecycleMethods() {
     return notEmpty(initMethod) || notEmpty(destroyMethod) || (typeReader != null && typeReader.isClosable() || beanCloseable);
   }

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

@@ -0,0 +1,19 @@
+package io.avaje.inject.generator;
+
+import org.junit.jupiter.api.Test;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+class MethodReaderTest {
+
+  @Test
+  void addPreDestroy() {
+    assertThat(MethodReader.addPreDestroy("close")).isEqualTo("$bean::close");
+    assertThat(MethodReader.addPreDestroy("foo")).isEqualTo("$bean::foo");
+  }
+
+  @Test
+  void addPreDestroyNested() {
+    assertThat(MethodReader.addPreDestroy("foo().bar()")).isEqualTo("() -> $bean.foo().bar()");
+  }
+}