io.stargate.sdk.grpc.utils.StreamObserverToReactivePublisher Maven / Gradle / Ivy

Go to download
package io.stargate.sdk.grpc.utils;

import io.grpc.stub.StreamObserver;
import org.reactivestreams.Publisher;
import org.reactivestreams.Subscriber;
import reactor.core.publisher.BaseSubscriber;

/**
 * Mapping Grpc StreamObserver to Flux.
 *
 * @param 
 *     current param.
 */
public class StreamObserverToReactivePublisher implements Publisher, StreamObserver {

    /** subscriber. */
    private Subscriber subscriber;

    /**
     * Default constructor.
     */
    public StreamObserverToReactivePublisher() {
    }

    @Override
    /** @{inheritDocs} */
    public void onNext(T l) {
        subscriber.onNext(l);
    }

    @Override
    /** @{inheritDocs} */
    public void onError(Throwable throwable) {
        subscriber.onError(throwable);
    }

    @Override
    /** @{inheritDocs} */
    public void onCompleted() {
        subscriber.onComplete();
    }

    @Override
    /** @{inheritDocs} */
    public void subscribe(Subscriber subscriber) {
        this.subscriber = subscriber;
        this.subscriber.onSubscribe(new BaseSubscriber() {});
    }
}