GH-36828: [C++][Parquet] Make buffered RowGroupSerializer using BufferedPageWriter (#36829)

mapleFU · web-flow · commit 78bbe7676c14 · 2023-07-24T17:05:51.000+08:00
### Rationale for this change See #36828 ### What changes are included in this PR? Add `buffered` argument when building `PageWriter` ### Are these changes tested? no ### Are there any user-facing changes? no * Closes: #36828 Authored-by: mwish <maplewish117@gmail.com> Signed-off-by: Gang Wu <ustcwg@gmail.com>
diff --git a/cpp/src/parquet/file_writer.cc b/cpp/src/parquet/file_writer.cc
@@ -310,17 +310,17 @@ class RowGroupSerializer : public RowGroupWriter::Contents {
 
       std::unique_ptr<PageWriter> pager;
       if (!codec_options) {
-        pager = PageWriter::Open(sink_, properties_->compression(path), col_meta,
-                                 row_group_ordinal_, static_cast<int16_t>(column_ordinal),
-                                 properties_->memory_pool(), false, meta_encryptor,
-                                 data_encryptor, properties_->page_checksum_enabled(),
-                                 ci_builder, oi_builder, CodecOptions());
+        pager = PageWriter::Open(
+            sink_, properties_->compression(path), col_meta, row_group_ordinal_,
+            static_cast<int16_t>(column_ordinal), properties_->memory_pool(),
+            buffered_row_group_, meta_encryptor, data_encryptor,
+            properties_->page_checksum_enabled(), ci_builder, oi_builder, CodecOptions());
       } else {
-        pager = PageWriter::Open(sink_, properties_->compression(path), col_meta,
-                                 row_group_ordinal_, static_cast<int16_t>(column_ordinal),
-                                 properties_->memory_pool(), false, meta_encryptor,
-                                 data_encryptor, properties_->page_checksum_enabled(),
-                                 ci_builder, oi_builder, *codec_options);
+        pager = PageWriter::Open(
+            sink_, properties_->compression(path), col_meta, row_group_ordinal_,
+            static_cast<int16_t>(column_ordinal), properties_->memory_pool(),
+            buffered_row_group_, meta_encryptor, data_encryptor,
+            properties_->page_checksum_enabled(), ci_builder, oi_builder, *codec_options);
       }
       column_writers_.push_back(
           ColumnWriter::Make(col_meta, std::move(pager), properties_));
