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 2016 Sony Mobile Communications, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License").
* You may not use this file except in compliance with the License.
* A copy of the License is located at
*
* http://aws.amazon.com/apache2.0
*
* or in the "license" file accompanying this file. This file is distributed
* on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
* express or implied. See the License for the specific language governing
* permissions and limitations under the License.
*/
package lumbermill.internal.aws;
import com.amazonaws.handlers.AsyncHandler;
import com.amazonaws.services.kinesis.AmazonKinesisAsync;
import com.amazonaws.services.kinesis.model.PutRecordsRequest;
import com.amazonaws.services.kinesis.model.PutRecordsRequestEntry;
import com.amazonaws.services.kinesis.model.PutRecordsResult;
import com.amazonaws.services.kinesis.model.PutRecordsResultEntry;
import lumbermill.api.Event;
import lumbermill.api.Observables;
import lumbermill.api.Timer;
import lumbermill.aws.FatalAWSException;
import lumbermill.internal.StringTemplate;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import rx.Observable;
import rx.subjects.PublishSubject;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import java.util.UUID;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.function.Supplier;
import static java.util.stream.Collectors.toList;
/**
* Simple wrapper around AmazonKinesisClient that converts Events to Records.
* It will do its best to retry if there are failed records.
*
* TODO: Handle too large items, today a large item will prevent the others from being processed.
*/
public class SimpleRetryableKinesisClient {
private static final Logger LOGGER = LoggerFactory.getLogger(SimpleRetryableKinesisClient.class);
public static final int DEFAULT_ATTEMPTS = 20;
private final AmazonKinesisAsync amazonKinesisClient;
private final String stream;
private final Optional> partitionKeySupplier;
private Timer.Factory timerFactory = Observables.fixedTimer(500);
private int maxAttempts = DEFAULT_ATTEMPTS;
SimpleRetryableKinesisClient(AmazonKinesisAsync amazonKinesisClient, String stream, Optional partitionKey) {
this.amazonKinesisClient = amazonKinesisClient;
this.stream = stream;
if (partitionKey.isPresent()) {
this.partitionKeySupplier = Optional.of(new Supplier() {
final StringTemplate partitionKeyTemplate = StringTemplate.compile(partitionKey.get());
@Override
public StringTemplate get() {
return partitionKeyTemplate;
}
});
} else {
this.partitionKeySupplier = Optional.empty();
}
}
/**
* Puts a single record to kinesis. It is recommended to always buffer into multiple
* events and do putRecords instead.
*/
public Observable putRecord(T event) {
amazonKinesisClient.putRecord(stream, event.raw().asByteBuffer(),
partitionKeySupplier.isPresent()
? partitionKeySupplier.get().get().format(event).get()
: UUID.randomUUID().toString());
return Observable.just(event);
}
/**
* Asynchronously puts records to kinesis.
*
* @param events - List of events to send
* @return - Observable with same list as parameter
*/
public Observable> putRecords(List events) {
if (LOGGER.isDebugEnabled()) {
LOGGER.debug("putRecords() with {} events", events.size());
}
RequestContext request = new RequestContext(events, new PutRecordsRequest()
.withRecords(events.stream()
.map(this::toRecordEntries)
.collect(toList()))
.withStreamName(stream));
putRecordsAsync(request);
return request.subject;
}
/**
* Recursively retry until there are no more failed records in response or to many retries
*/
private void putRecordsAsync(final RequestContext request) {
amazonKinesisClient.putRecordsAsync(request.putRecordsRequest, new AsyncHandler() {
@Override
public void onError(Exception exception) {
try {
Observable> observable = request.nextAttempt();
observable
.doOnNext(requestContext -> LOGGER.warn("About to retry request from exception: {}", exception.getMessage()))
.doOnNext(context -> {
if (context.isPresent()) {
putRecordsAsync(context.get());
} else {
request.error(new FatalAWSException("Too many kinesis retries, root cause:", exception));
}
})
.doOnError(throwable -> request.error(throwable))
.subscribe();
} catch (Throwable t) {
LOGGER.error("Unexpected exception in onError()", t);
request.error(t);
}
}
@Override
public void onSuccess(PutRecordsRequest putRecordsRequest, PutRecordsResult putRecordsResult) {
// Surround with try/catch to prevent any unexpected exceptions from beeing swallowed
try {
if (putRecordsResult.getFailedRecordCount() > 0) {
LOGGER.debug("Got {} failed records, retrying (attempts = {})",
putRecordsResult.getFailedRecordCount(), request.attempt);
// Try again with failing records,
//Optional nextAttempt = request.nextAttempt(putRecordsResult);
Observable> observable = request.nextAttempt(putRecordsResult);
observable.doOnNext(context -> {
if (context.isPresent()) {
putRecordsAsync(context.get());
} else {
request.error(new FatalAWSException("Too many kinesis retries"));
}
})
.doOnError(throwable -> request.error(throwable))
.subscribe();
} else {
request.done();
}
} catch (Throwable t) {
LOGGER.error("Unexpected exception in onSuccess()", t);
request.error(t);
}
}
});
}
/**
* Converts event to actual kinesis entry type
*/
private PutRecordsRequestEntry toRecordEntries(T event) {
//Optional partitionKey = partitionKeyTemplate.format(event);
return new PutRecordsRequestEntry().withData (
event.raw().asByteBuffer())
// FIXME: If partitionkey does not return a value, what approach is best?
.withPartitionKey(partitionKeySupplier.isPresent()
? partitionKeySupplier.get().get().format(event).get()
: UUID.randomUUID().toString());
}
public SimpleRetryableKinesisClient withRetryTimer(Timer.Factory timer, int attempts) {
this.timerFactory = timer;
this.maxAttempts = attempts;
return this;
}
/**
* Contains state in order to track retries as well as returning response to pipeline.
*/
private class RequestContext {
public final PublishSubject> subject = PublishSubject.create();
/**
* Keep them here until we are done, then return them
*/
public final List events;
/**
* Request to execute
*/
public PutRecordsRequest putRecordsRequest;
/**
* Attempt count
*/
public AtomicInteger attempt = new AtomicInteger(1);
private Timer timer = SimpleRetryableKinesisClient.this.timerFactory.create();
public RequestContext(List events, PutRecordsRequest putRecordsRequest) {
this.events = events;
this.putRecordsRequest = putRecordsRequest;
}
private boolean hasNextAttempt() {
return attempt.get() > SimpleRetryableKinesisClient.this.maxAttempts ? false : true;
}
/**
* Next attempt based on the same request as previous request, use when exception occured
*
* @return RequestContext IF there are more attempts, Optional.empty() otherwise
*/
public Observable> nextAttempt() {
this.attempt.incrementAndGet();
if (!hasNextAttempt()) {
return Observable.just(Optional.empty());
}
return Observables.just(Optional.of(this)).withDelay(timer);
}
/**
* Based on non successful records, returns a RequestContext with a correct PutRecordsRequest
*
* @param result is the last result returned from Kinesis
* @return RequestContext IF there are more attempts, Optional.empty() otherwise
*/
public Observable> nextAttempt(PutRecordsResult result) {
this.attempt.incrementAndGet();
if (!hasNextAttempt()) {
return Observable.just(Optional.empty());
}
this.putRecordsRequest = failedRecords(result);
return Observables.just(Optional.of(this)).withDelay(timer);
}
/**
* Based on the request and the result, returns a new request
* containing the records that failed.
* @param result is the last PutRecordsResult
* @return a new PutRecordsRequest with failing records
*/
private PutRecordsRequest failedRecords(PutRecordsResult result) {
List newRecords = new ArrayList<>();
List records = result.getRecords();
for (int i = 0; i < records.size(); i++) {
if (records.get(i).getErrorCode() != null) {
newRecords.add(putRecordsRequest.getRecords().get(i));
}
}
return new PutRecordsRequest()
.withRecords(newRecords)
.withStreamName(putRecordsRequest.getStreamName());
}
/**
* Invoked after request is successful and there are no failing records.
*/
public void done() {
if (LOGGER.isTraceEnabled()) {
LOGGER.trace("Done() mem free {}, mem max {}", Runtime.getRuntime().freeMemory(), Runtime.getRuntime().maxMemory());
}
this.subject.onNext(events);
this.subject.onCompleted();
}
/**
* Invoked if an error occurs of there are no more retries
* @param t is the original exception
*/
public void error(Throwable t) {
this.subject.onError(t);
}
}
}
