Streaming 通信

流的实现原理

Triple协议的流模式

  • 从协议层来说,Triple 是建立在 HTTP2 基础上的,所以直接拥有所有 HTTP2 的能力,故拥有了分 stream 和全双工的能力。

  • 框架层来说,StreamObserver 作为流的接口提供给用户,用于入参和出参提供流式处理。框架在收发 stream data 时进行相应的接口调用, 从而保证流的生命周期完整。

开启 Triple 新特性

Stream 流

Stream 是 Dubbo3 新提供的一种调用类型,在以下场景时建议使用流的方式:

  • 接口需要发送大量数据,这些数据无法被放在一个 RPC 的请求或响应中,需要分批发送,但应用层如果按照传统的多次 RPC 方式无法解决顺序和性能的问题,如果需要保证有序,则只能串行发送
  • 流式场景,数据需要按照发送顺序处理, 数据本身是没有确定边界的
  • 推送类场景,多个消息在同一个调用的上下文中被发送和处理

Stream 分为以下三种:

  • SERVER_STREAM(服务端流) SERVER_STREAM
  • CLIENT_STREAM(客户端流) CLIENT_STREAM
  • BIDIRECTIONAL_STREAM(双向流) BIDIRECTIONAL_STREAM

由于 java 语言的限制,BIDIRECTIONAL_STREAM 和 CLIENT_STREAM 的实现是一样的。

在 Dubbo3 中,流式接口以 SteamObserver 声明和使用,用户可以通过使用和实现这个接口来发送和处理流的数据、异常和结束。

对于 Dubbo2 用户来说,可能会对StreamObserver感到陌生,这是Dubbo3定义的一种流类型,Dubbo2 中并不存在 Stream 的类型,所以对于迁移场景没有任何影响。

流的语义保证

  • 提供消息边界,可以方便地对消息单独处理
  • 严格有序,发送端的顺序和接收端顺序一致
  • 全双工,发送不需要等待
  • 支持取消和超时

非 PB 序列化的流

api

  1. public interface IWrapperGreeter {
  2. StreamObserver<String> sayHelloStream(StreamObserver<String> response);
  3. void sayHelloServerStream(String request, StreamObserver<String> response);
  4. }

Stream 方法的方法入参和返回值是严格约定的,为防止写错而导致问题,Dubbo3 框架侧做了对参数的检查, 如果出错则会抛出异常。 对于 双向流(BIDIRECTIONAL_STREAM), 需要注意参数中的 StreamObserver 是响应流,返回参数中的 StreamObserver 为请求流。

实现类

  1. public class WrapGreeterImpl implements WrapGreeter {
  2. //...
  3. @Override
  4. public StreamObserver<String> sayHelloStream(StreamObserver<String> response) {
  5. return new StreamObserver<String>() {
  6. @Override
  7. public void onNext(String data) {
  8. System.out.println(data);
  9. response.onNext("hello,"+data);
  10. }
  11. @Override
  12. public void onError(Throwable throwable) {
  13. throwable.printStackTrace();
  14. }
  15. @Override
  16. public void onCompleted() {
  17. System.out.println("onCompleted");
  18. response.onCompleted();
  19. }
  20. };
  21. }
  22. @Override
  23. public void sayHelloServerStream(String request, StreamObserver<String> response) {
  24. for (int i = 0; i < 10; i++) {
  25. response.onNext("hello," + request);
  26. }
  27. response.onCompleted();
  28. }
  29. }

调用方式

  1. delegate.sayHelloServerStream("server stream", new StreamObserver<String>() {
  2. @Override
  3. public void onNext(String data) {
  4. System.out.println(data);
  5. }
  6. @Override
  7. public void onError(Throwable throwable) {
  8. throwable.printStackTrace();
  9. }
  10. @Override
  11. public void onCompleted() {
  12. System.out.println("onCompleted");
  13. }
  14. });
  15. StreamObserver<String> request = delegate.sayHelloStream(new StreamObserver<String>() {
  16. @Override
  17. public void onNext(String data) {
  18. System.out.println(data);
  19. }
  20. @Override
  21. public void onError(Throwable throwable) {
  22. throwable.printStackTrace();
  23. }
  24. @Override
  25. public void onCompleted() {
  26. System.out.println("onCompleted");
  27. }
  28. });
  29. for (int i = 0; i < n; i++) {
  30. request.onNext("stream request" + i);
  31. }
  32. request.onCompleted();

使用 Protobuf 序列化的流

对于 Protobuf 序列化方式,推荐编写 IDL 使用 compiler 插件进行编译生成。生成的代码大致如下:

  1. public interface PbGreeter {
  2. static final String JAVA_SERVICE_NAME = "org.apache.dubbo.sample.tri.PbGreeter";
  3. static final String SERVICE_NAME = "org.apache.dubbo.sample.tri.PbGreeter";
  4. static final boolean inited = PbGreeterDubbo.init();
  5. //...
  6. void greetServerStream(org.apache.dubbo.sample.tri.GreeterRequest request, org.apache.dubbo.common.stream.StreamObserver<org.apache.dubbo.sample.tri.GreeterReply> responseObserver);
  7. org.apache.dubbo.common.stream.StreamObserver<org.apache.dubbo.sample.tri.GreeterRequest> greetStream(org.apache.dubbo.common.stream.StreamObserver<org.apache.dubbo.sample.tri.GreeterReply> responseObserver);
  8. }

最后修改 December 16, 2022: Fix check (#1736) (97972c1)