diff --git a/amazon-kinesis-client/src/main/java/software/amazon/kinesis/retrieval/polling/PollingConfig.java b/amazon-kinesis-client/src/main/java/software/amazon/kinesis/retrieval/polling/PollingConfig.java index 3359c570..cbbf8d11 100644 --- a/amazon-kinesis-client/src/main/java/software/amazon/kinesis/retrieval/polling/PollingConfig.java +++ b/amazon-kinesis-client/src/main/java/software/amazon/kinesis/retrieval/polling/PollingConfig.java @@ -146,12 +146,13 @@ public class PollingConfig implements RetrievalSpecificConfig { 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; } /**