com.salesforce.reactorgrpc.stub.BackpressureChunkingLifter 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.reactorgrpc.stub;

import com.salesforce.reactivegrpc.common.ReactiveBackpressureChunker;
import org.reactivestreams.Subscriber;
import org.reactivestreams.Subscription;
import reactor.core.CoreSubscriber;
import reactor.core.Scannable;
import reactor.util.context.Context;

import java.util.function.BiFunction;

/**
 * Implements {@link ReactiveBackpressureChunker} as a liftable operator for {@link reactor.core.publisher.Operators#lift(BiFunction)}.
 *
 * @param 
 */
public class BackpressureChunkingLifter extends ReactiveBackpressureChunker implements BiFunction, CoreSubscriber> {

    public BackpressureChunkingLifter() {
        super(ReactiveBackpressureChunker.DEFAULT_CHUNK_SIZE);
    }

    @Override
    public CoreSubscriber apply(Scannable scannable, CoreSubscriber downstream) {
        Subscriber delegate = apply(downstream);

        return new CoreSubscriber() {
            @Override
            public Context currentContext() {
                return downstream.currentContext();
            }

            @Override
            public void onSubscribe(Subscription subscription) {
                delegate.onSubscribe(subscription);
            }

            @Override
            public void onNext(T t) {
                delegate.onNext(t);
            }

            @Override
            public void onError(Throwable throwable) {
                delegate.onError(throwable);
            }

            @Override
            public void onComplete() {
                delegate.onComplete();
            }
        };
    }
}