org.esbtools.eventhandler.lightblue.client.BulkLightblueRequester Maven / Gradle / Ivy
/*
* Copyright 2016 esbtools Contributors and/or its affiliates.
*
* This file is part of esbtools.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see This class may be used across multiple threads safely, however the returned {@code Future}s
* are not threadsafe. That is, a given future instance should not be shared among multiple threads,
* which should not be a relevant limitation (if it is, however, it would not be hard to make them
* thread safe). Requests are queued up atomically, and performed and cleared atomically as well.
* That is, when one future is resolved, the current batch of requests is frozen, copied, cleared,
* and performed. A thread queueing a request while another thread resolves a future will
* not result in a loss of requests. It will either make it in for the batch, or be queued
* for the next.
*
*
While this class is thread safe, the logical "scope" of returned {@code Future}s is
* significant to consider. If you know you are going to batch up a bunch of requests, you don't
* want some other thread interrupting your batch performing your requests before you've finished
* queueing all of them up. So, you should create a new {@code BulkLightblueRequester} instance per
* logical "batch," and generally should avoid sharing an instance among multiple threads.
*/
public class BulkLightblueRequester implements LightblueRequester {
private final LightblueClient lightblue;
private final List> queuedRequests =
Collections.synchronizedList(new ArrayList<>());
private final List> queuedTryRequests =
Collections.synchronizedList(new ArrayList<>());
public BulkLightblueRequester(LightblueClient lightblue) {
this.lightblue = lightblue;
}
@Override
public TransformableFuture request(CRUDRequest... requests) {
checkNoNullsInRequests(requests);
LazyRequestTransformableFuture responseFuture =
new LazyRequestTransformableFuture<>(requests);
queuedRequests.add(responseFuture);
return responseFuture;
}
@Override
public TransformableFuture request(
Collection requests) {
return request(requests.toArray(new CRUDRequest[requests.size()]));
}
@Override
public TransformableFuture tryRequest(CRUDRequest... req) {
checkNoNullsInRequests(req);
LazyRequestTransformableFuture responseFuture =
new LazyRequestTransformableFuture<>(req);
queuedTryRequests.add(responseFuture);
return responseFuture;
}
private void doQueuedRequestsAndCompleteFutures() {
List> batch;
List> tryBatch;
synchronized (queuedRequests) {
batch = new ArrayList<>(queuedRequests);
queuedRequests.clear();
}
synchronized (queuedTryRequests) {
tryBatch = new ArrayList<>(queuedTryRequests);
queuedTryRequests.clear();
}
DataBulkRequest bulkRequest = new DataBulkRequest();
Stream.concat(batch.stream(), tryBatch.stream())
.flatMap(requestFuture -> Arrays.stream(requestFuture.requests))
// TODO: Determine if any requests are equivalent / duplicated and filter out
.forEach(bulkRequest::add);
try {
LightblueBulkDataResponse bulkResponse = tryBulkRequest(bulkRequest);
for (LazyRequestTransformableFuture batchedFuture : batch) {
CRUDRequest[] requests = batchedFuture.requests;
Map responseMap =
new HashMap<>(requests.length);
List errors = new ArrayList<>();
for (CRUDRequest request : requests) {
LightblueDataResponse response = bulkResponse.getResponse(request);
if (response instanceof LightblueErrorResponse) {
LightblueErrorResponse errorResponse = (LightblueErrorResponse) response;
DataError[] dataErrors = errorResponse.getDataErrors();
Error[] lightblueErrors = errorResponse.getLightblueErrors();
if (dataErrors != null) {
for (DataError dataError : dataErrors) {
errors.addAll(dataError.getErrors());
}
}
if (lightblueErrors != null) {
Collections.addAll(errors, lightblueErrors);
}
}
responseMap.put(request, response);
}
if (errors.isEmpty()) {
batchedFuture.complete(new BulkDataResponses(responseMap));
} else {
batchedFuture.completeExceptionally(new LightblueResponseException(errors));
}
}
for (LazyRequestTransformableFuture batchedFuture : tryBatch) {
CRUDRequest[] requests = batchedFuture.requests;
Map responseMap =
new HashMap<>(requests.length);
for (CRUDRequest request : requests) {
LightblueDataResponse response = bulkResponse.getResponse(request);
responseMap.put(request, LightblueResponse.fromClientResponse(response));
}
batchedFuture.complete(new BulkResponses(responseMap));
}
} catch (Exception e) {
Stream.concat(batch.stream(), tryBatch.stream())
.forEach(batchedFuture -> batchedFuture.completeExceptionally(e));
}
}
/**
* Swallows exceptions related to errors in individual requests on purpose. The returned
* bulk response object may have failed responses.
*
* @throws LightblueException if something else went wrong, in which case there is no usable
* response at all.
*/
private LightblueBulkDataResponse tryBulkRequest(DataBulkRequest bulkRequest)
throws LightblueException {
try {
return lightblue.bulkData(bulkRequest);
} catch (LightblueBulkResponseException e) {
return e.getBulkResponse();
}
}
private static void checkNoNullsInRequests(CRUDRequest[] requests) {
Objects.requireNonNull(requests, "requests");
for (int i = 0; i < requests.length; i++) {
Objects.requireNonNull(requests[i], "requests[" + i + "]");
}
}
/**
* We have two response maps: one with guaranteed successful responses (
* {@link BulkDataResponses} and one with no guarantees ({@link BulkResponses}). They are both
* backed by simple maps, just with different generic types, hence the base class.
*/
static abstract class ResponseMap implements Responses {
private final Map responseMap;
ResponseMap(Map responseMap) {
this.responseMap = responseMap;
}
@Override
public T forRequest(CRUDRequest request) {
if (responseMap.containsKey(request)) {
return responseMap.get(request);
}
throw new NoSuchElementException("No response for request: " + request);
}
}
static class BulkDataResponses extends ResponseMap implements LightblueDataResponses {
BulkDataResponses(Map responseMap) {
super(responseMap);
}
}
static class BulkResponses extends ResponseMap implements LightblueResponses {
BulkResponses(Map responseMap) {
super(responseMap);
}
}
/**
* A core future implementation which accepts its result externally, triggering this result
* lazily on the first call to {@link #get()}.
*
* This future is used to back more specific cases in {@link LazyRequestTransformableFuture}
* and {@link LazyTransformingFuture}.
*
*
{@link Future}s typically work asynchronously in another thread. This implementation
* intentionally does not. To reap the performance benefits of batching requests, we need a
* future design which puts off doing any work as long as possible: until something
* calls {@code .get()}. Spinning of work in another thread is the opposite: it starts work
* immediately in the background. To the caller, this future implementation feels similar:
* creating one returns immediately, and the caller can use it almost exactly as a "normal"
* asynchronous future or future. The only difference is that with this solution, callbacks
* won't happen until you ask for the result with {@code .get()}. With a normal thread-based,
* async future the callbacks happen at some point in the future regardless of if anything ever
* calls {@code .get()}.
*
* @param The type of the result of the future. See {@link TransformableFuture}.
*/
static class LazyTransformableFuture implements TransformableFuture {
/**
* A function which should trigger the completion of this future with a result. If it does
* not, this is a runtime error.
*
*
This is what makes the future lazy: we can use this to complete the future
* on demand.
*/
private final Completer completer;
private U result;
private Exception exception;
private boolean completed = false;
private boolean cancelled = false;
/**
* Queued up futures which are the result of applying this future's value to some transform
* function ({@link FutureTransform}). Futures are queued up by calling APIs like
* {@link #transformSync(FutureTransform)} and {@link #transformAsync(FutureTransform)}.
*/
private final List> next = new ArrayList<>(1);
private final List doneCallbacks = new ArrayList<>();
private static Logger log = LoggerFactory.getLogger(LazyTransformableFuture.class);
/**
* @param completer Reference to a function which should complete this future when called.
* See {@link #completer}.
*/
LazyTransformableFuture(Completer completer) {
this.completer = completer;
}
void complete(U responses) {
if (isDone()) return;
result = responses;
completed = true;
callDoneCallbacks();
// It's important that these transforms are completed eagerly, since this may queue up
// additional requests. See LazyTransformingFuture#complete.
for (LazyTransformingFuture next : this.next) {
try {
next.complete(result);
} catch (Exception e) {
log.warn("Exception caught and ignored while completing next transforming " +
"future.", e);
}
}
}
void completeExceptionally(Exception exception) {
if (isDone()) return;
completed = true;
this.exception = exception;
callDoneCallbacks();
for (LazyTransformingFuture next : this.next) {
try {
next.completeExceptionally(exception);
} catch (Exception e) {
log.warn("Exception caught and ignored while completing next transforming " +
"future with exception.", e);
}
}
}
@Override
public boolean cancel(boolean mayInterruptIfRunning) {
if (completed) return false;
cancelled = true;
callDoneCallbacks();
return true;
}
@Override
public boolean isCancelled() {
return cancelled;
}
@Override
public boolean isDone() {
return cancelled || completed;
}
private static class TimeoutDuration {
final long duration;
final TimeUnit timeUnit;
TimeoutDuration(long duration, TimeUnit timeUnit) {
this.duration = duration;
this.timeUnit = timeUnit;
}
}
@Override
public U get() throws InterruptedException, ExecutionException {
try {
return get(Optional.empty());
} catch (TimeoutException e) {
throw new ExecutionException(
"A future without a timeout is attempting to throw a TimeoutException while it should not. "
+ "This implies there is a bug in the future code and will need to be corrected ",
e);
}
}
@Override
public U get(long timeout, TimeUnit unit) throws InterruptedException, ExecutionException, TimeoutException {
return get(Optional.of(new TimeoutDuration(timeout, unit)));
}
private U get(Optional timeout)
throws TimeoutException, InterruptedException, ExecutionException {
if (cancelled) {
throw new CancellationException();
}
if (!completed) {
if (timeout.isPresent()) {
ExecutorService executor = Executors.newSingleThreadExecutor();
try {
Future submittedTask = executor.submit(() -> {
completer.triggerFutureCompletion();
});
TimeoutDuration timeoutDuration = timeout.get();
submittedTask.get(timeoutDuration.duration, timeoutDuration.timeUnit);
} finally {
executor.shutdownNow();
}
} else {
completer.triggerFutureCompletion();
}
if (!completed) {
throw new ExecutionException(new IllegalStateException("Future attempted to "
+ "lazily trigger completion, but completer did not actually complete "
+ "the future. Check the provided completer function for correctness."));
}
}
if (exception != null) {
throw new ExecutionException(exception);
}
return result;
}
@Override
public TransformableFuture transformSync(FutureTransform futureTransform) {
LazyTransformingFuture future =
new LazyTransformingFuture<>(futureTransform, completer);
next.add(future);
return future;
}
@Override
public TransformableFuture transformAsync(
FutureTransform> futureTransform) {
LazyTransformingFuture> future =
new LazyTransformingFuture<>(futureTransform, completer);
next.add(future);
return new NestedTransformableFuture<>(future);
}
@Override
public TransformableFuture transformAsyncIgnoringReturn(
FutureTransform> futureTransform) {
LazyTransformingFuture> future =
new LazyTransformingFuture<>(futureTransform, completer);
next.add(future);
return new NestedTransformableFutureIgnoringReturn(future);
}
@Override
public TransformableFuture whenDoneOrCancelled(FutureDoneCallback callback) {
if (isDone()) {
try {
callback.onDoneOrCancelled();
} catch (Exception e) {
log.warn("Exception caught and ignored while running future done callback.", e);
}
return this;
}
doneCallbacks.add(callback);
return this;
}
private void callDoneCallbacks() {
for (FutureDoneCallback doneCallback : doneCallbacks) {
try {
doneCallback.onDoneOrCancelled();
} catch (Exception e) {
log.warn("Exception caught and ignored while running future done callback.", e);
}
}
}
}
/**
* Wraps a {@link LazyTransformableFuture} and some {@link AbstractLightblueDataRequest
* lightblue requests} which are used to complete this in
* {@link #doQueuedRequestsAndCompleteFutures()}. Naturally, then, that function is used as the
* lazy future's completer function. That function and this implementation are tightly coupled.
*/
class LazyRequestTransformableFuture implements TransformableFuture {
private final LazyTransformableFuture backingFuture =
new LazyTransformableFuture<>(() -> doQueuedRequestsAndCompleteFutures());
final CRUDRequest[] requests;
LazyRequestTransformableFuture(CRUDRequest[] requests) {
this.requests = requests;
}
void complete(T responses) {
backingFuture.complete(responses);
}
void completeExceptionally(Exception exception) {
backingFuture.completeExceptionally(exception);
}
@Override
public TransformableFuture transformSync(
FutureTransform futureTransform) {
return backingFuture.transformSync(futureTransform);
}
@Override
public TransformableFuture transformAsync(
FutureTransform> futureTransform) {
return backingFuture.transformAsync(futureTransform);
}
@Override
public TransformableFuture transformAsyncIgnoringReturn(
FutureTransform> futureTransform) {
return backingFuture.transformAsyncIgnoringReturn(futureTransform);
}
@Override
public TransformableFuture whenDoneOrCancelled(FutureDoneCallback callback) {
backingFuture.whenDoneOrCancelled(callback);
return this;
}
@Override
public boolean cancel(boolean mayInterruptIfRunning) {
if (!backingFuture.isDone()) {
queuedRequests.remove(this);
queuedTryRequests.remove(this);
}
return backingFuture.cancel(mayInterruptIfRunning);
}
@Override
public boolean isCancelled() {
return backingFuture.isCancelled();
}
@Override
public boolean isDone() {
return backingFuture.isDone();
}
@Override
public T get() throws InterruptedException, ExecutionException {
return backingFuture.get();
}
@Override
public T get(long timeout, TimeUnit unit) throws InterruptedException,
ExecutionException, TimeoutException {
return backingFuture.get(timeout, unit);
}
}
/**
* A simple lazy future implementation which applies some transformation function to the value
* it is completed with.
*
* This is used in {@link #transformSync(FutureTransform)} callbacks of
* {@link TransformableFuture} implementations.
*
* @param The input type used to complete the Future.
* @param The output type as a result of applying the transform function to the input.
*/
static class LazyTransformingFuture implements TransformableFuture {
private final FutureTransform transform;
private final LazyTransformableFuture backingFuture;
LazyTransformingFuture(FutureTransform transform, Completer completer) {
this.transform = transform;
this.backingFuture = new LazyTransformableFuture<>(completer);
}
/**
* Completes, applying the transform function immediately.
*
* Running the transform function immediately is crucial. If we waited until .get() to
* run the transform, then the aggregate effect is that transforms that queue up additional
* requests are run one at a time, just before performing those requests, because .get()
* needs to return with some results to the caller and so it must call its completion
* function (which makes all queued requests). We miss out on batching all requests from
* peer transforms. By applying transforms as soon as we have input for them, we queue up
* any new requests right away so when the next completer is called, all of the next lazy
* futures' requests are already queued up.
*/
void complete(T responses) {
if (isDone()) return;
try {
U transformed = transform.transform(responses);
backingFuture.complete(transformed);
} catch (Exception e) {
completeExceptionally(e);
}
}
void completeExceptionally(Exception exception) {
backingFuture.completeExceptionally(exception);
}
@Override
public boolean cancel(boolean mayInterruptIfRunning) {
return backingFuture.cancel(mayInterruptIfRunning);
}
@Override
public boolean isCancelled() {
return backingFuture.isCancelled();
}
@Override
public boolean isDone() {
return backingFuture.isDone();
}
@Override
public U get() throws InterruptedException, ExecutionException {
return backingFuture.get();
}
/**
* This future is not completed asynchronously so there is no way to "time out" unless we
* introduce another thread for processing requests which I don't think is hugely necessary.
*/
@Override
public U get(long timeout, TimeUnit unit) throws InterruptedException, ExecutionException,
TimeoutException {
return backingFuture.get(timeout, unit);
}
@Override
public TransformableFuture transformSync(FutureTransform futureTransform) {
return backingFuture.transformSync(futureTransform);
}
@Override
public TransformableFuture transformAsync(
FutureTransform> futureTransform) {
return backingFuture.transformAsync(futureTransform);
}
@Override
public TransformableFuture transformAsyncIgnoringReturn(
FutureTransform> futureTransform) {
return backingFuture.transformAsyncIgnoringReturn(futureTransform);
}
@Override
public TransformableFuture whenDoneOrCancelled(FutureDoneCallback callback) {
backingFuture.whenDoneOrCancelled(callback);
return this;
}
}
interface Completer {
void triggerFutureCompletion();
}
}