luciano/amazon-kinesis-client
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Adding some logging for listShards.

Browse source
This commit is contained in:
Joshua Kim 2021-02-18 18:20:41 -08:00
parent efc57af43f
commit 65dfb759f4
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
amazon-kinesis-client/src/main/java/software/amazon/kinesis/leases/KinesisShardDetector.java
View file

@ -192,12 +192,16 @@ public class KinesisShardDetector implements ShardDetector {
exceptionManager.add(ResourceInUseException.class, t -> t);
exceptionManager.add(KinesisException.class, t -> t);
ListShardsRequest.Builder request = KinesisRequestsBuilder.listShardsRequestBuilder();
ListShardsRequest.Builder builder = KinesisRequestsBuilder.listShardsRequestBuilder();
if (StringUtils.isEmpty(nextToken)) {
request = request.streamName(streamIdentifier.streamName()).shardFilter(shardFilter);
builder = builder.streamName(streamIdentifier.streamName()).shardFilter(shardFilter);
} else {
request = request.nextToken(nextToken);
builder = builder.nextToken(nextToken);
}
final ListShardsRequest request = builder.build();
log.info("Stream {}: listing shards with list shards request {}", streamIdentifier, request);
ListShardsResponse result = null;
LimitExceededException lastException = null;
int remainingRetries = maxListShardsRetryAttempts;
@ -205,7 +209,7 @@ public class KinesisShardDetector implements ShardDetector {
while (result == null) {
try {
try {
result = getListShardsResponse(request.build());
result = getListShardsResponse(request);
} catch (ExecutionException e) {
throw exceptionManager.apply(e.getCause());
} catch (InterruptedException e) {