From 65dfb759f460c97f8e7f0ea245e7c9ba1d51844e Mon Sep 17 00:00:00 2001 From: Joshua Kim Date: Thu, 18 Feb 2021 18:20:41 -0800 Subject: [PATCH] Adding some logging for listShards. --- .../amazon/kinesis/leases/KinesisShardDetector.java | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/amazon-kinesis-client/src/main/java/software/amazon/kinesis/leases/KinesisShardDetector.java b/amazon-kinesis-client/src/main/java/software/amazon/kinesis/leases/KinesisShardDetector.java index 8baf61d2..189ba18b 100644 --- a/amazon-kinesis-client/src/main/java/software/amazon/kinesis/leases/KinesisShardDetector.java +++ b/amazon-kinesis-client/src/main/java/software/amazon/kinesis/leases/KinesisShardDetector.java @@ -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) {