Fix benchmark

apache · Jan 8, 2025 · 1b82cd3 · 1b82cd3
1 parent 313ce5e
commit 1b82cd3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/cpp/src/parquet/arrow/size_stats_benchmark.cc b/cpp/src/parquet/arrow/size_stats_benchmark.cc
@@ -95,13 +95,15 @@ void WriteColumn(::benchmark::State& state, const std::shared_ptr<::arrow::Table
         std::static_pointer_cast<::arrow::io::OutputStream>(output),
         DEFAULT_MAX_ROW_GROUP_LENGTH, properties));
 
-    state.PauseTiming();
-
-    auto metadata = parquet::ReadMetaData(
-        std::make_shared<::arrow::io::BufferReader>(output->Finish().ValueOrDie()));
-    state.counters["output_size"] = static_cast<double>(output->Tell().ValueOrDie());
-    state.counters["page_index_size"] =
-        static_cast<double>(GetTotalPageIndexSize(metadata));
+    if (state.counters.find("page_index_size") == state.counters.end()) {
+      state.PauseTiming();
+      auto metadata = parquet::ReadMetaData(
+          std::make_shared<::arrow::io::BufferReader>(output->Finish().ValueOrDie()));
+      state.counters["output_size"] = static_cast<double>(output->Tell().ValueOrDie());
+      state.counters["page_index_size"] =
+          static_cast<double>(GetTotalPageIndexSize(metadata));
+      state.ResumeTiming();
+    }
   }
 
   state.SetItemsProcessed(state.iterations() * kBenchmarkSize);
@@ -158,5 +160,3 @@ BENCHMARK_TEMPLATE(BM_WriteListColumn, SizeStatisticsLevel::PageAndColumnChunk,
                    ::arrow::StringType);
 
 }  // namespace parquet::benchmark
-
-BENCHMARK_MAIN();
diff --git a/cpp/submodules/parquet-testing b/cpp/submodules/parquet-testing
+ −		bad_data/ARROW-RS-GH-6229-DICTHEADER.parquet
+20 −1		data/README.md
+ −		data/alltypes_dictionary.parquet
+ −		data/alltypes_plain.parquet
+ −		data/alltypes_plain.snappy.parquet
+ −		data/encrypt_columns_and_footer.parquet.encrypted
+ −		data/encrypt_columns_and_footer_aad.parquet.encrypted
+ −		data/encrypt_columns_and_footer_ctr.parquet.encrypted
+ −		data/encrypt_columns_and_footer_disable_aad_storage.parquet.encrypted
+ −		data/encrypt_columns_plaintext_footer.parquet.encrypted
+164 −0		data/map_no_value.md
+ −		data/map_no_value.parquet
+ −		data/nan_in_stats.parquet
+ −		data/repeated_primitive_no_list.parquet
+ −		data/uniform_encryption.parquet.encrypted