com.salesforce.reactivegrpc.common.ReactiveStreamObserverPublisherClient Maven / Gradle / Ivy

Go to download
/*
 *  Copyright (c) 2017, salesforce.com, inc.
 *  All rights reserved.
 *  Licensed under the BSD 3-Clause license.
 *  For full license text, see LICENSE.txt file in the repo root  or https://opensource.org/licenses/BSD-3-Clause
 */

package com.salesforce.reactivegrpc.common;

import io.grpc.stub.ClientCallStreamObserver;
import org.reactivestreams.Subscriber;

/**
 * The gRPC client-side implementation of {@link ReactiveStreamObserverPublisherBase}.
 *
 * @param 
 */
public class ReactiveStreamObserverPublisherClient extends ReactiveStreamObserverPublisherBase {
    private ClientCallStreamObserver callStreamObserver;

    public ReactiveStreamObserverPublisherClient(ClientCallStreamObserver callStreamObserver) {
        super(callStreamObserver);
        this.callStreamObserver = callStreamObserver;
    }

    @Override
    protected ReactiveStreamObserverPublisherSubscriptionBase createSubscription() {
        return new ReactiveStreamObserverPublisherSubscriptionBase() {
            @Override
            public void cancel() {
                ReactiveStreamObserverPublisherClient.this.cancel();
                if (callStreamObserver != null) {
                    callStreamObserver.cancel("Client canceled request", null);

                    // Release the subscriber, we don't need a reference to it anymore
                    ReactiveStreamObserverPublisherClient.super.freeSubscriber();
                    callStreamObserver = null;
                }
            }
        };
    }

    // These methods are overridden to give more descriptive stack traces
    @Override
    public void subscribe(Subscriber subscriber) {
        super.subscribe(subscriber);
    }

    @Override
    public void onNext(T value) {
        super.onNext(value);
    }

    @Override
    public void onError(Throwable t) {
        super.onError(t);
    }

    @Override
    public void onCompleted() {
        super.onCompleted();
    }
}