Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Copyright 2017, Google LLC All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the following disclaimer
* in the documentation and/or other materials provided with the
* distribution.
* * Neither the name of Google LLC nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package nl.topicus.jdbc.shaded.com.google.api.gax.rpc.testing;
import nl.topicus.jdbc.shaded.com.google.api.core.ApiFuture;
import nl.topicus.jdbc.shaded.com.google.api.core.ApiFutures;
import nl.topicus.jdbc.shaded.com.google.api.core.InternalApi;
import nl.topicus.jdbc.shaded.com.google.api.gax.batching.PartitionKey;
import nl.topicus.jdbc.shaded.com.google.api.gax.batching.RequestBuilder;
import nl.topicus.jdbc.shaded.com.google.api.gax.rpc.ApiCallContext;
import nl.topicus.jdbc.shaded.com.google.api.gax.rpc.BatchedRequestIssuer;
import nl.topicus.jdbc.shaded.com.google.api.gax.rpc.BatchingDescriptor;
import nl.topicus.jdbc.shaded.com.google.api.gax.rpc.UnaryCallable;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.List;
@InternalApi("for testing")
public class FakeBatchableApi {
public static class LabeledIntList {
public String label;
public List ints;
public LabeledIntList(String label, Integer... numbers) {
this(label, new ArrayList<>(Arrays.asList(numbers)));
}
public LabeledIntList(String label, List ints) {
this.label = label;
this.ints = ints;
}
@Override
public boolean equals(Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
LabeledIntList that = (LabeledIntList) o;
if (!label.equals(that.label)) {
return false;
}
return ints.equals(that.ints);
}
@Override
public int hashCode() {
int result = label.hashCode();
result = 31 * result + ints.hashCode();
return result;
}
}
public static LabeledIntSquarerCallable callLabeledIntSquarer = new LabeledIntSquarerCallable();
public static class LabeledIntSquarerCallable
extends UnaryCallable> {
@Override
public ApiFuture> futureCall(LabeledIntList request, ApiCallContext context) {
List result = new ArrayList<>();
for (Integer i : request.ints) {
result.add(i * i);
}
return ApiFutures.immediateFuture(result);
}
}
public static SquarerBatchingDescriptor SQUARER_BATCHING_DESC = new SquarerBatchingDescriptor();
public static class SquarerBatchingDescriptor
implements BatchingDescriptor> {
@Override
public PartitionKey getBatchPartitionKey(LabeledIntList request) {
return new PartitionKey(request.label);
}
@Override
public RequestBuilder getRequestBuilder() {
return new RequestBuilder() {
LabeledIntList list;
@Override
public void appendRequest(LabeledIntList request) {
if (list == null) {
list = request;
} else {
list.ints.addAll(request.ints);
}
}
@Override
public LabeledIntList build() {
return list;
}
};
}
@Override
public void splitResponse(
List batchResponse,
Collection>> batch) {
int batchMessageIndex = 0;
for (BatchedRequestIssuer> responder : batch) {
List messageIds = new ArrayList<>();
long messageCount = responder.getMessageCount();
for (int i = 0; i < messageCount; i++) {
messageIds.add(batchResponse.get(batchMessageIndex));
batchMessageIndex += 1;
}
responder.setResponse(messageIds);
}
}
@Override
public void splitException(
Throwable throwable, Collection>> batch) {
for (BatchedRequestIssuer> responder : batch) {
responder.setException(throwable);
}
}
@Override
public long countElements(LabeledIntList request) {
return request.ints.size();
}
@Override
public long countBytes(LabeledIntList request) {
long counter = 0;
for (Integer i : request.ints) {
counter += i;
}
// Limit the byte size to simulate merged messages having smaller serialized size that the
// sum of their components
return Math.min(counter, 5);
}
}
}