fix build errors

Author: fastio

benchmarks/clickhouse-bench/src/main.rs

@@ -14,6 +14,8 @@ use vortex_bench::Opts;
 use vortex_bench::create_benchmark;
 use vortex_bench::create_output_writer;
 use vortex_bench::display::DisplayFormat;
+use vortex_bench::runner::BenchmarkMode;
+use vortex_bench::runner::BenchmarkQueryResult;
 use vortex_bench::runner::SqlBenchmarkRunner;
 use vortex_bench::runner::filter_queries;
 use vortex_bench::setup_logging_and_tracing;
@@ -59,6 +61,20 @@ struct Args {
     options: Vec<Opt>,
 }
 
+struct ClickHouseQueryResult {
+    row_count: usize,
+}
+
+impl BenchmarkQueryResult for ClickHouseQueryResult {
+    fn row_count(&self) -> usize {
+        self.row_count
+    }
+
+    fn display(self) -> String {
+        format!("{} rows", self.row_count)
+    }
+}
+
 fn main() -> anyhow::Result<()> {
     let args = Args::parse();
     let opts = Opts::from(args.options);
@@ -91,9 +107,14 @@ fn main() -> anyhow::Result<()> {
 
     runner.run_all(
         &filtered_queries,
-        args.iterations,
+        BenchmarkMode::Run {
+            iterations: args.iterations,
+        },
         |format| ClickHouseClient::new(benchmark.as_ref(), format),
-        |ctx, _query_idx, _format, query| ctx.execute_query(query),
+        |ctx, _query_idx, _format, query| {
+            let (row_count, duration) = ctx.execute_query(query)?;
+            Ok((duration, ClickHouseQueryResult { row_count }))
+        },
     )?;
 
     let benchmark_id = format!("clickhouse-{}", benchmark.dataset_name());