io.vertx.rxjava3.grpc.common.GrpcWriteStream Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of vertx-rx-java3 Show documentation
There is a newer version: 5.0.0.CR2
/*
 * Copyright 2014 Red Hat, Inc.
 *
 * Red Hat licenses this file to you under the Apache License, version 2.0
 * (the "License"); you may not use this file except in compliance with the
 * License.  You may obtain a copy of the License at:
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
 * License for the specific language governing permissions and limitations
 * under the License.
 */

package io.vertx.rxjava3.grpc.common;

import io.vertx.rxjava3.RxHelper;
import io.vertx.rxjava3.ObservableHelper;
import io.vertx.rxjava3.FlowableHelper;
import io.vertx.rxjava3.impl.AsyncResultMaybe;
import io.vertx.rxjava3.impl.AsyncResultSingle;
import io.vertx.rxjava3.impl.AsyncResultCompletable;
import io.vertx.rxjava3.WriteStreamObserver;
import io.vertx.rxjava3.WriteStreamSubscriber;
import java.util.Map;
import java.util.Set;
import java.util.List;
import java.util.Iterator;
import java.util.function.Function;
import java.util.stream.Collectors;
import io.vertx.core.Handler;
import io.vertx.core.AsyncResult;
import io.vertx.core.json.JsonObject;
import io.vertx.core.json.JsonArray;
import io.vertx.lang.rx.RxGen;
import io.vertx.lang.rx.TypeArg;
import io.vertx.lang.rx.MappingIterator;


@RxGen(io.vertx.grpc.common.GrpcWriteStream.class)
public class GrpcWriteStream implements io.vertx.rxjava3.core.streams.WriteStream {

  @Override
  public String toString() {
    return delegate.toString();
  }

  @Override
  public boolean equals(Object o) {
    if (this == o) return true;
    if (o == null || getClass() != o.getClass()) return false;
    GrpcWriteStream that = (GrpcWriteStream) o;
    return delegate.equals(that.delegate);
  }
  
  @Override
  public int hashCode() {
    return delegate.hashCode();
  }

  public static final TypeArg __TYPE_ARG = new TypeArg<>(    obj -> new GrpcWriteStream((io.vertx.grpc.common.GrpcWriteStream) obj),
    GrpcWriteStream::getDelegate
  );

  private final io.vertx.grpc.common.GrpcWriteStream delegate;
  public final TypeArg __typeArg_0;
  
  public GrpcWriteStream(io.vertx.grpc.common.GrpcWriteStream delegate) {
    this.delegate = delegate;
    this.__typeArg_0 = TypeArg.unknown();  }

  public GrpcWriteStream(Object delegate, TypeArg typeArg_0) {
    this.delegate = (io.vertx.grpc.common.GrpcWriteStream)delegate;
    this.__typeArg_0 = typeArg_0;
  }

  public io.vertx.grpc.common.GrpcWriteStream getDelegate() {
    return delegate;
  }

  private WriteStreamObserver observer;
  private WriteStreamSubscriber subscriber;

  public synchronized WriteStreamObserver toObserver() {
    if (observer == null) {
      Function conv = (Function) __typeArg_0.unwrap;
      observer = RxHelper.toObserver(getDelegate(), conv);
    }
    return observer;
  }

  public synchronized WriteStreamSubscriber toSubscriber() {
    if (subscriber == null) {
      Function conv = (Function) __typeArg_0.unwrap;
      subscriber = RxHelper.toSubscriber(getDelegate(), conv);
    }
    return subscriber;
  }


  /**
   * Same as  but with an handler called when the operation completes
   * @param data 
   * @return 
   */
  public io.reactivex.rxjava3.core.Completable write(T data) { 
    io.reactivex.rxjava3.core.Completable ret = rxWrite(data);
    ret = ret.cache();
    ret.subscribe(io.vertx.rxjava3.CompletableHelper.nullObserver());
    return ret;
  }

  /**
   * Same as  but with an handler called when the operation completes
   * @param data 
   * @return 
   */
  public io.reactivex.rxjava3.core.Completable rxWrite(T data) { 
    return AsyncResultCompletable.toCompletable( handler -> {
      delegate.write(__typeArg_0.unwrap(data), handler);
    });
  }

  /**
   * Same as {@link io.vertx.rxjava3.core.streams.WriteStream#end} but with an handler called when the operation completes
   * @return 
   */
  public io.reactivex.rxjava3.core.Completable end() { 
    io.reactivex.rxjava3.core.Completable ret = rxEnd();
    ret = ret.cache();
    ret.subscribe(io.vertx.rxjava3.CompletableHelper.nullObserver());
    return ret;
  }

  /**
   * Same as {@link io.vertx.rxjava3.core.streams.WriteStream#end} but with an handler called when the operation completes
   * @return 
   */
  public io.reactivex.rxjava3.core.Completable rxEnd() { 
    return AsyncResultCompletable.toCompletable( handler -> {
      delegate.end(handler);
    });
  }

  /**
   * Same as  but with an handler called when the operation completes
   * @param data 
   * @return 
   */
  public io.reactivex.rxjava3.core.Completable end(T data) { 
    io.reactivex.rxjava3.core.Completable ret = rxEnd(data);
    ret = ret.cache();
    ret.subscribe(io.vertx.rxjava3.CompletableHelper.nullObserver());
    return ret;
  }

  /**
   * Same as  but with an handler called when the operation completes
   * @param data 
   * @return 
   */
  public io.reactivex.rxjava3.core.Completable rxEnd(T data) { 
    return AsyncResultCompletable.toCompletable( handler -> {
      delegate.end(__typeArg_0.unwrap(data), handler);
    });
  }

  /**
   * This will return true if there are more bytes in the write queue than the value set using {@link io.vertx.rxjava3.grpc.common.GrpcWriteStream#setWriteQueueMaxSize}
   * @return true if write queue is full
   */
  public boolean writeQueueFull() { 
    boolean ret = delegate.writeQueueFull();
    return ret;
  }

  /**
   * @return the  to reader metadata headers
   */
  public io.vertx.rxjava3.core.MultiMap headers() { 
    io.vertx.rxjava3.core.MultiMap ret = io.vertx.rxjava3.core.MultiMap.newInstance((io.vertx.core.MultiMap)delegate.headers());
    return ret;
  }

  /**
   * Set the stream encoding, e.g identity or gzip.
   *
   * It must be called before sending any message, otherwise {@code identity will be used.
   * @param encoding the target message encoding
   * @return a reference to this, so the API can be used fluently
   */
  public io.vertx.rxjava3.grpc.common.GrpcWriteStream encoding(java.lang.String encoding) { 
    delegate.encoding(encoding);
    return this;
  }

  public io.vertx.rxjava3.grpc.common.GrpcWriteStream exceptionHandler(io.vertx.core.Handler handler) { 
    delegate.exceptionHandler(handler);
    return this;
  }

  public io.vertx.rxjava3.grpc.common.GrpcWriteStream setWriteQueueMaxSize(int i) { 
    delegate.setWriteQueueMaxSize(i);
    return this;
  }

  public io.vertx.rxjava3.grpc.common.GrpcWriteStream drainHandler(io.vertx.core.Handler handler) { 
    delegate.drainHandler(handler);
    return this;
  }

  /**
   * Write an encoded gRPC message.
   * @param message the message
   * @return a future completed with the result
   */
  public io.reactivex.rxjava3.core.Completable writeMessage(io.vertx.rxjava3.grpc.common.GrpcMessage message) { 
    io.reactivex.rxjava3.core.Completable ret = rxWriteMessage(message);
    ret = ret.cache();
    ret.subscribe(io.vertx.rxjava3.CompletableHelper.nullObserver());
    return ret;
  }

  /**
   * Write an encoded gRPC message.
   * @param message the message
   * @return a future completed with the result
   */
  public io.reactivex.rxjava3.core.Completable rxWriteMessage(io.vertx.rxjava3.grpc.common.GrpcMessage message) { 
    return AsyncResultCompletable.toCompletable(delegate.writeMessage(message.getDelegate()));
  }

  /**
   * End the stream with an encoded gRPC message.
   * @param message the message
   * @return a future completed with the result
   */
  public io.reactivex.rxjava3.core.Completable endMessage(io.vertx.rxjava3.grpc.common.GrpcMessage message) { 
    io.reactivex.rxjava3.core.Completable ret = rxEndMessage(message);
    ret = ret.cache();
    ret.subscribe(io.vertx.rxjava3.CompletableHelper.nullObserver());
    return ret;
  }

  /**
   * End the stream with an encoded gRPC message.
   * @param message the message
   * @return a future completed with the result
   */
  public io.reactivex.rxjava3.core.Completable rxEndMessage(io.vertx.rxjava3.grpc.common.GrpcMessage message) { 
    return AsyncResultCompletable.toCompletable(delegate.endMessage(message.getDelegate()));
  }

  /**
   * Cancel the stream.
   */
  public void cancel() { 
    delegate.cancel();
  }

  public static  GrpcWriteStream newInstance(io.vertx.grpc.common.GrpcWriteStream arg) {
    return arg != null ? new GrpcWriteStream(arg) : null;
  }

  public static  GrpcWriteStream newInstance(io.vertx.grpc.common.GrpcWriteStream arg, TypeArg __typeArg_T) {
    return arg != null ? new GrpcWriteStream(arg, __typeArg_T) : null;
  }

}