com.salesforce.reactivegrpc.jmh.BenchmarkRxServerServiceImpl 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.jmh;

import java.util.Arrays;

import com.salesforce.reactivegrpc.jmh.proto.Messages;
import com.salesforce.reactivegrpc.jmh.proto.RxBenchmarkServiceGrpc;
import io.reactivex.Flowable;
import io.reactivex.Single;

/**
 * RxJava benchmarking service.
 */
public class BenchmarkRxServerServiceImpl extends RxBenchmarkServiceGrpc.BenchmarkServiceImplBase {

    private final Single   responseMono;
    private final Flowable responseFlux;

    public BenchmarkRxServerServiceImpl(int times) {
        Messages.SimpleResponse[] array = new Messages.SimpleResponse[times];
        Arrays.fill(array, Messages.SimpleResponse.getDefaultInstance());

        this.responseFlux = Flowable.fromArray(array);
        this.responseMono = Single.just(Messages.SimpleResponse.getDefaultInstance());
    }

    @Override
    public Single unaryCall(Single request) {
        return request.ignoreElement().andThen(responseMono);
    }

    @Override
    public Flowable streamingFromServer(Single request) {
        return request.ignoreElement().andThen(responseFlux);
    }

    @Override
    public Single streamingFromClient(Flowable request) {
        return request.ignoreElements().andThen(responseMono);
    }

    @Override
    public Flowable streamingBothWays(Flowable request) {
        request.subscribe();
        return responseFlux;
    }
}