From 10dc26d91930727ee66a707c98395fed437fa636 Mon Sep 17 00:00:00 2001
From: Bingyi Sun <sunbingyi1992@gmail.com>
Date: Wed, 26 Jul 2023 14:12:56 +0800
Subject: [PATCH] Fix writer count error (#28)

Signed-off-by: sunby <sunbingyi1992@gmail.com>
---
 cpp/src/format/parquet/file_writer.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cpp/src/format/parquet/file_writer.cpp b/cpp/src/format/parquet/file_writer.cpp
index 010d30b6..721c82ee 100644
--- a/cpp/src/format/parquet/file_writer.cpp
+++ b/cpp/src/format/parquet/file_writer.cpp
@@ -19,7 +19,7 @@ Status ParquetFileWriter::Init() {
 
 Status ParquetFileWriter::Write(arrow::RecordBatch* record) {
   RETURN_ARROW_NOT_OK(writer_->WriteRecordBatch(*record));
-  count_ += record->num_columns();
+  count_ += record->num_rows();
   return Status::OK();
 }