diff --git a/samples/grpc-server/src/test/java/org/springframework/grpc/sample/GrpcClientApplicationTests.java b/samples/grpc-server/src/test/java/org/springframework/grpc/sample/GrpcClientApplicationTests.java new file mode 100644 index 0000000..4b16fd8 --- /dev/null +++ b/samples/grpc-server/src/test/java/org/springframework/grpc/sample/GrpcClientApplicationTests.java @@ -0,0 +1,82 @@ +package org.springframework.grpc.sample; + +import static org.assertj.core.api.Assertions.assertThat; + +import org.junit.jupiter.api.Nested; +import org.junit.jupiter.api.Test; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.boot.test.context.SpringBootTest; +import org.springframework.boot.test.context.TestConfiguration; +import org.springframework.context.ApplicationContext; +import org.springframework.grpc.client.FutureStubFactory; +import org.springframework.grpc.client.ImportGrpcClients; +import org.springframework.grpc.sample.proto.SimpleGrpc; +import org.springframework.grpc.test.AutoConfigureInProcessTransport; + +import io.grpc.stub.AbstractStub; + +public class GrpcClientApplicationTests { + + @Nested + @SpringBootTest + @AutoConfigureInProcessTransport + class NoAutowiredClients { + + @Autowired + private ApplicationContext context; + + @Test + void noStubIsCreated() { + assertThat(context.containsBeanDefinition("simpleBlockingStub")).isFalse(); + assertThat(context.containsBeanDefinition("simpleStub")).isFalse(); + assertThat(context.containsBeanDefinition("simpleFutureStub")).isFalse(); + assertThat(context.getBeanNamesForType(AbstractStub.class)).isEmpty(); + } + + } + + @Nested + @SpringBootTest(properties = "spring.grpc.client.default-channel.address=0.0.0.0:9090") + @AutoConfigureInProcessTransport + class DefaultAutowiredClients { + + @Autowired + private ApplicationContext context; + + @Test + void onlyDefaultStubIsCreated() { + assertThat(context.containsBeanDefinition("simpleBlockingStub")).isTrue(); + assertThat(context.getBean(SimpleGrpc.SimpleBlockingStub.class)).isNotNull(); + assertThat(context.containsBeanDefinition("simpleStub")).isFalse(); + assertThat(context.containsBeanDefinition("simpleFutureStub")).isFalse(); + assertThat(context.getBeanNamesForType(AbstractStub.class)).hasSize(1); + } + + } + + @Nested + @SpringBootTest(properties = "spring.grpc.client.default-channel.address=0.0.0.0:9090") + @AutoConfigureInProcessTransport + class SpecificAutowiredClients { + + @Autowired + private ApplicationContext context; + + @Test + void stubOfCorrectTypeIsCreated() { + assertThat(context.containsBeanDefinition("simpleFutureStub")).isTrue(); + assertThat(context.getBean(SimpleGrpc.SimpleFutureStub.class)).isNotNull(); + assertThat(context.containsBeanDefinition("simpleStub")).isFalse(); + assertThat(context.containsBeanDefinition("simpleBlockingStub")).isFalse(); + assertThat(context.getBeanNamesForType(AbstractStub.class)).hasSize(1); + } + + @TestConfiguration + @ImportGrpcClients(basePackageClasses = SimpleGrpc.class, factory = FutureStubFactory.class) + static class TestConfig { + + } + + } + +} diff --git a/spring-grpc-core/src/main/java/org/springframework/grpc/client/BlockingStubFactory.java b/spring-grpc-core/src/main/java/org/springframework/grpc/client/BlockingStubFactory.java index 76a80e9..6fb4b47 100644 --- a/spring-grpc-core/src/main/java/org/springframework/grpc/client/BlockingStubFactory.java +++ b/spring-grpc-core/src/main/java/org/springframework/grpc/client/BlockingStubFactory.java @@ -25,6 +25,11 @@ public class BlockingStubFactory extends AbstractStubFactory type) { + return super.supports(type) && !type.getSimpleName().contains("BlockingV2"); + } + @Override public int getOrder() { return SimpleStubFactory.SIMPLE_STUB_ORDER - 30; diff --git a/spring-grpc-core/src/main/java/org/springframework/grpc/client/BlockingV2StubFactory.java b/spring-grpc-core/src/main/java/org/springframework/grpc/client/BlockingV2StubFactory.java new file mode 100644 index 0000000..d2ce4e2 --- /dev/null +++ b/spring-grpc-core/src/main/java/org/springframework/grpc/client/BlockingV2StubFactory.java @@ -0,0 +1,43 @@ +/* + * Copyright 2024-2024 the original author or authors. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * https://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.springframework.grpc.client; + +import org.springframework.core.Ordered; + +import io.grpc.stub.AbstractBlockingStub; + +public class BlockingV2StubFactory extends AbstractStubFactory> implements Ordered { + + public BlockingV2StubFactory() { + super(AbstractBlockingStub.class); + } + + @Override + public boolean supports(Class type) { + return super.supports(type) && type.getSimpleName().contains("BlockingV2"); + } + + @Override + public int getOrder() { + return SimpleStubFactory.SIMPLE_STUB_ORDER - 30; + } + + @Override + protected String methodName() { + return "newBlockingStub"; + } + +} diff --git a/spring-grpc-core/src/main/java/org/springframework/grpc/client/GrpcClientRegistry.java b/spring-grpc-core/src/main/java/org/springframework/grpc/client/GrpcClientRegistry.java index a52de34..f7760ce 100644 --- a/spring-grpc-core/src/main/java/org/springframework/grpc/client/GrpcClientRegistry.java +++ b/spring-grpc-core/src/main/java/org/springframework/grpc/client/GrpcClientRegistry.java @@ -56,6 +56,7 @@ public class GrpcClientRegistry { public GrpcClientRegistry(GenericApplicationContext context) { this.context = context; stubs(new BlockingStubFactory()); + stubs(new BlockingV2StubFactory()); stubs(new FutureStubFactory()); stubs(new ReactorStubFactory()); stubs(new SimpleStubFactory());