1
0

Merge pull request #1143 from BigDataArtisans/outoflimit

[MINOR] Fix out of limits for results
This commit is contained in:
Suneel Marthi
2019-12-31 02:08:54 -05:00
committed by GitHub

View File

@@ -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) {
@@ -215,9 +214,10 @@ public class HoodieLogFileCommand implements CommandMarker {
if (n instanceof HoodieAvroDataBlock) {
HoodieAvroDataBlock blk = (HoodieAvroDataBlock) n;
List<IndexedRecord> records = blk.getRecords();
allRecords.addAll(records);
if (allRecords.size() >= limit) {
break;
for (IndexedRecord record : records) {
if (allRecords.size() < limit) {
allRecords.add(record);
}
}
}
}