From 66fac819e4f242061758321565c684c9240c4fc0 Mon Sep 17 00:00:00 2001
From: Gabor Szadovszky <gabor@apache.org>
Date: Thu, 8 Nov 2018 15:08:19 +0100
Subject: [PATCH] PARQUET-1414: Simplify next row count check calculation

---
 .../org/apache/parquet/column/impl/ColumnWriteStoreBase.java    | 2 +-
 1 file changed, 1 insertion(+), 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
index f79c09de2f..ac9aaca26a 100644
--- 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 ?