From 619f5010549220c98cb6849ebcfb3592a811c679 Mon Sep 17 00:00:00 2001
From: lamber-ken <lamberken@163.com>
Date: Tue, 31 Dec 2019 13:59:26 +0800
Subject: [PATCH] Clean up code

---
 .../apache/hudi/cli/commands/HoodieLogFileCommand.java | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/hudi-cli/src/main/java/org/apache/hudi/cli/commands/HoodieLogFileCommand.java b/hudi-cli/src/main/java/org/apache/hudi/cli/commands/HoodieLogFileCommand.java
index dc0f43e69..f9bea4785 100644
--- a/hudi-cli/src/main/java/org/apache/hudi/cli/commands/HoodieLogFileCommand.java
+++ b/hudi-cli/src/main/java/org/apache/hudi/cli/commands/HoodieLogFileCommand.java
@@ -198,10 +198,9 @@ public class HoodieLogFileCommand implements CommandMarker {
               HoodieMemoryConfig.DEFAULT_SPILLABLE_MAP_BASE_PATH);
       for (HoodieRecord<? extends HoodieRecordPayload> hoodieRecord : scanner) {
         Option<IndexedRecord> record = hoodieRecord.getData().getInsertValue(readerSchema);
-        if (allRecords.size() >= limit) {
-          break;
+        if (allRecords.size() < limit) {
+          allRecords.add(record.get());
         }
-        allRecords.add(record.get());
       }
     } else {
       for (String logFile : logFilePaths) {
@@ -216,10 +215,9 @@ public class HoodieLogFileCommand implements CommandMarker {
             HoodieAvroDataBlock blk = (HoodieAvroDataBlock) n;
             List<IndexedRecord> records = blk.getRecords();
             for (IndexedRecord record : records) {
-              if (allRecords.size() >= limit) {
-                break;
+              if (allRecords.size() < limit) {
+                allRecords.add(record);
               }
-              allRecords.add(record);
             }
           }
         }