Try something

Signed-off-by: Adam Gutglick <[email protected]>

Author: AdamGS

benchmarks/ddb-bench/src/main.rs

@@ -8,6 +8,7 @@ use std::path::PathBuf;
 use clap::Parser;
 use clap::value_parser;
 use ddb_bench::DuckClient;
+use tokio::runtime::Runtime;
 use vortex_bench::BenchmarkArg;
 use vortex_bench::CompactionStrategy;
 use vortex_bench::Engine;
@@ -68,8 +69,7 @@ struct Args {
     options: Vec<Opt>,
 }
 
-#[tokio::main]
-async fn main() -> anyhow::Result<()> {
+fn main() -> anyhow::Result<()> {
     let args = Args::parse();
     let opts = Opts::from(args.options);
 
@@ -85,25 +85,31 @@ async fn main() -> anyhow::Result<()> {
 
     // Generate Vortex files from Parquet for any Vortex formats requested
     if benchmark.data_url().scheme() == "file" {
-        benchmark.generate_base_data().await?;
-
-        let base_path = benchmark
-            .data_url()
-            .to_file_path()
-            .map_err(|_| anyhow::anyhow!("Invalid file URL: {}", benchmark.data_url()))?;
-
-        for format in args.formats.iter().copied() {
-            match format {
-                Format::OnDiskVortex => {
-                    convert_parquet_to_vortex(&base_path, CompactionStrategy::Default).await?;
+        let runtime = Runtime::new()?;
+
+        runtime.block_on(async {
+            benchmark.generate_base_data().await?;
+
+            let base_path = benchmark
+                .data_url()
+                .to_file_path()
+                .map_err(|_| anyhow::anyhow!("Invalid file URL: {}", benchmark.data_url()))?;
+
+            for format in args.formats.iter().copied() {
+                match format {
+                    Format::OnDiskVortex => {
+                        convert_parquet_to_vortex(&base_path, CompactionStrategy::Default).await?;
+                    }
+                    Format::VortexCompact => {
+                        convert_parquet_to_vortex(&base_path, CompactionStrategy::Compact).await?;
+                    }
+                    // OnDiskDuckDB tables are created during register_tables by loading from Parquet
+                    _ => {}
                 }
-                Format::VortexCompact => {
-                    convert_parquet_to_vortex(&base_path, CompactionStrategy::Compact).await?;
-                }
-                // OnDiskDuckDB tables are created during register_tables by loading from Parquet
-                _ => {}
             }
-        }
+
+            anyhow::Ok(())
+        })?;
     }
 
     let mut runner = SqlBenchmarkRunner::new(