From 807027fb4e58ec205ee56f46587922e72237b06c Mon Sep 17 00:00:00 2001
From: Zihan Li <zihli@zihli-mn2.linkedin.biz>
Date: Tue, 14 Nov 2023 16:51:54 -0800
Subject: [PATCH] small refractor

---
 .../extractor/extract/kafka/KafkaStreamingExtractor.java       | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/gobblin-modules/gobblin-kafka-common/src/main/java/org/apache/gobblin/source/extractor/extract/kafka/KafkaStreamingExtractor.java b/gobblin-modules/gobblin-kafka-common/src/main/java/org/apache/gobblin/source/extractor/extract/kafka/KafkaStreamingExtractor.java
index 911dd3acac3..8320a82ec06 100644
--- a/gobblin-modules/gobblin-kafka-common/src/main/java/org/apache/gobblin/source/extractor/extract/kafka/KafkaStreamingExtractor.java
+++ b/gobblin-modules/gobblin-kafka-common/src/main/java/org/apache/gobblin/source/extractor/extract/kafka/KafkaStreamingExtractor.java
@@ -91,7 +91,6 @@ public class KafkaStreamingExtractor<S> extends FlushingExtractor<S, DecodeableK
   public static final Long DEFAULT_MAX_KAFKA_BUFFER_SIZE_IN_BYTES = Long.valueOf(50 * 1024 * 1024);
   // Max number of records to be pulled in single polling.
   private static final String KAFKA_MAX_POLL_RECORDS_KEY = "kafka.consumer.maxPollRecords";
-  // Changed from Integer.MAX_VALUE to 1 to workaround and issue with lack of error handling support in LiKafka10
   private static final int DEFAULT_MAX_POLL_RECORDS = 100;
   private static final Long MAX_LOG_ERRORS = 100L;
 
@@ -232,7 +231,7 @@ public KafkaStreamingExtractor(WorkUnitState state) {
     } else {
       // As there is no avg record size available, using lower number to make sure we don't hit OOM issue
       state.setProp(KAFKA_MAX_POLL_RECORDS_KEY, DEFAULT_MAX_POLL_RECORDS);
-      log.info("set max.poll.records to be 100");
+      log.info("set max.poll.records to be {}", DEFAULT_MAX_POLL_RECORDS);
     }
     this.kafkaConsumerClientResolver =
         new ClassAliasResolver<>(GobblinKafkaConsumerClient.GobblinKafkaConsumerClientFactory.class);