Support 3rd party gRPC Java compilers via StubFactory

Author: liqweed

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

@@ -46,6 +46,9 @@
 import net.devh.boot.grpc.client.interceptor.AnnotationGlobalClientInterceptorConfigurer;
 import net.devh.boot.grpc.client.interceptor.GlobalClientInterceptorRegistry;
 import net.devh.boot.grpc.client.nameresolver.NameResolverRegistration;
+import net.devh.boot.grpc.client.stubfactory.AsyncStubFactory;
+import net.devh.boot.grpc.client.stubfactory.BlockingStubFactory;
+import net.devh.boot.grpc.client.stubfactory.FutureStubFactory;
 import net.devh.boot.grpc.common.autoconfigure.GrpcCommonCodecAutoConfiguration;
 
 /**
@@ -65,6 +68,21 @@ static GrpcClientBeanPostProcessor grpcClientBeanPostProcessor(final Application
         return new GrpcClientBeanPostProcessor(applicationContext);
     }
 
+    @Bean
+    AsyncStubFactory asyncStubFactory() {
+        return new AsyncStubFactory();
+    }
+
+    @Bean
+    BlockingStubFactory blockingStubFactory() {
+        return new BlockingStubFactory();
+    }
+
+    @Bean
+    FutureStubFactory futureStubFactory() {
+        return new FutureStubFactory();
+    }
+
     @ConditionalOnMissingBean
     @Bean
     GrpcChannelsProperties grpcChannelsProperties() {

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

@@ -19,15 +19,11 @@
 
 import static java.util.Objects.requireNonNull;
 
-import java.lang.reflect.Constructor;
 import java.lang.reflect.Field;
 import java.lang.reflect.Member;
 import java.lang.reflect.Method;
-import java.util.ArrayList;
-import java.util.Collection;
-import java.util.List;
+import java.util.*;
 
-import org.springframework.beans.BeanInstantiationException;
 import org.springframework.beans.BeansException;
 import org.springframework.beans.InvalidPropertyException;
 import org.springframework.beans.factory.BeanCreationException;
@@ -41,12 +37,10 @@
 
 import io.grpc.Channel;
 import io.grpc.ClientInterceptor;
-import io.grpc.stub.AbstractAsyncStub;
-import io.grpc.stub.AbstractBlockingStub;
-import io.grpc.stub.AbstractFutureStub;
 import io.grpc.stub.AbstractStub;
 import net.devh.boot.grpc.client.channelfactory.GrpcChannelFactory;
 import net.devh.boot.grpc.client.nameresolver.NameResolverRegistration;
+import net.devh.boot.grpc.client.stubfactory.StubFactory;
 
 /**
  * This {@link BeanPostProcessor} searches for fields and methods in beans that are annotated with {@link GrpcClient}
@@ -63,6 +57,7 @@ public class GrpcClientBeanPostProcessor implements BeanPostProcessor {
     // which could lead to problems with the correct bean setup.
     private GrpcChannelFactory channelFactory = null;
     private List<StubTransformer> stubTransformers = null;
+    private final List<StubFactory> stubFactories;
 
     /**
      * Creates a new GrpcClientBeanPostProcessor with the given ApplicationContext.
@@ -72,6 +67,7 @@ public class GrpcClientBeanPostProcessor implements BeanPostProcessor {
      */
     public GrpcClientBeanPostProcessor(final ApplicationContext applicationContext) {
         this.applicationContext = requireNonNull(applicationContext, "applicationContext");
+        stubFactories = new ArrayList<>(applicationContext.getBeansOfType(StubFactory.class).values());
     }
 
     @Override
@@ -217,7 +213,8 @@ protected <T> T valueForMember(final String name, final Member injectionTarget,
         } else if (AbstractStub.class.isAssignableFrom(injectionType)) {
 
             @SuppressWarnings("unchecked") // Eclipse incorrectly marks this as not required
-            AbstractStub<?> stub = createStub(injectionType.asSubclass(AbstractStub.class), channel);
+            AbstractStub<?> stub = createStub(
+                    (Class<? extends AbstractStub<?>>) injectionType.asSubclass(AbstractStub.class), channel);
             for (final StubTransformer stubTransformer : getStubTransformers()) {
                 stub = stubTransformer.transform(name, stub);
             }
@@ -229,53 +226,25 @@ protected <T> T valueForMember(final String name, final Member injectionTarget,
     }
 
     /**
-     * Creates a stub of the given type.
-     *
-     * @param <T> The type of the instance to be injected.
-     * @param stubType The type of the stub to create.
-     * @param channel The channel used to create the stub.
-     * @return The newly created stub.
-     *
-     * @throws BeanInstantiationException If the stub couldn't be created.
-     */
-    protected <T extends AbstractStub<T>> T createStub(final Class<T> stubType, final Channel channel) {
-        try {
-            // First try the public static factory method
-            final String methodName = deriveStubFactoryMethodName(stubType);
-            final Class<?> enclosingClass = stubType.getEnclosingClass();
-            final Method factoryMethod = enclosingClass.getMethod(methodName, Channel.class);
-            return stubType.cast(factoryMethod.invoke(null, channel));
-        } catch (final Exception e) {
-            try {
-                // Use the private constructor as backup
-                final Constructor<T> constructor = stubType.getDeclaredConstructor(Channel.class);
-                constructor.setAccessible(true);
-                return constructor.newInstance(channel);
-            } catch (final Exception e1) {
-                e.addSuppressed(e1);
-            }
-            throw new BeanInstantiationException(stubType, "Failed to create gRPC client", e);
-        }
-    }
-
-    /**
-     * Derives the name of the factory method from the given stub type.
-     *
-     * @param stubType The type of the stub to get it for.
-     * @return The name of the factory method.
+     * Creates a stub instance for the specified stub type.
+     * 
+     * @param stubClass
+     * @param channel
      * @throws IllegalArgumentException If the method was called with an unsupported stub type.
+     * @throws IllegalStateException If failed creating the stub instance.
+     * @return
      */
-    protected String deriveStubFactoryMethodName(final Class<? extends AbstractStub<?>> stubType) {
-        if (AbstractAsyncStub.class.isAssignableFrom(stubType)) {
-            return "newStub";
-        } else if (AbstractBlockingStub.class.isAssignableFrom(stubType)) {
-            return "newBlockingStub";
-        } else if (AbstractFutureStub.class.isAssignableFrom(stubType)) {
-            return "newFutureStub";
-        } else {
-            throw new IllegalArgumentException(
-                    "Unsupported stub type: " + stubType.getName() + " -> Please report this issue.");
+    private AbstractStub<?> createStub(Class<? extends AbstractStub<?>> stubClass, Channel channel) {
+        final StubFactory factory = this.stubFactories.stream()
+                .filter(stubFactory -> stubFactory.isApplicable(stubClass))
+                .findFirst()
+                .orElseThrow(() -> new IllegalArgumentException(
+                        "Unsupported stub type: " + stubClass.getName() + " -> Please report this issue."));
+
+        try {
+            return factory.createStub(stubClass, channel);
+        } catch (Exception exception) {
+            throw new IllegalStateException("Failed to create gRPC stub of type " + stubClass.getName(), exception);
         }
     }
-
 }

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

@@ -0,0 +1,34 @@
+/*
+ * Copyright (c) 2016-2020 Michael Zhang <[email protected]>
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
+ * documentation files (the "Software"), to deal in the Software without restriction, including without limitation the
+ * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all copies or substantial portions of the
+ * Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
+ * WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+ * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
+ * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+package net.devh.boot.grpc.client.stubfactory;
+
+import io.grpc.stub.AbstractAsyncStub;
+import io.grpc.stub.AbstractStub;
+
+public class AsyncStubFactory extends StandardJavaGrpcStubFactory {
+
+    @Override
+    public boolean isApplicable(Class<? extends AbstractStub<?>> stubType) {
+        return AbstractAsyncStub.class.isAssignableFrom(stubType);
+    }
+
+    @Override
+    protected String getFactoryMethodName() {
+        return "newStub";
+    }
+}

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

@@ -0,0 +1,34 @@
+/*
+ * Copyright (c) 2016-2020 Michael Zhang <[email protected]>
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
+ * documentation files (the "Software"), to deal in the Software without restriction, including without limitation the
+ * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all copies or substantial portions of the
+ * Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
+ * WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+ * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
+ * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+package net.devh.boot.grpc.client.stubfactory;
+
+import io.grpc.stub.AbstractBlockingStub;
+import io.grpc.stub.AbstractStub;
+
+public class BlockingStubFactory extends StandardJavaGrpcStubFactory {
+
+    @Override
+    public boolean isApplicable(Class<? extends AbstractStub<?>> stubType) {
+        return AbstractBlockingStub.class.isAssignableFrom(stubType);
+    }
+
+    @Override
+    protected String getFactoryMethodName() {
+        return "newBlockingStub";
+    }
+}

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

@@ -0,0 +1,34 @@
+/*
+ * Copyright (c) 2016-2020 Michael Zhang <[email protected]>
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
+ * documentation files (the "Software"), to deal in the Software without restriction, including without limitation the
+ * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all copies or substantial portions of the
+ * Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
+ * WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+ * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
+ * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+package net.devh.boot.grpc.client.stubfactory;
+
+import io.grpc.stub.AbstractFutureStub;
+import io.grpc.stub.AbstractStub;
+
+public class FutureStubFactory extends StandardJavaGrpcStubFactory {
+
+    @Override
+    public boolean isApplicable(Class<? extends AbstractStub<?>> stubType) {
+        return AbstractFutureStub.class.isAssignableFrom(stubType);
+    }
+
+    @Override
+    protected String getFactoryMethodName() {
+        return "newFutureStub";
+    }
+}

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

@@ -0,0 +1,71 @@
+/*
+ * Copyright (c) 2016-2020 Michael Zhang <[email protected]>
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
+ * documentation files (the "Software"), to deal in the Software without restriction, including without limitation the
+ * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all copies or substantial portions of the
+ * Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
+ * WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+ * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
+ * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+package net.devh.boot.grpc.client.stubfactory;
+
+import java.lang.reflect.Constructor;
+import java.lang.reflect.Method;
+
+import org.springframework.beans.BeanInstantiationException;
+
+import io.grpc.Channel;
+import io.grpc.stub.AbstractStub;
+
+/**
+ * A factory for creating stubs provided by standard grpc Java library. This is an abstract super-type that can be
+ * extended to support the different provided types.
+ */
+public abstract class StandardJavaGrpcStubFactory implements StubFactory {
+
+    /**
+     * Creates a stub of the given type.
+     *
+     * @param stubType The type of the stub to create.
+     * @param channel The channel used to create the stub.
+     * @return The newly created stub.
+     *
+     * @throws BeanInstantiationException If the stub couldn't be created.
+     */
+    @Override
+    public AbstractStub<?> createStub(final Class<? extends AbstractStub<?>> stubType, final Channel channel) {
+        try {
+            // First try the public static factory method
+            final String methodName = getFactoryMethodName();
+            final Class<?> enclosingClass = stubType.getEnclosingClass();
+            final Method factoryMethod = enclosingClass.getMethod(methodName, Channel.class);
+            return stubType.cast(factoryMethod.invoke(null, channel));
+        } catch (final Exception e) {
+            try {
+                // Use the private constructor as backup
+                final Constructor<? extends AbstractStub<?>> constructor =
+                        stubType.getDeclaredConstructor(Channel.class);
+                constructor.setAccessible(true);
+                return constructor.newInstance(channel);
+            } catch (final Exception e1) {
+                e.addSuppressed(e1);
+            }
+            throw new BeanInstantiationException(stubType, "Failed to create gRPC client", e);
+        }
+    }
+
+    /**
+     * Derives the name of the factory method from the given stub type.
+     *
+     * @return The name of the factory method.
+     */
+    protected abstract String getFactoryMethodName();
+}

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

@@ -0,0 +1,28 @@
+/*
+ * Copyright (c) 2016-2020 Michael Zhang <[email protected]>
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
+ * documentation files (the "Software"), to deal in the Software without restriction, including without limitation the
+ * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all copies or substantial portions of the
+ * Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
+ * WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+ * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
+ * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+package net.devh.boot.grpc.client.stubfactory;
+
+import io.grpc.Channel;
+import io.grpc.stub.AbstractStub;
+
+public interface StubFactory {
+
+    AbstractStub<?> createStub(final Class<? extends AbstractStub<?>> stubType, final Channel channel);
+
+    boolean isApplicable(Class<? extends AbstractStub<?>> stubType);
+}