Update PollingConfig maxRecords method to return PollingConfig (#1275)

awslabs · Mar 15, 2024 · a95aa9e · a95aa9e
1 parent ddff314
commit a95aa9e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...kinesis-client/src/main/java/software/amazon/kinesis/retrieval/polling/PollingConfig.java b/...kinesis-client/src/main/java/software/amazon/kinesis/retrieval/polling/PollingConfig.java
@@ -146,12 +146,13 @@ public PollingConfig idleTimeBetweenReadsInMillis(long idleTimeBetweenReadsInMil
         return this;
     }
 
-    public void maxRecords(int maxRecords) {
+    public PollingConfig maxRecords(int maxRecords) {
         if (maxRecords > DEFAULT_MAX_RECORDS) {
             throw new IllegalArgumentException(
                     "maxRecords must be less than or equal to " +  DEFAULT_MAX_RECORDS + " but current value is " + maxRecords());
         }
         this.maxRecords = maxRecords;
+        return this;
     }
 
     /**