Basic stream impl

Author: vietj

src/main/generated/io/vertx/core/net/ConnectOptionsConverter.java

@@ -15,7 +15,6 @@
 import io.vertx.codegen.annotations.GenIgnore;
 import io.vertx.codegen.annotations.Nullable;
 import io.vertx.codegen.annotations.VertxGen;
-import io.vertx.core.AsyncResult;
 import io.vertx.core.Future;
 import io.vertx.core.Handler;
 import io.vertx.core.eventbus.impl.DefaultSerializableChecker;
@@ -199,6 +198,10 @@ default <T> Future<Message<T>> request(String address, @Nullable Object message)
    */
   <T> MessageProducer<T> publisher(String address, DeliveryOptions options);
 
+  Future<Void> bindStream(String address, Handler<MessageStream> handler);
+
+  Future<MessageStream> connectStream(String address);
+
   /**
    * Register a message codec.
    * <p>

src/main/java/io/vertx/core/eventbus/EventBus.java

@@ -0,0 +1,27 @@
+/*
+ * Copyright (c) 2011-2021 Contributors to the Eclipse Foundation
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License 2.0 which is available at
+ * http://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
+ * which is available at https://www.apache.org/licenses/LICENSE-2.0.
+ *
+ * SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
+ */
+package io.vertx.core.eventbus;
+
+import io.vertx.codegen.annotations.VertxGen;
+import io.vertx.core.Handler;
+
+@VertxGen
+public interface MessageStream {
+
+  void handler(Handler<Message<String>> handler);
+
+  void endHandler(Handler<Void> handler);
+
+  void write(String msg);
+
+  void end();
+
+}

src/main/java/io/vertx/core/eventbus/MessageStream.java

@@ -0,0 +1,50 @@
+package io.vertx.core.eventbus.impl;
+
+import io.vertx.core.Handler;
+import io.vertx.core.Promise;
+import io.vertx.core.eventbus.Message;
+import io.vertx.core.eventbus.MessageStream;
+import io.vertx.core.impl.ContextInternal;
+
+import java.util.concurrent.TimeoutException;
+
+class ClientStream extends StreamBase implements Handler<Long> {
+
+  private final Promise<MessageStream> promise2;
+  private final long timeoutID;
+
+  public ClientStream(EventBusImpl eventBus, String sourceAddress, ContextInternal ctx, Promise<MessageStream> promise2) {
+    super(sourceAddress, ctx, eventBus, sourceAddress, true);
+    this.promise2 = promise2;
+    this.timeoutID = ctx.setTimer(3_000, this);
+  }
+
+  @Override
+  public void handle(Long event) {
+    unregister();
+    promise2.fail(new TimeoutException());
+  }
+
+  @Override
+  protected boolean doReceive(Message msg) {
+    if (msg.body() instanceof SynFrame) {
+      if (context.owner().cancelTimer(timeoutID)) {
+        base = (MessageImpl) msg;
+        SynFrame syn = (SynFrame) msg.body();
+        remoteAddress = syn.src;
+        promise2.complete(this);
+      }
+      return true;
+    } else {
+      if (base != null) {
+        return super.doReceive(msg);
+      } else {
+        if (context.owner().cancelTimer(timeoutID)) {
+          unregister();
+          promise2.fail(new IllegalStateException());
+        }
+        return true;
+      }
+    }
+  }
+}

src/main/java/io/vertx/core/eventbus/impl/ClientStream.java

@@ -63,7 +63,7 @@ public CodecManager() {
     this.systemCodecs = codecs(NULL_MESSAGE_CODEC, PING_MESSAGE_CODEC, STRING_MESSAGE_CODEC, BUFFER_MESSAGE_CODEC, JSON_OBJECT_MESSAGE_CODEC, JSON_ARRAY_MESSAGE_CODEC,
       BYTE_ARRAY_MESSAGE_CODEC, INT_MESSAGE_CODEC, LONG_MESSAGE_CODEC, FLOAT_MESSAGE_CODEC, DOUBLE_MESSAGE_CODEC,
       BOOLEAN_MESSAGE_CODEC, SHORT_MESSAGE_CODEC, CHAR_MESSAGE_CODEC, BYTE_MESSAGE_CODEC, REPLY_EXCEPTION_MESSAGE_CODEC,
-      clusterSerializableCodec, serializableCodec);
+      clusterSerializableCodec, serializableCodec, SynFrame.CODEC, FinFrame.CODEC);
   }
 
   public MessageCodec lookupCodec(Object body, String codecName, boolean local) {
@@ -98,6 +98,10 @@ public MessageCodec lookupCodec(Object body, String codecName, boolean local) {
       codec = CHAR_MESSAGE_CODEC;
     } else if (body instanceof Byte) {
       codec = BYTE_MESSAGE_CODEC;
+    } else if (body instanceof SynFrame) {
+      codec = SynFrame.CODEC;
+    } else if (body instanceof FinFrame) {
+      codec = FinFrame.CODEC;
     } else if (body instanceof ReplyException) {
       codec = defaultCodecMap.get(body.getClass());
       if (codec == null) {

src/main/java/io/vertx/core/eventbus/impl/CodecManager.java

@@ -158,6 +158,34 @@ public <T> MessageProducer<T> publisher(String address, DeliveryOptions options)
     return new MessageProducerImpl<>(vertx, address, false, options);
   }
 
+  @Override
+  public Future<Void> bindStream(String address, Handler<MessageStream> handler) {
+    ContextInternal ctx = vertx.getOrCreateContext();
+    HandlerRegistration reg = new StreamServer(this, ctx, address, handler);
+    Promise<Void> promise = ctx.promise();
+    reg.register(true, false, promise);
+    return promise.future();
+  }
+
+  @Override
+  public Future<MessageStream> connectStream(String address) {
+    ContextInternal ctx = vertx.getOrCreateContext();
+    String sourceAddress = generateReplyAddress();
+    Promise<MessageStream> promise2 = ctx.promise();
+    StreamBase reg = new ClientStream(this, sourceAddress, ctx, promise2);
+    Promise<Void> promise = ctx.promise();
+    reg.register(false, false, promise);
+    promise.future().onComplete(ar -> {
+      if (ar.succeeded()) {
+        boolean localOnly = isLocalOnly(new DeliveryOptions()); // Change when options
+        MessageImpl msg = createMessage(true, localOnly, address, MultiMap.caseInsensitiveMultiMap(), new SynFrame(sourceAddress, address), null);
+        msg.setReplyAddress(sourceAddress);
+        sendOrPub(ctx, msg, new DeliveryOptions(), ctx.promise());
+      }
+    });
+    return promise2.future();
+  }
+
   @Override
   public EventBus publish(String address, Object message) {
     return publish(address, message, new DeliveryOptions());

src/main/java/io/vertx/core/eventbus/impl/EventBusImpl.java

@@ -0,0 +1,47 @@
+package io.vertx.core.eventbus.impl;
+
+import io.netty.util.CharsetUtil;
+import io.vertx.core.buffer.Buffer;
+import io.vertx.core.eventbus.MessageCodec;
+
+class FinFrame implements Frame {
+
+  public static final MessageCodec<FinFrame, FinFrame> CODEC = new MessageCodec<>() {
+    @Override
+    public void encodeToWire(Buffer buffer, FinFrame synFrame) {
+      byte[] strBytes = synFrame.addr.getBytes(CharsetUtil.UTF_8);
+      buffer.appendInt(strBytes.length);
+      buffer.appendBytes(strBytes);
+    }
+
+    @Override
+    public FinFrame decodeFromWire(int pos, Buffer buffer) {
+      int length = buffer.getInt(pos);
+      pos += 4;
+      byte[] bytes = buffer.getBytes(pos, pos + length);
+      String src = new String(bytes, CharsetUtil.UTF_8);
+      return new FinFrame(src);
+    }
+
+    @Override
+    public FinFrame transform(FinFrame finFrame) {
+      return finFrame;
+    }
+
+    @Override
+    public String name() {
+      return "frame.fin";
+    }
+
+    @Override
+    public byte systemCodecID() {
+      return 19;
+    }
+  };
+
+  final String addr;
+
+  public FinFrame(String addr) {
+    this.addr = addr;
+  }
+}

src/main/java/io/vertx/core/eventbus/impl/FinFrame.java

@@ -0,0 +1,4 @@
+package io.vertx.core.eventbus.impl;
+
+public interface Frame {
+}

src/main/java/io/vertx/core/eventbus/impl/Frame.java

@@ -0,0 +1,77 @@
+package io.vertx.core.eventbus.impl;
+
+import io.vertx.core.Handler;
+import io.vertx.core.MultiMap;
+import io.vertx.core.eventbus.DeliveryOptions;
+import io.vertx.core.eventbus.Message;
+import io.vertx.core.eventbus.MessageStream;
+import io.vertx.core.impl.ContextInternal;
+
+class StreamBase extends HandlerRegistration implements MessageStream {
+
+  MessageImpl base;
+  private Handler<Message<String>> handler;
+  private Handler<Void> endHandler;
+  final String localAddress;
+  String remoteAddress;
+  private boolean halfClosed;
+
+  StreamBase(String localAddress, ContextInternal context, EventBusImpl bus, String address, boolean src) {
+    super(context, bus, address, src);
+    this.localAddress = localAddress;
+  }
+
+  @Override
+  protected boolean doReceive(Message msg) {
+    if (msg.body() instanceof FinFrame) {
+      Handler<Void> h = endHandler;
+      if (h != null) {
+        h.handle(null);
+      }
+      if (halfClosed) {
+        unregister();
+      } else {
+        halfClosed = true;
+      }
+      return true;
+    } else {
+      Handler<Message<String>> h = handler;
+      if (h != null) {
+        h.handle(msg);
+      }
+      return true;
+    }
+  }
+
+  @Override
+  protected void dispatch(Message msg, ContextInternal context, Handler handler) {
+
+  }
+
+  @Override
+  public void handler(Handler<Message<String>> handler) {
+    this.handler = handler;
+  }
+
+  @Override
+  public void endHandler(Handler<Void> handler) {
+    this.endHandler = handler;
+  }
+
+  @Override
+  public void write(String body) {
+    MessageImpl msg = base.createReply(body, new DeliveryOptions());
+    bus.sendOrPub(context, msg, new DeliveryOptions(), context.promise());
+  }
+
+  @Override
+  public void end() {
+    MessageImpl msg = base.createReply(new FinFrame(remoteAddress), new DeliveryOptions());
+    bus.sendOrPub(context, msg, new DeliveryOptions(), context.promise());
+    if (halfClosed) {
+      unregister();
+    } else {
+      halfClosed = true;
+    }
+  }
+}