PARQUET-1414: Simplify next row count check calculation (#537)

apache · Nov 8, 2018 · 7f561b6 · 7f561b6
1 parent ca294f9
commit 7f561b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/parquet-column/src/main/java/org/apache/parquet/column/impl/ColumnWriteStoreBase.java b/parquet-column/src/main/java/org/apache/parquet/column/impl/ColumnWriteStoreBase.java
@@ -200,7 +200,7 @@ private void sizeCheck() {
         writer.writePage();
         remainingMem = props.getPageSizeThreshold();
       } else {
-        rowCountForNextRowCountCheck = min(rowCountForNextRowCountCheck, rowCount + (pageRowCountLimit - rows));
+        rowCountForNextRowCountCheck = min(rowCountForNextRowCountCheck, writer.getRowsWrittenSoFar() + pageRowCountLimit);
       }
       long rowsToFillPage =
           usedMem == 0 ?