From 5ca680008592b0c752e8cf57d6ff96bf3090660d Mon Sep 17 00:00:00 2001
From: eha sah <ehasah@amazon.com>
Date: Tue, 11 Mar 2025 13:13:44 -0700
Subject: [PATCH] Fall back to previously calculated totalScan

---
 .../kinesis/leases/dynamodb/DynamoDBLeaseRefresher.java   | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/amazon-kinesis-client/src/main/java/software/amazon/kinesis/leases/dynamodb/DynamoDBLeaseRefresher.java b/amazon-kinesis-client/src/main/java/software/amazon/kinesis/leases/dynamodb/DynamoDBLeaseRefresher.java
index 06808d25..0a6a2ad1 100644
--- a/amazon-kinesis-client/src/main/java/software/amazon/kinesis/leases/dynamodb/DynamoDBLeaseRefresher.java
+++ b/amazon-kinesis-client/src/main/java/software/amazon/kinesis/leases/dynamodb/DynamoDBLeaseRefresher.java
@@ -622,11 +622,9 @@ public class DynamoDBLeaseRefresher implements LeaseRefresher {
         if (isTotalSegmentsCacheValid()) {
             return cachedTotalSegments;
         }
-        int parallelScanTotalSegments = DEFAULT_LEASE_TABLE_SCAN_PARALLELISM_FACTOR;
-        if (isTotalSegmentsCacheValid()) {
-            return cachedTotalSegments;
-        }
-        DescribeTableResponse describeTableResponse = describeLeaseTable();
+
+        int parallelScanTotalSegments = cachedTotalSegments == null ? DEFAULT_LEASE_TABLE_SCAN_PARALLELISM_FACTOR  : cachedTotalSegments;
+        final DescribeTableResponse describeTableResponse = describeLeaseTable();
         if (describeTableResponse == null) {
             log.info("DescribeTable returned null so using default totalSegments : {}", parallelScanTotalSegments);
         } else {