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

Go to download
/*
 *  Copyright (c) 2019, 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 java.util.Queue;

import io.grpc.stub.CallStreamObserver;
import io.grpc.stub.ClientCallStreamObserver;
import io.grpc.stub.ClientResponseObserver;

/**
 * The gRPC client-side implementation of
 * {@link AbstractStreamObserverAndPublisher}.
 *
 * @param  T
 */
public abstract class AbstractClientStreamObserverAndPublisher
        extends AbstractStreamObserverAndPublisher
        implements ClientResponseObserver {

    public AbstractClientStreamObserverAndPublisher(
            Queue queue,
            Consumer> onSubscribe) {
        super(queue, onSubscribe);
    }

    public AbstractClientStreamObserverAndPublisher(
            Queue queue,
            Consumer> onSubscribe,
            Runnable onTerminate) {
        super(queue, onSubscribe, onTerminate);
    }

    public AbstractClientStreamObserverAndPublisher(
            Queue queue,
            Consumer> onSubscribe,
            Runnable onTerminate,
            int prefetch,
            int lowTide) {
        super(queue, prefetch, lowTide, onSubscribe, onTerminate);
    }

    @Override
    public void beforeStart(ClientCallStreamObserver requestStream) {
        super.onSubscribe(requestStream);
    }

    @Override
    protected void doOnCancel() {
        if (subscription != null) {
            ((ClientCallStreamObserver) subscription).cancel("Client canceled request", null);
        }
    }
}