io.vertx.grpcio.common.impl.ReadStreamAdapter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vertx-grpcio-common Show documentation

The newest version!

/*
 * Copyright (c) 2011-2022 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.grpcio.common.impl;

import io.vertx.grpc.common.GrpcReadStream;

/**
 * An adapter between gRPC and Vert.x back-pressure.
 */
public class ReadStreamAdapter {

  private GrpcReadStream stream;
  private int request = 0;

  /**
   * Init the adapter with the stream.
   */
  public final void init(GrpcReadStream stream, BridgeMessageDecoder decoder) {
    stream.messageHandler(msg -> {
      handleMessage(decoder.decode(msg));
    });
    stream.endHandler(v -> {
      handleClose();
    });
    this.stream = stream;
    stream.pause();
    if (request > 0) {
      stream.fetch(request);
    }
  }

  /**
   * Override this to handle close event
   */
  protected void handleClose() {

  }

  /**
   * Override this to handle message event
   */
  protected void handleMessage(T msg) {

  }

  /**
   * Request {@code num} messages
   */
  public final void request(int num) {
    if (stream != null) {
      stream.fetch(num);
    } else {
      request += num;
    }
  }
}