This is an automated email from the ASF dual-hosted git repository. jiayu pushed a commit to branch main in repository https://gitbox.apache.org/repos/asf/sedona-spatialbench.git
commit 306343d88be2c29377cce1e56df0753a42fc007a Author: Pranav Toggi <[email protected]> AuthorDate: Mon Jun 30 09:25:54 2025 -0700 suppress non-sedonabench tables --- tpchgen-cli/src/main.rs | 141 ++++++++++++++++++++++++------------------------ 1 file changed, 71 insertions(+), 70 deletions(-) diff --git a/tpchgen-cli/src/main.rs b/tpchgen-cli/src/main.rs index fa2449a..34d2eec 100644 --- a/tpchgen-cli/src/main.rs +++ b/tpchgen-cli/src/main.rs @@ -130,13 +130,13 @@ struct Cli { #[derive(Debug, Copy, Clone, PartialEq, Eq, PartialOrd, Ord)] enum Table { - Nation, - Region, + // Nation, + // Region, Vehicle, Driver, Customer, - Orders, - Lineitem, + // Orders, + // Lineitem, Trip, Building, } @@ -172,13 +172,13 @@ impl TypedValueParser for TableValueParser { ) -> Option<Box<dyn Iterator<Item = clap::builder::PossibleValue> + '_>> { Some(Box::new( [ - clap::builder::PossibleValue::new("region").help("Region table (alias: r)"), - clap::builder::PossibleValue::new("nation").help("Nation table (alias: n)"), + // clap::builder::PossibleValue::new("region").help("Region table (alias: r)"), + // clap::builder::PossibleValue::new("nation").help("Nation table (alias: n)"), clap::builder::PossibleValue::new("driver").help("Driver table (alias: d)"), clap::builder::PossibleValue::new("customer").help("Customer table (alias: c)"), clap::builder::PossibleValue::new("vehicle").help("Vehicle table (alias: V)"), - clap::builder::PossibleValue::new("orders").help("Orders table (alias: O)"), - clap::builder::PossibleValue::new("lineitem").help("LineItem table (alias: L)"), + // clap::builder::PossibleValue::new("orders").help("Orders table (alias: O)"), + // clap::builder::PossibleValue::new("lineitem").help("LineItem table (alias: L)"), clap::builder::PossibleValue::new("trip").help("Trip table (alias: T)"), clap::builder::PossibleValue::new("building").help("Trip table (alias: b)"), ] @@ -198,13 +198,13 @@ impl FromStr for Table { /// only support the exclusive abbreviations. fn from_str(s: &str) -> Result<Self, Self::Err> { match s { - "n" | "nation" => Ok(Table::Nation), - "r" | "region" => Ok(Table::Region), + // "n" | "nation" => Ok(Table::Nation), + // "r" | "region" => Ok(Table::Region), "d" | "driver" => Ok(Table::Driver), "V" | "vehicle" => Ok(Table::Vehicle), "c" | "customer" => Ok(Table::Customer), - "O" | "orders" => Ok(Table::Orders), - "L" | "lineitem" => Ok(Table::Lineitem), + // "O" | "orders" => Ok(Table::Orders), + // "L" | "lineitem" => Ok(Table::Lineitem), "T" | "trip" => Ok(Table::Trip), "b" | "building" => Ok(Table::Building), _ => Err("Invalid table name {s}"), @@ -215,13 +215,13 @@ impl FromStr for Table { impl Table { fn name(&self) -> &'static str { match self { - Table::Nation => "nation", - Table::Region => "region", + // Table::Nation => "nation", + // Table::Region => "region", Table::Vehicle => "vehicle", Table::Driver => "driver", Table::Customer => "customer", - Table::Orders => "orders", - Table::Lineitem => "lineitem", + // Table::Orders => "orders", + // Table::Lineitem => "lineitem", Table::Trip => "trip", Table::Building => "building", } @@ -295,14 +295,15 @@ impl Cli { tables.clone() } else { vec![ - Table::Nation, - Table::Region, + // Table::Nation, + // Table::Region, Table::Vehicle, Table::Driver, Table::Customer, - Table::Orders, - Table::Lineitem, + // Table::Orders, + // Table::Lineitem, Table::Trip, + Table::Building, ] }; @@ -318,13 +319,13 @@ impl Cli { // Generate each table for table in tables { match table { - Table::Nation => self.generate_nation().await?, - Table::Region => self.generate_region().await?, + // Table::Nation => self.generate_nation().await?, + // Table::Region => self.generate_region().await?, Table::Vehicle => self.generate_vehicle().await?, Table::Driver => self.generate_driver().await?, Table::Customer => self.generate_customer().await?, - Table::Orders => self.generate_orders().await?, - Table::Lineitem => self.generate_lineitem().await?, + // Table::Orders => self.generate_orders().await?, + // Table::Lineitem => self.generate_lineitem().await?, Table::Trip => self.generate_trip().await?, Table::Building => self.generate_building().await?, } @@ -334,22 +335,22 @@ impl Cli { Ok(()) } - define_generate!( - generate_nation, - Table::Nation, - NationGenerator, - NationTblSource, - NationCsvSource, - NationArrow - ); - define_generate!( - generate_region, - Table::Region, - RegionGenerator, - RegionTblSource, - RegionCsvSource, - RegionArrow - ); + // define_generate!( + // generate_nation, + // Table::Nation, + // NationGenerator, + // NationTblSource, + // NationCsvSource, + // NationArrow + // ); + // define_generate!( + // generate_region, + // Table::Region, + // RegionGenerator, + // RegionTblSource, + // RegionCsvSource, + // RegionArrow + // ); define_generate!( generate_vehicle, Table::Vehicle, @@ -374,22 +375,22 @@ impl Cli { CustomerCsvSource, CustomerArrow ); - define_generate!( - generate_orders, - Table::Orders, - OrderGenerator, - OrderTblSource, - OrderCsvSource, - OrderArrow - ); - define_generate!( - generate_lineitem, - Table::Lineitem, - LineItemGenerator, - LineItemTblSource, - LineItemCsvSource, - LineItemArrow - ); + // define_generate!( + // generate_orders, + // Table::Orders, + // OrderGenerator, + // OrderTblSource, + // OrderCsvSource, + // OrderArrow + // ); + // define_generate!( + // generate_lineitem, + // Table::Lineitem, + // LineItemGenerator, + // LineItemTblSource, + // LineItemCsvSource, + // LineItemArrow + // ); define_generate!( generate_trip, Table::Trip, @@ -440,8 +441,8 @@ impl Cli { // Avg row size is an estimate of the average row size in bytes from the first 100 rows // of the table in tbl format let (avg_row_size_bytes, row_count) = match table { - Table::Nation => (88, 1), - Table::Region => (77, 1), + // Table::Nation => (88, 1), + // Table::Region => (77, 1), Table::Vehicle => ( 115, VehicleGenerator::calculate_row_count(self.scale_factor, 1, 1), @@ -454,18 +455,18 @@ impl Cli { 160, CustomerGenerator::calculate_row_count(self.scale_factor, 1, 1), ), - Table::Orders => ( - 114, - OrderGenerator::calculate_row_count(self.scale_factor, 1, 1), - ), - Table::Lineitem => { - // there are on average 4 line items per order. - // For example, in SF=10, - // * orders has 15,000,000 rows - // * lineitem has around 60,000,000 rows - let row_count = 4 * OrderGenerator::calculate_row_count(self.scale_factor, 1, 1); - (128, row_count) - }, + // Table::Orders => ( + // 114, + // OrderGenerator::calculate_row_count(self.scale_factor, 1, 1), + // ), + // Table::Lineitem => { + // // there are on average 4 line items per order. + // // For example, in SF=10, + // // * orders has 15,000,000 rows + // // * lineitem has around 60,000,000 rows + // let row_count = 4 * OrderGenerator::calculate_row_count(self.scale_factor, 1, 1); + // (128, row_count) + // }, &Table::Trip => (130, TripGenerator::calculate_row_count(self.scale_factor, 1, 1)), Table::Building => ( 115,
