gRPC入门-双向流式通信

Gradle工程的环境搭建依然和前文一致，参考：gRPC入门-Hello World

示例代码

• 在src\main\proto下新建.proto契约，如下：

  syntax = "proto3";
  
  package com.mattie.netty.grpc;
  
  option java_package = "com.mattie.netty.grpc";
  option java_outer_classname = "HelloWorldProtos";
  
  service HelloService {
      //双向流
      rpc biStream (stream StreamReq) returns (stream StreamResp) {};
  }
  
  //双向流
   message StreamReq {
     string req_info = 1;
  }
  
  message StreamResp {
      string resp_info = 1;
  }
  

• 执行gradle clean build，自动生成gRPC相关代码: HelloServiceGrpc, 我们需要使用就是它和它的几个内部类。

• 自定义服务类MyServiceImpl继承HelloServiceGrpc的内部类HelloServiceImplBase并重写契约中定义的服务biStream。注意到重写的方法接收的参数类型是StreamObserver（针对response），返回值的类型也是StreamObserver（针对request）。

  public class MyServiceImpl extends HelloServiceImplBase {
      @Override
      public StreamObserver biStream(StreamObserver responseObserver) {
    return new StreamObserver() {
  
        //接收请求
        @Override
        public void onNext(HelloWorldProtos.StreamReq streamReq) {
            System.out.println(streamReq.getReqInfo());
            //接收请求后就返回一个响应
            responseObserver.onNext(HelloWorldProtos.StreamResp.newBuilder().setRespInfo("from server").build());
        }
  
        @Override
        public void onError(Throwable throwable) {
  
        }
  
        //客户端发送数据完毕
        @Override
        public void onCompleted() {
            //服务端也完毕
            responseObserver.onCompleted();
        }
      };
    }
  

RequestObserver在服务端实现，responseObserver在客户端实现。

因此，在服务代码MyServiceImpl中需要返回一个requestObserver，实现其中的回调方法:
1. onNext表示接收到一个request，这里通过responseObserver的onNext()立刻返回了一条数据。
2. onCompleted表示客户端已经发送数据完毕，这里调用responseObserver的onCompleted()也告诉客户端连接关闭。

• 客户端不仅需要发送数据，而且需要实现一个responseObserver:

    public void biCommute() {
        StreamObserver reqStreamObserver = this.stub.biStream(new StreamObserver() {
            //接收到服务返回
            @Override
            public void onNext(HelloWorldProtos.StreamResp streamResp) {
                System.out.println(streamResp.getRespInfo());
            }
  
            @Override
            public void onError(Throwable throwable) {
  
            }
  
            //服务发送完毕
            @Override
            public void onCompleted() {
  
            }
        });
  
        reqStreamObserver.onNext(HelloWorldProtos.StreamReq.newBuilder().setReqInfo("hello server1").build());
        reqStreamObserver.onNext(HelloWorldProtos.StreamReq.newBuilder().setReqInfo("hello server2").build());
        reqStreamObserver.onNext(HelloWorldProtos.StreamReq.newBuilder().setReqInfo("hello server3").build());
        reqStreamObserver.onNext(HelloWorldProtos.StreamReq.newBuilder().setReqInfo("hello server4").build());
        reqStreamObserver.onCompleted();
  
       try {
          Thread.sleep(10000);
       } catch (InterruptedException e) {
          e.printStackTrace();
       }
     }
  

1. 这里调用服务方法biStream使用的是stub（异步），而非之前的blockingStub（同步）。
2. 调用biStream需要传入responseObserver作为参数，同时返回值是requestObserver。
3. 通过requestObserver的onNext()不断发送数据。

• 执行程序，客户端发送四条数据，服务端每接收到一条数据就响应一条from server给客户端。

为了能更好的看到执行效果，可以在程序最后增加sleep(10000)，观察服务的异步响应过程。

双向流