Basic support for unary, serverStream, streamClient and bidi invocati…

…ons. Signed-off-by: Santiago Pericasgeertsen <[email protected]>
spericas · Feb 9, 2024 · 1cd3d02 · 1cd3d02
1 parent a22f8fe
commit 1cd3d02
Show file tree

Hide file tree

Showing 5 changed files with 374 additions and 75 deletions.
diff --git a/...s/webserver/protocols/src/test/java/io/helidon/examples/webserver/protocols/GrpcTest.java b/...s/webserver/protocols/src/test/java/io/helidon/examples/webserver/protocols/GrpcTest.java
@@ -16,9 +16,12 @@
 
 package io.helidon.examples.webserver.protocols;
 
+import java.util.Iterator;
+import java.util.List;
 import java.util.Locale;
 
 import com.google.protobuf.StringValue;
+import io.grpc.stub.StreamObserver;
 import io.helidon.examples.grpc.strings.Strings;
 import io.helidon.webclient.grpc.GrpcClient;
 import io.helidon.webclient.grpc.GrpcClientMethodDescriptor;
@@ -27,7 +30,7 @@
 import io.helidon.webserver.grpc.GrpcRouting;
 import io.helidon.webserver.testing.junit5.ServerTest;
 import io.helidon.webserver.testing.junit5.SetUpServer;
-import org.junit.jupiter.api.RepeatedTest;
+import org.junit.jupiter.api.Test;
 
 import static org.hamcrest.CoreMatchers.is;
 import static org.hamcrest.MatcherAssert.assertThat;
@@ -44,20 +47,94 @@ private GrpcTest(GrpcClient grpcClient) {
     @SetUpServer
     public static void setup(WebServerConfig.Builder builder) {
         builder.addRouting(GrpcRouting.builder()
-                        .unary(Strings.getDescriptor(),
-                                "StringService",
-                                "Upper",
-                                GrpcTest::blockingGrpcUpper));
+                .unary(Strings.getDescriptor(),
+                        "StringService",
+                        "Upper",
+                        GrpcTest::upper)
+                .serverStream(Strings.getDescriptor(),
+                        "StringService",
+                        "Split",
+                        GrpcTest::split)
+                .clientStream(Strings.getDescriptor(),
+                        "StringService",
+                        "Join",
+                        GrpcTest::join)
+                .bidi(Strings.getDescriptor(),
+                        "StringService",
+                        "Echo",
+                        GrpcTest::echo));
     }
 
-    private static Strings.StringMessage blockingGrpcUpper(Strings.StringMessage reqT) {
+    private static Strings.StringMessage upper(Strings.StringMessage reqT) {
         return Strings.StringMessage.newBuilder()
                 .setText(reqT.getText().toUpperCase(Locale.ROOT))
                 .build();
     }
 
-    @RepeatedTest(3)
-    void testSimpleCall() {
+    private static void split(Strings.StringMessage reqT,
+                              StreamObserver<Strings.StringMessage> streamObserver) {
+        String[] strings = reqT.getText().split(" ");
+        for (String string : strings) {
+            streamObserver.onNext(Strings.StringMessage.newBuilder()
+                    .setText(string)
+                    .build());
+
+        }
+        streamObserver.onCompleted();
+    }
+
+    private static StreamObserver<Strings.StringMessage> join(StreamObserver<Strings.StringMessage> streamObserver) {
+        return new StreamObserver<>() {
+            private StringBuilder builder;
+
+            @Override
+            public void onNext(Strings.StringMessage value) {
+                if (builder == null) {
+                    builder = new StringBuilder();
+                    builder.append(value.getText());
+                } else {
+                    builder.append(" ").append(value.getText());
+                }
+            }
+
+            @Override
+            public void onError(Throwable t) {
+                streamObserver.onError(t);
+            }
+
+            @Override
+            public void onCompleted() {
+                streamObserver.onNext(Strings.StringMessage.newBuilder()
+                        .setText(builder.toString())
+                        .build());
+                streamObserver.onCompleted();
+            }
+        };
+    }
+
+    private static StreamObserver<Strings.StringMessage> echo(StreamObserver<Strings.StringMessage> streamObserver) {
+        return new StreamObserver<>() {
+            private StringBuilder builder;
+
+            @Override
+            public void onNext(Strings.StringMessage value) {
+                streamObserver.onNext(value);
+            }
+
+            @Override
+            public void onError(Throwable t) {
+                streamObserver.onError(t);
+            }
+
+            @Override
+            public void onCompleted() {
+                streamObserver.onCompleted();
+            }
+        };
+    }
+
+    @Test
+    void testUnaryUpper() {
         GrpcServiceDescriptor serviceDescriptor =
                 GrpcServiceDescriptor.builder()
                         .serviceName("StringService")
@@ -70,7 +147,63 @@ void testSimpleCall() {
 
         StringValue r = grpcClient.serviceClient(serviceDescriptor)
                 .unary("Upper", StringValue.of("hello"));
-        System.out.println("r = " + r.getValue());
         assertThat(r.getValue(), is("HELLO"));
     }
+
+    @Test
+    void testServerStreamingSplit() {
+        GrpcServiceDescriptor serviceDescriptor =
+                GrpcServiceDescriptor.builder()
+                        .serviceName("StringService")
+                        .putMethod("Split",
+                                GrpcClientMethodDescriptor.serverStreaming("StringService", "Split")
+                                        .requestType(StringValue.class)
+                                        .responseType(StringValue.class)
+                                        .build())
+                        .build();
+
+        Iterator<StringValue> r = grpcClient.serviceClient(serviceDescriptor)
+                .serverStream("Split", StringValue.of("hello world"));
+        assertThat(r.next().getValue(), is("hello"));
+        assertThat(r.next().getValue(), is("world"));
+        assertThat(r.hasNext(), is(false));
+    }
+
+    @Test
+    void testClientStreamingJoin() {
+        GrpcServiceDescriptor serviceDescriptor =
+                GrpcServiceDescriptor.builder()
+                        .serviceName("StringService")
+                        .putMethod("Join",
+                                GrpcClientMethodDescriptor.clientStreaming("StringService", "Join")
+                                        .requestType(StringValue.class)
+                                        .responseType(StringValue.class)
+                                        .build())
+                        .build();
+
+        StringValue r = grpcClient.serviceClient(serviceDescriptor)
+                .clientStream("Join", List.of(StringValue.of("hello"),
+                                                          StringValue.of("world")).iterator());
+        assertThat(r.getValue(), is("hello world"));
+    }
+
+    @Test
+    void testBidirectionalEcho() {
+        GrpcServiceDescriptor serviceDescriptor =
+                GrpcServiceDescriptor.builder()
+                        .serviceName("StringService")
+                        .putMethod("Echo",
+                                GrpcClientMethodDescriptor.bidirectional("StringService", "Echo")
+                                        .requestType(StringValue.class)
+                                        .responseType(StringValue.class)
+                                        .build())
+                        .build();
+
+        Iterator<StringValue> r = grpcClient.serviceClient(serviceDescriptor)
+                .bidi("Echo", List.of(StringValue.of("hello"),
+                                                  StringValue.of("world")).iterator());
+        assertThat(r.next().getValue(), is("hello"));
+        assertThat(r.next().getValue(), is("world"));
+        assertThat(r.hasNext(), is(false));
+    }
 }