Add LoadTimeWeaverAware and test

Author: Lim-Soohyun

grpc-client-spring-boot-autoconfigure/src/main/java/net/devh/boot/grpc/client/autoconfigure/GrpcClientAutoConfiguration.java

@@ -20,6 +20,7 @@
 import java.util.Collections;
 import java.util.List;
 
+import javax.annotation.PostConstruct;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.autoconfigure.AutoConfigureAfter;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
@@ -52,6 +53,8 @@
 import net.devh.boot.grpc.client.stubfactory.BlockingStubFactory;
 import net.devh.boot.grpc.client.stubfactory.FutureStubFactory;
 import net.devh.boot.grpc.common.autoconfigure.GrpcCommonCodecAutoConfiguration;
+import org.springframework.context.weaving.LoadTimeWeaverAware;
+import org.springframework.instrument.classloading.LoadTimeWeaver;
 
 /**
  * The auto configuration used by Spring-Boot that contains all beans to create and inject grpc clients into beans.
@@ -194,4 +197,18 @@ GrpcChannelFactory inProcessGrpcChannelFactory(
         return new InProcessChannelFactory(properties, globalClientInterceptorRegistry, channelConfigurers);
     }
 
+    @Configuration(proxyBeanMethods = false)
+    static class GrpcClientConstructorInjectionConfiguration implements LoadTimeWeaverAware {
+        @Autowired
+        private GrpcClientBeanPostProcessor grpcClientBeanPostProcessor;
+
+        @PostConstruct
+        public void init() {
+            grpcClientBeanPostProcessor.initGrpClientConstructorInjections();
+        }
+
+        @Override
+        public void setLoadTimeWeaver(LoadTimeWeaver ltw) {
+        }
+    }
 }

grpc-client-spring-boot-autoconfigure/src/main/java/net/devh/boot/grpc/client/inject/GrpcClientBeanPostProcessor.java

@@ -34,7 +34,6 @@
 import java.util.Map;
 import java.util.Set;
 import java.util.concurrent.ConcurrentHashMap;
-import javax.annotation.PostConstruct;
 
 import org.springframework.beans.BeanInstantiationException;
 import org.springframework.beans.BeanUtils;
@@ -108,15 +107,10 @@ public GrpcClientBeanPostProcessor(final ApplicationContext applicationContext)
         this.grpcClientAnnotationTypes.add(GrpcClient.class);
     }
 
-    @PostConstruct
-    public void init() {
-        initGrpClientConstructorInjections();
-    }
-
     /**
      * Triggers registering grpc client beans from GrpcClientConstructorInjection.
      */
-    private void initGrpClientConstructorInjections() {
+    public void initGrpClientConstructorInjections() {
         Iterable<GrpcClientConstructorInjection.Registry> registries;
         try {
             registries = getConfigurableBeanFactory().getBean(GrpcClientConstructorInjection.class).getRegistries();

tests/src/test/java/net/devh/boot/grpc/test/inject/GrpcClientMetricsTest.java

@@ -0,0 +1,91 @@
+package net.devh.boot.grpc.test.inject;
+
+import static org.assertj.core.api.AssertionsForInterfaceTypes.assertThat;
+
+import io.grpc.stub.AbstractStub;
+import io.micrometer.core.instrument.MeterRegistry;
+import net.devh.boot.grpc.client.inject.GrpcClient;
+import net.devh.boot.grpc.client.stubfactory.StandardJavaGrpcStubFactory;
+import net.devh.boot.grpc.client.stubfactory.StubFactory;
+import net.devh.boot.grpc.test.config.BaseAutoConfiguration;
+import net.devh.boot.grpc.test.config.InProcessConfiguration;
+import net.devh.boot.grpc.test.config.MetricConfiguration;
+import net.devh.boot.grpc.test.config.ServiceConfiguration;
+import net.devh.boot.grpc.test.proto.TestServiceGrpc;
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.actuate.autoconfigure.metrics.JvmMetricsAutoConfiguration;
+import org.springframework.boot.actuate.autoconfigure.metrics.MetricsAutoConfiguration;
+import org.springframework.boot.autoconfigure.ImportAutoConfiguration;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.stereotype.Component;
+import org.springframework.test.annotation.DirtiesContext;
+import org.springframework.test.context.junit.jupiter.SpringJUnitConfig;
+
+@SpringBootTest
+@SpringJUnitConfig(
+    classes = {
+        GrpcClientMetricsTest.TestConfig.class,
+        GrpcClientMetricsTest.GrpcClientConstructorInjectionBean.class,
+        InProcessConfiguration.class,
+        ServiceConfiguration.class,
+        BaseAutoConfiguration.class,
+        MetricConfiguration.class
+    })
+@DirtiesContext
+public class GrpcClientMetricsTest {
+    @Autowired
+    MeterRegistry registry;
+
+    @Test
+    void jvmMetricsTest() {
+        assertThat(registry.getMeters())
+            .filteredOn(meter -> meter.getId().getName().contains("jvm"))
+            .isNotEmpty();
+    }
+
+    @Component
+    public static class GrpcClientConstructorInjectionBean {
+        public TestServiceGrpc.TestServiceBlockingStub blockingStub;
+        public TestServiceGrpc.TestServiceFutureStub futureStubForClientTest;
+        public TestServiceGrpc.TestServiceBlockingStub anotherBlockingStub;
+        public TestServiceGrpc.TestServiceBlockingStub unnamedTestServiceBlockingStub;
+        public CustomGrpc.FactoryMethodAccessibleStub anotherServiceClientBean;
+
+        public GrpcClientConstructorInjectionBean(
+            @GrpcClient("test") TestServiceGrpc.TestServiceBlockingStub blockingStub,
+            @GrpcClient("test") TestServiceGrpc.TestServiceFutureStub futureStubForClientTest,
+            @GrpcClient("anotherTest") TestServiceGrpc.TestServiceBlockingStub anotherBlockingStub,
+            @GrpcClient("unnamed") TestServiceGrpc.TestServiceBlockingStub unnamedTestServiceBlockingStub,
+            @GrpcClient("test") CustomGrpc.FactoryMethodAccessibleStub anotherServiceClientBean) {
+          this.blockingStub = blockingStub;
+          this.futureStubForClientTest = futureStubForClientTest;
+          this.anotherBlockingStub = anotherBlockingStub;
+          this.unnamedTestServiceBlockingStub = unnamedTestServiceBlockingStub;
+          this.anotherServiceClientBean = anotherServiceClientBean;
+        }
+    }
+
+    @Configuration
+    @ImportAutoConfiguration({MetricsAutoConfiguration.class, JvmMetricsAutoConfiguration.class})
+    public static class TestConfig {
+        @Bean
+        StubFactory customStubFactory() {
+            return new StandardJavaGrpcStubFactory() {
+
+                @Override
+                public boolean isApplicable(final Class<? extends AbstractStub<?>> stubType) {
+          return CustomStub.class.isAssignableFrom(stubType);
+        }
+
+                @Override
+                protected String getFactoryMethodName() {
+          return "custom";
+        }
+
+            };
+        }
+    }
+}